Update app.py

app.py

@@ -173,18 +173,16 @@ def create_ui(agent: TxAgent):
                     if isinstance(update, str):
                         final_response.append(update)
                     elif isinstance(update, list):
-                        for msg in update:
-                            if hasattr(msg, 'content'):
-                                final_response.append(msg.content)
+                        final_response.extend(msg.content for msg in update if hasattr(msg, 'content'))
 
                     if len(final_response) % 3 == 0:
-                        content = "".join(final_response).strip().replace("[TOOL_CALLS]", "")
-                        history[-1] = {"role": "assistant", "content": content or "❌ No response."}
+                        current_output = "".join(final_response).strip().replace("[TOOL_CALLS]", "")
+                        history[-1] = {"role": "assistant", "content": current_output or "⏳ Generating..."}
                         yield history
 
-                final_cleaned = "".join(final_response).strip().replace("[TOOL_CALLS]", "")
-                history[-1] = {"role": "assistant", "content": final_cleaned or "❌ No response."}
-                print("Final model response:\n", final_cleaned)
+                full_output = "".join(final_response).strip().replace("[TOOL_CALLS]", "")
+                print("Final model response:\n", full_output)
+                history[-1] = {"role": "assistant", "content": full_output or "❌ No response."}
                 print(f"Model processing took: {time.time() - model_start:.2f}s")
                 yield history