Update app.py

app.py

@@ -3,6 +3,8 @@ from pydantic import BaseModel
 from typing import Optional
 from datetime import datetime
 import logging
+import uvicorn
+import os
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -29,7 +31,7 @@ class VendorOutput(BaseModel):
     Alert_Flag__c: bool
 
 # API token for security (replace with your actual token or env variable)
-EXPECTED_API_TOKEN = "hf_your_secure_token_here"
+EXPECTED_API_TOKEN = os.getenv("API_TOKEN", "hf_your_secure_token_here")
 
 # Scoring function
 def calculate_vendor_score(data: VendorInput) -> VendorOutput:
@@ -97,4 +99,9 @@ async def score_vendor(data: VendorInput, authorization: Optional[str] = Header(
 # Root endpoint for health check
 @app.get("/")
 async def root():
-    return {"message": "Vendor Scoring API is running"}
+    return {"message": "Vendor Scoring API is running"}
+
+# Start Uvicorn server when running directly
+if __name__ == "__main__":
+    port = int(os.getenv("PORT", 7860))  # Default to 7860 for Hugging Face Spaces
+    uvicorn.run(app, host="0.0.0.0", port=port)