Update app.py

app.py

@@ -110,23 +110,29 @@ convers_chain = RunnableWithMessageHistory(
 
 # =============FONCTION CHAT ================
 
-SESSION_ID = str(uuid.uuid4())  # session fixe pour tout le Space
+# SESSION_ID = str(uuid.uuid4())  # session fixe pour tout le Space
 
-def chat_fn(message, history):
-    result = convers_chain.invoke(
-        {"input": message},
-        config={"configurable": {"session_id": SESSION_ID}}
-    )
-    return result.get("answer", str(result))
-
-# def chat_fn(message, history, request:gr.Request):
-#     session_id=request.session_id or "default"
+# def chat_fn(message, history):
 #     result = convers_chain.invoke(
 #         {"input": message},
-#         config={"configurable": {"session_id": session_id}}
+#         config={"configurable": {"session_id": SESSION_ID}}
 #     )
 #     return result.get("answer", str(result))
 
+session_id_state = gr.State(value=None)
+
+def chat_fn(message, history, session_id_state):
+    if session_id_state is None:
+        session_id_state = str(uuid.uuid4())
+
+    result = convers_chain.invoke(
+        {"input": message},
+        config={"configurable": {"session_id": session_id_state}}
+    )
+
+    response = result.get("answer", str(result))
+    return response, session_id_state
+
 
 # ================= GRADIO =================