first232hjhkhkdsafdsf

app.py

@@ -4,53 +4,95 @@ import tempfile
 import os
 import shutil
 from urllib.parse import urlparse
-import re
 
-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', '')
+def validate_repo_url(url):
+    """Validate if the input is a valid repository URL or GitHub shorthand."""
+    if not url:
+        return False
     
-    # Handle full GitHub URL
-    try:
-        # Parse URL and path
-        parsed = urlparse(repo_url)
-        path_parts = parsed.path.strip('/').split('/')
+    # Check if it's a GitHub shorthand (user/repo)
+    if '/' in url and len(url.split('/')) == 2:
+        return True
         
-        # Extract username and repo
-        if len(path_parts) >= 2:
-            username = path_parts[0]
-            repo = path_parts[1].replace('.git', '')
-            return username, repo
+    try:
+        result = urlparse(url)
+        return all([result.scheme, result.netloc])
     except:
-        pass
-    
-    return None, None
-
-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"
+        return False
 
-[Previous helper functions remain the same]
+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", None
+        
+    if not validate_repo_url(repo_url):
+        return "Error: Invalid repository URL format", None
+        
+    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}", None
+            
+        # 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}", None
+            
+        # Read the output file
+        try:
+            with open(output_file, 'r', encoding='utf-8') as f:
+                content = f.read()
+            return content, content
+        except Exception as e:
+            return f"Error reading output file: {str(e)}", None
+            
+    except Exception as e:
+        return f"Error: {str(e)}", None
+        
+    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)}")
 
 # Create the Gradio interface
-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.Blocks(title="Repo to TXT", theme=gr.themes.Soft()) as demo:
     
     with gr.Row():
         with gr.Column():
@@ -93,8 +135,7 @@ with gr.Blocks(title="Repomix Web Interface", theme=gr.themes.Soft()) as demo:
             lines=20,
             show_copy_button=True
         )
-    
-    # Add download button
+        
     with gr.Row():
         download_button = gr.File(
             label="Download Output",
@@ -104,39 +145,11 @@ with gr.Blocks(title="Repomix Web Interface", theme=gr.themes.Soft()) as demo:
         )
     
     # 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=on_pack,
+        fn=pack_repository,
         inputs=[repo_url, branch, output_style, remove_comments, remove_empty_lines, security_check],
         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()