Update app.py

app.py

@@ -1,12 +1,23 @@
 import gradio as gr
 from langchain_core.messages import HumanMessage
-from agent import alfred
+from agent import assistant, AgentState
 
 def respond(message, history=None):
-    messages = [HumanMessage(content=message)]
-    result = alfred.invoke({"messages": messages})
-    response = result["messages"][0] if isinstance(result["messages"], list) else result["messages"]
-    return response
+    history = history or []
+    # Append new user message
+    messages = history + [HumanMessage(content=message)]
+    
+    state = {"messages": messages}
+    result = assistant(state)
+    
+    # Get the response text
+    response_msg = result["messages"][-1].content
+    
+    # Append bot response to history
+    history.append(HumanMessage(content=message))
+    history.append(HumanMessage(content=response_msg))
+    
+    return response_msg, history
 
 demo = gr.ChatInterface(respond)