first232hjhkhk

app.py

@@ -4,96 +4,53 @@ import tempfile
 import os
 import shutil
 from urllib.parse import urlparse
+import re
 
-# [Previous helper functions remain the same]
-def validate_repo_url(url):
-    """Validate if the input is a valid repository URL or GitHub shorthand."""
-    if not url:
-        return False
+def get_repo_details(repo_url):
+    """Extract username and repo name from GitHub URL or shorthand."""
+    # Handle GitHub shorthand (user/repo)
+    if '/' in repo_url and len(repo_url.split('/')) == 2:
+        username, repo = repo_url.split('/')
+        return username, repo.replace('.git', '')
     
-    # Check if it's a GitHub shorthand (user/repo)
-    if '/' in url and len(url.split('/')) == 2:
-        return True
-        
+    # Handle full GitHub URL
     try:
-        result = urlparse(url)
-        return all([result.scheme, result.netloc])
+        # Parse URL and path
+        parsed = urlparse(repo_url)
+        path_parts = parsed.path.strip('/').split('/')
+        
+        # Extract username and repo
+        if len(path_parts) >= 2:
+            username = path_parts[0]
+            repo = path_parts[1].replace('.git', '')
+            return username, repo
     except:
-        return False
+        pass
+    
+    return None, None
 
-def pack_repository(repo_url, branch, output_style, remove_comments, remove_empty_lines, security_check):
-    """Pack a repository using Repomix and return the output."""
-    if not repo_url:
-        return "Error: Please provide a repository URL"
-        
-    if not validate_repo_url(repo_url):
-        return "Error: Invalid repository URL format"
-        
-    temp_dir = None
-    try:
-        # Create temporary directory that persists until we're done
-        temp_dir = tempfile.mkdtemp()
-        
-        # Prepare command
-        cmd = ["npx", "repomix"]
-        
-        # Add remote repository options
-        cmd.extend(["--remote", repo_url])
-        
-        if branch:
-            cmd.extend(["--remote-branch", branch])
-            
-        # Add style option
-        cmd.extend(["--style", output_style])
-        
-        # Add other options
-        if remove_comments:
-            cmd.append("--remove-comments")
-        if remove_empty_lines:
-            cmd.append("--remove-empty-lines")
-        if not security_check:
-            cmd.append("--no-security-check")
-            
-        # Set output path
-        output_file = os.path.join(temp_dir, "repomix-output.txt")
-        cmd.extend(["-o", output_file])
-        
-        # Execute Repomix
-        result = subprocess.run(
-            cmd, 
-            capture_output=True, 
-            text=True,
-            cwd=temp_dir
-        )
-        
-        if result.returncode != 0:
-            return f"Error running Repomix: {result.stderr}"
-            
-        # Check if the file exists
-        if not os.path.exists(output_file):
-            return f"Error: Output file was not created. Repomix output: {result.stdout}\n{result.stderr}"
-            
-        # Read the output file
-        try:
-            with open(output_file, 'r', encoding='utf-8') as f:
-                content = f.read()
-            return content
-        except Exception as e:
-            return f"Error reading output file: {str(e)}"
-            
-    except Exception as e:
-        return f"Error: {str(e)}"
-        
-    finally:
-        # Clean up temporary directory
-        if temp_dir and os.path.exists(temp_dir):
-            try:
-                shutil.rmtree(temp_dir)
-            except Exception as e:
-                print(f"Warning: Could not remove temporary directory: {str(e)}")
+def format_filename(repo_url, branch):
+    """Generate filename based on repository details."""
+    username, repo_name = get_repo_details(repo_url)
+    if not username or not repo_name:
+        return "repository-output.txt"
+    
+    # Clean and format branch name
+    branch = branch if branch else "main"
+    branch = re.sub(r'[^\w\-\.]', '-', branch)
+    
+    return f"{username}-{repo_name}-{branch}.txt"
+
+[Previous helper functions remain the same]
 
 # Create the Gradio interface
-with gr.Blocks(title="Repo to TXT", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Repomix Web Interface", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("""
+    # 📦 Repomix Web Interface
+    Pack your GitHub repository into a single, AI-friendly file. Perfect for use with LLMs like Claude, ChatGPT, and Gemini.
+    
+    Enter a GitHub repository URL (e.g., `https://github.com/user/repo`) or shorthand format (e.g., `user/repo`).
+    """)
     
     with gr.Row():
         with gr.Column():
@@ -134,15 +91,52 @@ with gr.Blocks(title="Repo to TXT", theme=gr.themes.Soft()) as demo:
             label="Output",
             placeholder="Packed repository content will appear here...",
             lines=20,
-            show_copy_button=True  # Enable built-in copy button
+            show_copy_button=True
+        )
+    
+    # Add download button
+    with gr.Row():
+        download_button = gr.File(
+            label="Download Output",
+            file_count="single",
+            type="file",
+            interactive=False
         )
     
     # Handle the pack button click
+    def on_pack(repo_url, branch, output_style, remove_comments, remove_empty_lines, security_check):
+        content = pack_repository(repo_url, branch, output_style, remove_comments, remove_empty_lines, security_check)
+        
+        # Generate file for download if content is not an error message
+        if not content.startswith("Error"):
+            filename = format_filename(repo_url, branch)
+            temp_path = tempfile.mktemp(suffix='.txt')
+            
+            with open(temp_path, 'w', encoding='utf-8') as f:
+                f.write(content)
+            
+            return content, temp_path
+        
+        return content, None
+    
     pack_button.click(
-        fn=pack_repository,
+        fn=on_pack,
         inputs=[repo_url, branch, output_style, remove_comments, remove_empty_lines, security_check],
-        outputs=output_text
+        outputs=[output_text, download_button]
     )
+    
+    gr.Markdown("""
+    ### 📝 Notes
+    - The packed output is optimized for use with AI models
+    - Security check helps identify potentially sensitive information
+    - Different output styles (plain, XML, markdown) are available for different use cases
+    - Click the copy icon in the output box to copy the content to clipboard
+    - Use the download button to save the output as a text file
+    
+    ### 🔗 Links
+    - [Repomix GitHub Repository](https://github.com/yamadashy/repomix)
+    - [Documentation](https://github.com/yamadashy/repomix#-quick-start)
+    """)
 
 if __name__ == "__main__":
     demo.launch()