Sync with local repository - 2025-08-12 08:12:32

README.md

@@ -79,4 +79,4 @@ This project is licensed under the MIT License.
 
 ---
 
-**Note**: This application processes public repositories only. Private repositories require authentication tokens.
+**Note**: This application processes public repositories only. Private repositories require authentication tokens.

app.py

@@ -49,14 +49,16 @@ def validate_github_url(url: str) -> bool:
 async def process_repository(repo_url: str, branch: str = "main") -> tuple:
     """Process a GitHub repository and return status"""
     if not validate_github_url(repo_url):
-        return "❌ Error", "Invalid GitHub URL. Please provide a valid GitHub repository URL."
+        yield "❌ Error", "Invalid GitHub URL. Please provide a valid GitHub repository URL."
+        return
     
     try:
         repo_id = github_service.generate_repo_id(repo_url)
         
         # Check if already processed
         if repo_id in repo_data:
-            return "✅ Ready", f"Repository already processed! You can now ask questions about the code."
+            yield "✅ Ready", f"Repository already processed! You can now ask questions about the code."
+            return
         
         # Clone repository
         yield "🔄 Processing", "Cloning repository..."
@@ -68,7 +70,8 @@ async def process_repository(repo_url: str, branch: str = "main") -> tuple:
         
         if not files:
             github_service.cleanup_repo(repo_path)
-            return "❌ Error", "No supported files found in the repository."
+            yield "❌ Error", "No supported files found in the repository."
+            return
         
         # Create embeddings
         yield "🔄 Processing", f"Creating embeddings for {len(files)} files (this may take a while)..."
@@ -88,30 +91,25 @@ async def process_repository(repo_url: str, branch: str = "main") -> tuple:
         yield "✅ Ready", f"Repository processed successfully! Found {len(files)} files. You can now ask questions about the code."
         
     except Exception as e:
-        return "❌ Error", f"Error processing repository: {str(e)}"
+        yield "❌ Error", f"Error processing repository: {str(e)}"
+        return
 
 def process_repo_sync(repo_url: str, branch: str = "main"):
-    """Synchronous wrapper for repository processing"""
+    """Synchronous wrapper for repository processing that streams updates to Gradio."""
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
     try:
-        # Run the async function
-        loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(loop)
-        
-        async def run_process():
-            async for status, message in process_repository(repo_url, branch):
+        agen = process_repository(repo_url, branch)
+        while True:
+            try:
+                status, message = loop.run_until_complete(agen.__anext__())
                 yield status, message
-        
-        # Get the final result
-        gen = run_process()
-        result = None
-        async for result in gen:
-            pass
-        
-        loop.close()
-        return result if result else ("❌ Error", "Processing failed")
-        
+            except StopAsyncIteration:
+                break
     except Exception as e:
-        return "❌ Error", f"Error: {str(e)}"
+        yield "❌ Error", f"Error: {str(e)}"
+    finally:
+        loop.close()
 
 async def chat_with_repository(message: str, repo_url: str, history: List) -> tuple:
     """Chat with the processed repository"""