Update app.py

app.py

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