Update app/main.py

app/main.py

@@ -29,6 +29,8 @@ app = FastAPI(title="GitGud AI Service")
 # Global embedding cache
 REPO_CACHE: Dict[str, Dict[str, List[float]]] = {}
 
+executor = ThreadPoolExecutor(max_workers=4)
+
 # 3. Data Models
 class FileRequest(BaseModel):
     fileName: str
@@ -58,10 +60,8 @@ def calculate_repo_health(total_vulns: int, avg_maint: float) -> int:
     penalty = total_vulns * 8
     return int(max(10, min(100, base_score - penalty)))
 
-def sync_review_worker(file_req: FileRequest):
-    """Wrapper for the synchronous reviewer service"""
-    # Note: We pass the individual file object directly
-    return reviewer.service.review_batch_code([file_req])
+def sync_review_worker(file_list: List[FileRequest]):
+    return reviewer.service.review_batch_code(file_list)
 
 def parse_tree_to_list(raw_tree: str):
     """Parses text tree into JSON objects for Compose LazyColumn"""
@@ -117,31 +117,9 @@ async def classify_file(request: FileRequest):
 async def review_batch_code(request: BatchReviewRequest):
     try:
         loop = asyncio.get_running_loop()
-        
-        # Dispatch all files to the thread pool immediately
-        tasks = [
-            loop.run_in_executor(executor, sync_review_worker, f) 
-            for f in request.files
-        ]
-        
-        # Wait for all to complete
-        raw_results = await asyncio.gather(*tasks, return_exceptions=True)
-
-        final_reviews = []
-        for i, result in enumerate(raw_results):
-            if isinstance(result, Exception):
-                logger.error(f"Error reviewing {request.files[i].fileName}: {result}")
-                # We return a placeholder error for this specific file so the UI knows it failed
-                final_reviews.append({
-                    "fileName": request.files[i].fileName,
-                    "error": "Analysis failed",
-                    "details": str(result)
-                })
-            else:
-                final_reviews.append(result)
-
+        raw_results = await loop.run_in_executor(executor, sync_review_worker, request.files)
+        final_reviews = raw_results if isinstance(raw_results, list) else [raw_results]
         return {"results": final_reviews}
-
     except Exception as e:
         logger.error(f"Batch review critical failure: {traceback.format_exc()}")
         raise HTTPException(status_code=500, detail="Internal processing error")
@@ -150,31 +128,23 @@ async def review_batch_code(request: BatchReviewRequest):
 async def get_dashboard_stats(request: BatchReviewRequest):
     try:
         loop = asyncio.get_running_loop()
-        
-        # Parallel execution for stats
-        tasks = [loop.run_in_executor(executor, sync_review_worker, f) for f in request.files]
-        raw_reviews = await asyncio.gather(*tasks, return_exceptions=True)
+        raw_reviews = await loop.run_in_executor(executor, sync_review_worker, request.files)
+        if not isinstance(raw_reviews, list):
+            raw_reviews = [raw_reviews]
 
         total_vulns = 0
         maint_scores = []
         found_apis = set()
         api_regex = re.compile(r'(?:get|post|put|delete|patch)\([\'"]\/(.*?)[\'"]', re.IGNORECASE)
 
-        for i, review in enumerate(raw_reviews):
-            if isinstance(review, Exception):
-                continue 
-
-            # Handle the structure returned by your reviewer service
-            # (Assuming it returns a list or a dict based on your current service)
-            current_review = review[0] if isinstance(review, list) else review
-            
+        for i, current_review in enumerate(raw_reviews):
             vulns = current_review.get("vulnerabilities", [])
             total_vulns += len(vulns)
-            
+
             m_score = current_review.get("metrics", {}).get("maintainability", 8.0)
             maint_scores.append(m_score)
-            
-            content = request.files[i].content
+
+            content = request.files[i].content if i < len(request.files) else None
             if content:
                 matches = api_regex.findall(content)
                 for match in matches: