Update app.py

app.py

@@ -3,8 +3,9 @@ from fastapi.middleware.cors import CORSMiddleware
 from model import predict_logic
 import uvicorn
 
-app = FastAPI()
+app = FastAPI(title="Titanium Protocol API")
 
+# Enable CORS for Vercel Frontend
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -13,11 +14,17 @@ app.add_middleware(
 )
 
 @app.get("/")
-def home(): return {"status": "Titanium Brain Active"}
+def health_check():
+    """Health check for Hugging Face monitoring."""
+    return {"status": "Titanium Protocol Online"}
 
 @app.post("/predict")
-async def predict_endpoint(book_text: str = Form(...), backstory: str = Form(...)):
-    return predict_logic(book_text, backstory)
+async def handle_prediction(book_text: str = Form(...), backstory: str = Form(...)):
+    """The main endpoint called by your frontend."""
+    # This calls the logic we defined in model.py
+    result = predict_logic(book_text, backstory)
+    return result
 
 if __name__ == "__main__":
+    # Standard port for Hugging Face Spaces
     uvicorn.run(app, host="0.0.0.0", port=7860)