Update app.py

app.py

@@ -1,11 +1,15 @@
-from fastapi import FastAPI, HTTPException
+from starlette.applications import Starlette
+from starlette.routing import Route
+from starlette.requests import Request
+from starlette.responses import JSONResponse
 from pydantic import BaseModel, Field
 from typing import List, Optional
 from datetime import datetime
 import uvicorn
 import numpy as np
+import json
 
-app = FastAPI(title="Vendor Performance Scoring API", description="API for calculating vendor performance scores with trend detection")
+app = Starlette(debug=True)
 
 # In-memory store for historical scores (replace with database for production)
 historical_scores = []
@@ -41,9 +45,12 @@ WEIGHTS = {
     "communication": 0.2
 }
 
-@app.post("/score", response_model=ScoreResponse)
-async def calculate_score(logs: VendorLog):
+async def calculate_score(request: Request):
     try:
+        # Parse JSON body
+        body = await request.json()
+        logs = VendorLog(**body)
+
         # Validate month format
         datetime.strptime(logs.month, "%Y-%m")
 
@@ -91,28 +98,44 @@ async def calculate_score(logs: VendorLog):
         # Mock certification URL (replace with actual logic)
         certification_url = f"https://example.com/cert/{logs.vendor_id}"
 
-        return {
-            "vendor_id": logs.vendor_id,
-            "month": logs.month,
-            "final_score": round(final_score, 2),
-            "quality_score": round(quality_score, 2),
-            "timeliness_score": round(timeliness_score, 2),
-            "safety_score": round(safety_score, 2),
-            "communication_score": round(communication_score, 2),
-            "alert_flag": alert_flag,
-            "trend_deviation": round(trend_deviation, 2) if trend_deviation is not None else None,
-            "certification_url": certification_url
-        }
+        # Prepare response
+        response = ScoreResponse(
+            vendor_id=logs.vendor_id,
+            month=logs.month,
+            final_score=round(final_score, 2),
+            quality_score=round(quality_score, 2),
+            timeliness_score=round(timeliness_score, 2),
+            safety_score=round(safety_score, 2),
+            communication_score=round(communication_score, 2),
+            alert_flag=alert_flag,
+            trend_deviation=round(trend_deviation, 2) if trend_deviation is not None else None,
+            certification_url=certification_url
+        )
+
+        return JSONResponse(content=response.dict())
 
     except ValueError as e:
-        raise HTTPException(status_code=400, detail=f"Invalid month format: {str(e)}")
+        return JSONResponse(
+            content={"detail": f"Invalid input: {str(e)}"},
+            status_code=400
+        )
+    except json.JSONDecodeError:
+        return JSONResponse(
+            content={"detail": "Invalid JSON format"},
+            status_code=400
+        )
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
+        return JSONResponse(
+            content={"detail": f"Internal server error: {str(e)}"},
+            status_code=500
+        )
+
+async def health_check(request: Request):
+    return JSONResponse(content={"status": "healthy"})
 
-# Health check endpoint
-@app.get("/health")
-async def health_check():
-    return {"status": "healthy"}
+# Define routes
+app.routes.append(Route("/score", endpoint=calculate_score, methods=["POST"]))
+app.routes.append(Route("/health", endpoint=health_check, methods=["GET"]))
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=8000)