Upload app.py

app.py

@@ -1,22 +1,20 @@
 import gradio as gr
-import asyncio
 from agents import build_graph
 
-# Build + compile once at import time
 graph = build_graph()
 
 async def run_book_recommender(user_input):
-    # 1) build initial state
     initial_state = {"user_input": user_input}
 
-    # 2) run the graph in one shot
-    final_state = await graph.run(initial_state)
+    final_state = None
+    # .astream yields after each node runs
+    async for state in graph.astream(initial_state):
+        final_state = state
 
-    # 3) extract node-3 outputs from the top-level state
+    # now state after the last node is in final_state
     recs = final_state.get("final_recommendations", [])
     reasoning = final_state.get("final_reasoning", "")
 
-    # 4) format for display
     recs_text = "\n\n".join(
         f"📘 {r['title']}\n🔗 {r.get('link','')}\n💡 {r.get('reason','')}"
         for r in recs