Update app/main.py

app/main.py

@@ -1,7 +1,8 @@
 # main.py - Final Fixed Version
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
-from typing import List, Optional  
+from typing import List, Optional
+import re
 import logging
 
 from app.services.reviewer_service import AIReviewerService
@@ -74,6 +75,45 @@ async def review_code(request: ReviewRequest):
         logger.error(f"Review endpoint failed: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
+@app.post("/repo-dashboard-stats")
+async def get_dashboard_stats(request: ReviewRequest):
+    """
+    Specific endpoint for the Jetpack Compose Dashboard.
+    Combines CodeBERT classification and Gemini review stats.
+    """
+    try:
+        # 1. Run the existing Review Logic
+        raw_reviews = reviewer_service.review_batch_code(request.files)
+        
+        # 2. Extract Security Count
+        total_vulns = sum(len(r.get("vulnerabilities", [])) for r in raw_reviews)
+        
+        # 3. Calculate Performance (Maintainability)
+        scores = [r.get("metrics", {}).get("maintainability", 8) for r in raw_reviews]
+        avg_maintainability = (sum(scores) / len(scores)) * 10 if scores else 0
+        
+        # 4. Extract Exposed APIs (Sniffing from content)
+        found_apis = []
+        for f in request.files:
+            if f.content:
+                # Simple regex to find common route patterns
+                matches = re.findall(r'(?:get|post|put|delete|patch)\([\'"]\/(.*?)[\'"]', f.content.lower())
+                for match in matches:
+                    found_apis.append(f"/{match}")
+
+        # 5. Repo Health Logic
+        health_score = max(10, 100 - (total_vulns * 10))
+
+        return {
+            "repo_health": f"{health_score}%",
+            "security_issues": total_vulns,
+            "performance_ratio": f"{int(avg_maintainability)}%",
+            "exposed_apis": list(set(found_apis))[:10] # Top 10 unique routes
+        }
+    except Exception as e:
+        logger.error(f"Dashboard stats failed: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+
 # 5. Application Entry Point
 if __name__ == "__main__":
     import uvicorn