Update app.py

app.py

@@ -82,6 +82,11 @@ supervisor = create_supervisor(
     output_mode="full_history"
 ).compile()
 
+#final answer
+def stream_graph_updates(user_input: str):
+    for event in graph.stream({"messages": [{"role": "user", "content": user_input}]}):
+        for value in event.values():
+            print("Assistant:", value["messages"][-1].content)
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -108,7 +113,19 @@ class BasicAgent:
                         ]
                     }
 
-        answer = supervisor.invoke(messages)
+        events = supervisor.stream(
+            messages,
+            stream_mode="values",
+        )
+        
+        for event in events:
+            messages.extend(event["messages"])
+
+        answer = messages[-1].content
+
+        #print(f"Answer: {answer}...")
+        
+        #answer = supervisor.invoke(messages)
         print(f"Agent returning answer: {answer}")
         return answer