Upload folder using huggingface_hub

server.py

@@ -3,27 +3,46 @@ import traceback
 import uuid
 from fastapi import APIRouter, Request
 from fastapi.responses import JSONResponse
-from gradio_client import Client
 from pydantic import BaseModel
 from app import chat
 
 router = APIRouter()
 
+# In-memory mapping from session_id -> thread_id
+# For production, you may want Redis or a DB for persistence
+thread_map = {}
 
 class Message(BaseModel):
     text: str
-
+    session_id: str | None = None  # Optional session ID from client
 
 @router.post("/chat")
-async def handle_chat(msg: Message):
+async def handle_chat(msg: Message, request: Request):
     try:
-        thread_id = uuid.uuid4()
+        # Use existing session_id if provided, else generate new
+        session_id = msg.session_id
+        if not session_id:
+            session_id = str(uuid.uuid4())
+
+        # Get or create a persistent thread_id for this session
+        if session_id not in thread_map:
+            thread_map[session_id] = str(uuid.uuid4())
+        thread_id = thread_map[session_id]
+
+        # Call your graph/chat function
         reply_text = chat(
-            debug_mode=False, message=msg.text, history=None, thread_id=thread_id
+            debug_mode=False,
+            message=msg.text,
+            history=None,
+            thread_id=thread_id
         )
 
-        return {"reply": reply_text}
+        # Return both reply and session_id to the client
+        return {"reply": reply_text, "session_id": session_id}
 
     except Exception as e:
         traceback.print_exc()
-        return {"reply": f"Error: {e}"}
+        return JSONResponse(
+            status_code=500,
+            content={"reply": f"Error: {e}"}
+        )