Update app.py

app.py

@@ -62,16 +62,17 @@ graph = workflow.compile()
 # Define the function to run the graph
 def run_langgraph(user_input):
     # Start with user input
-    events = graph.stream(
+     events = graph.stream(
         {"messages": [("user", user_input)]},
         {"recursion_limit": 150}
     )
-    # Collect outputs from the workflow
-    output = []
+    
+     final_message = None
     for event in events:
-        output.append(event)
-    # Return the final output
-    return output[-1]["messages"][-1].content if output else "No output generated"
+        if "messages" in event and event["messages"]:
+            final_message = event["messages"][-1].content
+
+    return final_message or "No output generated"
 
 # Create Gradio interface
 def process_input(user_input):