first23

app.py

@@ -22,50 +22,74 @@ def validate_repo_url(url):
 
 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
-        with tempfile.TemporaryDirectory() as temp_dir:
-            # Prepare command
-            cmd = ["npx", "repomix"]
-            
-            # Add remote repository options
-            cmd.extend(["--remote", repo_url])
+        # 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])
             
-            if branch:
-                cmd.extend(["--remote-branch", branch])
-                
-            # Add style option
-            cmd.extend(["--style", output_style])
+        # 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")
             
-            # 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])
+        # 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}"
             
-            # Execute Repomix
-            result = subprocess.run(cmd, 
-                                 capture_output=True, 
-                                 text=True,
-                                 cwd=temp_dir)
+        # 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}"
             
-            if result.returncode != 0:
-                return f"Error: {result.stderr}"
-                
-            # Read the output file
-            with open(output_file, 'r') as f:
+        # 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)}")
 
 # Create the Gradio interface
 with gr.Blocks(title="Repomix Web Interface", theme=gr.themes.Soft()) as demo: