Fixed endpoints

app.py

@@ -14,9 +14,9 @@ from dashboard_analytics import dashboard_router
 app = FastAPI(title="EY Catalyst Risk Analysis API", version="1.0.0")
 
 # Include the routers
-app.mount("/enterprise", enterprise_ra_router)
-app.mount("/threat", threat_ra_router) 
-app.mount("/dashboard", dashboard_router)
+app.include_router(enterprise_ra_router, prefix="/enterprise")
+app.include_router(threat_ra_router, prefix="/threat") 
+app.include_router(dashboard_router, prefix="/dashboard")
 
 # Environment Variables
 GROQ_API_KEY = os.environ.get("GROQ_API_KEY").strip()

dashboard_analytics.py

@@ -1,5 +1,5 @@
 # dashboard_analytics.py
-from fastapi import FastAPI, HTTPException
+from fastapi import APIRouter, HTTPException
 from pydantic import BaseModel
 from typing import List, Optional, Dict, Any
 import os
@@ -73,7 +73,7 @@ class RecentActivitiesResponse(BaseModel):
     message: str
 
 # Dashboard Analytics Router
-dashboard_router = FastAPI()
+dashboard_router = APIRouter()
 
 @dashboard_router.post("/api/dashboard/generate-kpis", response_model=DashboardKPIResponse)
 def generate_dashboard_kpis(request: DashboardAnalyticsRequest):

enterprise_ra.py

@@ -1,5 +1,5 @@
 # enterprise_ra.py
-from fastapi import FastAPI, HTTPException
+from fastapi import APIRouter, HTTPException
 from pydantic import BaseModel
 from typing import List, Optional
 import os
@@ -68,7 +68,7 @@ class ThreatGenerationResponse(BaseModel):
     message: str
 
 # Enterprise RA Router
-enterprise_ra_router = FastAPI()
+enterprise_ra_router = APIRouter()
 
 @enterprise_ra_router.post("/api/enterprise-ra/generate-risks", response_model=RiskGenerationResponse)
 def generate_enterprise_risks(request: RiskGenerationRequest):

threat_ra.py

@@ -1,5 +1,5 @@
 # threat_ra.py
-from fastapi import FastAPI, HTTPException
+from fastapi import APIRouter, HTTPException
 from pydantic import BaseModel
 from typing import List, Optional
 import os
@@ -66,7 +66,7 @@ class ThreatRiskAnalysisResponse(BaseModel):
     message: str
 
 # Threat RA Router
-threat_ra_router = FastAPI()
+threat_ra_router = APIRouter()
 
 @threat_ra_router.post("/api/threat-ra/generate-threat-risks", response_model=ThreatRiskGenerationResponse)
 def generate_threat_risks(request: ThreatRiskGenerationRequest):