Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 from fastapi import FastAPI, Request, HTTPException
 import uvicorn
 from gradio_client import Client
+import uuid
 
 # Keys
 VALID_API_KEY = os.getenv("APP_API_KEY")       # for your FastAPI security
@@ -24,8 +25,12 @@ async def chat(request: Request):
         raise HTTPException(status_code=403, detail="Invalid API Key")
 
     user_message = data.get("message")
-    # chat_history = data.get("chat_history", []) # Chat history is handled by Gradio's ChatInterface internally
-    user_id = data.get("user_id") # Extract user_id from request
+    # Check for user_id in the request data, generate one if not present
+    user_id = data.get("user_id")
+    if not user_id:
+        user_id = str(uuid.uuid4()) # Generate a unique ID if none is provided
+        print(f"Generated user_id: {user_id} for this request.")
+
 
     if not user_message:
         raise HTTPException(status_code=400, detail="Message is required")
@@ -40,7 +45,7 @@ async def chat(request: Request):
     )
 
 
-    return {"response": result}
+    return {"response": result, "user_id": user_id} # Return the user_id so the client can potentially reuse it
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)