Update app.py

app.py

@@ -6,14 +6,10 @@ import shutil
 from urllib.parse import urlparse
 
 def validate_repo_url(url):
-    """Validate if the input is a valid repository URL or GitHub shorthand."""
     if not url:
         return False
-    
-    # Check if it's a GitHub shorthand (user/repo)
     if '/' in url and len(url.split('/')) == 2:
         return True
-        
     try:
         result = urlparse(url)
         return all([result.scheme, result.netloc])
@@ -21,12 +17,9 @@ def validate_repo_url(url):
         return False
 
 def create_download_file(content):
-    """Create a temporary file for download and return its path."""
     if not content:
         return None
-    
     try:
-        # Create a temporary file
         temp_file = tempfile.NamedTemporaryFile(delete=False, mode='w', suffix='.txt')
         temp_file.write(content)
         temp_file.close()
@@ -35,8 +28,7 @@ def create_download_file(content):
         print(f"Error creating download file: {str(e)}")
         return 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."""
+def pack_repository(repo_url, branch, gh_token, output_style, remove_comments, remove_empty_lines, security_check):
     if not repo_url:
         return "Error: Please provide a repository URL", None
         
@@ -45,22 +37,18 @@ def pack_repository(repo_url, branch, output_style, remove_comments, remove_empt
         
     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
+        if gh_token:
+            cmd.extend(["--gh-token", gh_token])
+            
         cmd.extend(["--style", output_style])
         
-        # Add other options
         if remove_comments:
             cmd.append("--remove-comments")
         if remove_empty_lines:
@@ -68,11 +56,9 @@ def pack_repository(repo_url, branch, output_style, remove_comments, remove_empt
         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, 
@@ -83,16 +69,13 @@ def pack_repository(repo_url, branch, output_style, remove_comments, remove_empt
         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()
             
-            # Create a download file and return its path
             download_path = create_download_file(content)
             return content, download_path
             
@@ -103,16 +86,13 @@ def pack_repository(repo_url, branch, output_style, remove_comments, remove_empt
         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="Repo to TXT", theme=gr.themes.Soft()) as demo:
-    
     with gr.Row():
         with gr.Column():
             repo_url = gr.Textbox(
@@ -123,6 +103,11 @@ with gr.Blocks(title="Repo to TXT", theme=gr.themes.Soft()) as demo:
                 label="Branch/Tag/Commit (optional)",
                 placeholder="e.g., main, master, v1.0.0"
             )
+            gh_token = gr.Textbox(
+                label="GitHub Token (for private repos)",
+                placeholder="ghp_xxxxxxxxxxxxxx",
+                type="password"
+            )
             
             with gr.Row():
                 output_style = gr.Radio(
@@ -163,10 +148,9 @@ with gr.Blocks(title="Repo to TXT", theme=gr.themes.Soft()) as demo:
             interactive=False
         )
     
-    # Handle the pack button click
     pack_button.click(
         fn=pack_repository,
-        inputs=[repo_url, branch, output_style, remove_comments, remove_empty_lines, security_check],
+        inputs=[repo_url, branch, gh_token, output_style, remove_comments, remove_empty_lines, security_check],
         outputs=[output_text, download_button]
     )