Update app/main.py

app/main.py

@@ -113,14 +113,40 @@ async def classify_file(request: FileRequest):
 async def review_batch_code(request: BatchReviewRequest):
     try:
         loop = asyncio.get_event_loop()
-        with ThreadPoolExecutor() as executor:
-            reviews = await loop.run_in_executor(
-                executor, reviewer.service.review_batch_code, request.files
-            )
-        return {"results": reviews}
+        
+        # Helper function to process one file at a time, WITH the fix to unwrap the list
+        def process_single_file(file_req):
+            return reviewer.service.review_batch_code([file_req])
+
+        # Spin up a ThreadPoolExecutor with up to 15 concurrent workers
+        with ThreadPoolExecutor(max_workers=15) as executor:
+            # Create a concurrent task for every file in the request
+            tasks = [
+                loop.run_in_executor(executor, process_single_file, f) 
+                for f in request.files
+            ]
+            
+            # asyncio.gather fires them all off at the exact same time
+            raw_reviews = await asyncio.gather(*tasks, return_exceptions=True)
+
+        # Clean up the results and handle any individual file failures gracefully
+        valid_reviews = []
+        for i, result in enumerate(raw_reviews):
+            if isinstance(result, Exception):
+                # If we hit a rate limit, bubble it up immediately
+                if "429" in str(result):
+                    raise HTTPException(status_code=429, detail="AI Quota Exceeded")
+                
+                # Otherwise, log the specific file error but don't crash the whole batch
+                logger.error(f"Failed to analyze {request.files[i].fileName}: {result}")
+            else:
+                valid_reviews.append(result)
+
+        return {"results": valid_reviews}
+
     except Exception as e:
-        if "429" in str(e):
-            raise HTTPException(status_code=429, detail="AI Quota Exceeded")
+        if isinstance(e, HTTPException):
+            raise e
         traceback.print_exc()
         raise HTTPException(status_code=500, detail=str(e))
 
@@ -128,17 +154,30 @@ async def review_batch_code(request: BatchReviewRequest):
 async def get_dashboard_stats(request: BatchReviewRequest):
     try:
         loop = asyncio.get_event_loop()
-        with ThreadPoolExecutor() as executor:
-            raw_reviews = await loop.run_in_executor(
-                executor, reviewer.service.review_batch_code, request.files
-            )
+        
+        # 1. Use the same parallel helper for blazing fast stats!
+        def process_single_file(file_req):
+            return reviewer.service.review_batch_code([file_req])
+
+        # 2. Run all files concurrently
+        with ThreadPoolExecutor(max_workers=15) as executor:
+            tasks = [
+                loop.run_in_executor(executor, process_single_file, f) 
+                for f in request.files
+            ]
+            raw_reviews = await asyncio.gather(*tasks, return_exceptions=True)
 
         total_vulns = 0
         maint_scores = []
         found_apis = set()
         api_regex = re.compile(r'(?:get|post|put|delete|patch)\([\'"]\/(.*?)[\'"]', re.IGNORECASE)
 
+        # 3. Process the results, ignoring any parallel tasks that failed
         for i, review in enumerate(raw_reviews):
+            if isinstance(review, Exception):
+                logger.error(f"Skipping stats for {request.files[i].fileName} due to error: {review}")
+                continue # Skip failed files so the dashboard doesn't crash
+
             vulns = review.get("vulnerabilities", [])
             total_vulns += len(vulns)
             m_score = review.get("metrics", {}).get("maintainability", 8.0)
@@ -166,7 +205,7 @@ async def get_dashboard_stats(request: BatchReviewRequest):
     except Exception as e:
         if "429" in str(e):
             raise HTTPException(status_code=429, detail="Quota exceeded")
-        logger.error(f"Dashboard stats failed: {e}")
+        logger.error(f"Dashboard stats failed: {traceback.format_exc()}")
         raise HTTPException(status_code=500, detail="Failed to sync dashboard metrics")
 
 @app.post("/analyze-file")