Update app.py

app.py

@@ -57,7 +57,12 @@ def stream_graph_updates(user_input: str):
     assistant_reply = ""
     for event in graph.stream({"messages": [("user", user_input)]}):
         for value in event.values():
-            assistant_reply = value["messages"][-1].content
+            if isinstance(value["messages"][-1], dict):
+                # If it's a dict, extract 'content'
+                assistant_reply = value["messages"][-1].get("content", "")
+            elif isinstance(value["messages"][-1], str):
+                # If it's a string, use it directly
+                assistant_reply = value["messages"][-1]
     return assistant_reply
 
 # Gradio chatbot function using the streaming updates
@@ -81,4 +86,4 @@ interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch(share=True)