Update app.py

app.py

@@ -2,8 +2,16 @@ import gradio as gr
 from llm_handler import sanchaari
 
 def handle_query(query):
-    thought_process, response = sanchaari(query)
-    return {"Thought Process": thought_process, "Response": response}
+    thoughts = []
+    final_answer = None
+    
+    for result in sanchaari(query):
+        if "thought" in result:
+            thoughts.append(result["thought"])
+            yield {"Thought Process": thoughts, "Response Text": ""}
+        if "final_answer" in result:
+            final_answer = result["final_answer"]
+            yield {"Thought Process": thoughts, "Response Text": final_answer}
 
 gr.Interface(
     fn=handle_query,