Update app.py

app.py

@@ -349,35 +349,39 @@ async def multi_agent_conversation(task_message: str, log_queue: queue.Queue, ap
     log_queue.put(("result", context.conversation_history))
 
 # -------------------- Process Generator and Human Input --------------------
-
 def process_conversation_generator(task_message: str, api_key: str,
                                    human_event: threading.Event, human_input_queue: queue.Queue,
                                    log_queue: queue.Queue) -> Generator[str, None, None]:
     """
-    Runs the conversation and yields log messages.
+    Runs the multi-agent conversation in a background thread and yields log messages
+    to update the chat box without blocking the UI.
     """
-    # Run the conversation asynchronously.
-    asyncio.run(multi_agent_conversation(task_message, log_queue, api_key, human_event, human_input_queue))
-
-    final_result = None
-    while True:
+    def run_conversation():
+        asyncio.run(multi_agent_conversation(task_message, log_queue, api_key, human_event, human_input_queue))
+    
+    # Start the asynchronous conversation in a separate thread.
+    conversation_thread = threading.Thread(target=run_conversation)
+    conversation_thread.start()
+    
+    # Continuously yield log messages until the conversation is complete.
+    while conversation_thread.is_alive() or not log_queue.empty():
         try:
-            msg = log_queue.get_nowait()
+            msg = log_queue.get(timeout=0.1)
             if isinstance(msg, tuple) and msg[0] == "result":
-                final_result = msg[1]
-                yield gr.Chatbot.update(value=final_result, visible=True)
-                yield "Conversation complete."
-                break
+                # Update the chat box with the final conversation history.
+                yield gr.Chatbot.update(value=msg[1], visible=True)
             else:
                 yield msg
         except queue.Empty:
             pass
-
-        # If human feedback is requested, yield an appropriate message.
+        
         if human_event.is_set():
             yield "Waiting for human feedback..."
-        # Use a short asynchronous sleep to avoid busy-waiting.
-        asyncio.run(asyncio.sleep(0.1))
+        
+        # Small sleep to prevent busy-waiting.
+        time.sleep(0.1)
+    
+    yield "Conversation complete."
 
 def get_human_feedback(placeholder_text: str, human_input_queue: queue.Queue) -> gr.Blocks:
     """