Update app.py

app.py

@@ -3,37 +3,38 @@ from fastapi import FastAPI, Request, HTTPException
 import uvicorn
 from gradio_client import Client
 
-# Connect to your first Hugging Face Space
-client = Client("Futuresony/Mr.Events")
+# --- Load environment keys ---
+HF_TOKEN = os.getenv("HF_TOKEN")       # Hugging Face token
+VALID_API_KEY = os.getenv("APP_API_KEY")  # Your app security key
 
-# Get your secure API key from environment
-VALID_API_KEY = os.getenv("APP_API_KEY")
+# --- Connect to Hugging Face Space ---
+client = Client("Futuresony/Mr.Events", hf_token=HF_TOKEN)
 
+# --- Create FastAPI app ---
 app = FastAPI()
 
 @app.post("/chat")
 async def chat(request: Request):
     data = await request.json()
 
-    # API Key Check
-    api_key = request.headers.get("X-API-Key") # Get API key from headers
+    # --- Check API Key from headers ---
+    api_key = request.headers.get("X-API-Key")
     if api_key != VALID_API_KEY:
         raise HTTPException(status_code=403, detail="Invalid API Key")
 
-    # Get user message
+    # --- Get user message ---
     user_message = data.get("message")
     if not user_message:
         raise HTTPException(status_code=400, detail="Message is required")
 
-    # Call the model hosted on your first space
+    # --- Call Hugging Face Space ---
     result = client.predict(
         query=user_message,
-        api_key=VALID_API_KEY,     # your hosted model's internal key if needed
-        api_name="/chat"
+        api_name="/chat"   # API name exposed by your Space
     )
 
     return {"response": result}
 
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)  # HF default port
+    uvicorn.run(app, host="0.0.0.0", port=7860)