Update app.py

app.py

@@ -64,9 +64,14 @@ def setup_multiple_chains():
     )
 
     conversation_memory = ConversationBufferMemory(
-        memory_key="chat_history", max_len=30, return_messages=True
+        memory_key="chat_history",
+        input_key="question",  # ?
+        output_key="text",     # ?
+        max_len=30,
+        return_messages=True
     )
 
+
     llm_chain = LLMChain(
         llm=llm,
         prompt=daysoff_assistant_prompt,
@@ -118,9 +123,19 @@ async def handle_message(message: cl.Message):
         except requests.exceptions.RequestException as e:
             await cl.Message(content=f"Request failed: {str(e)}").send()
     else:
-        response = await llm_chain.ainvoke(user_message)
+        try:
+            response = await llm_chain.ainvoke({
+                "question": user_message,
+                "chat_history": ""
+            }, callbacks=[cl.AsyncLangchainCallbackHandler()])
+            
+            await cl.Message(content=response["text"]).send()
+            
+        except Exception as e:
+            await cl.Message(content=f"Error: {str(e)}").send()
+
 
 
-    response_key = "output" if "output" in response else "text"
-    await cl.Message(response.get(response_key, "")).send()
-    return message.content
+    #response_key = "output" if "output" in response else "text"
+    #await cl.Message(response.get(response_key, "")).send()
+    #return message.content