Update app.py

app.py

@@ -166,23 +166,23 @@ def create_ui(agent: TxAgent):
                     max_round=10
                 )
 
-                final_response = []
+                final_response = ""
                 for update in generator:
                     if not update:
                         continue
-                    if isinstance(update, str):
-                        final_response.append(update)
-                    elif isinstance(update, list):
-                        final_response.extend(msg.content for msg in update if hasattr(msg, 'content'))
-
-                    if len(final_response) % 3 == 0:
-                        current_output = "".join(final_response).strip().replace("[TOOL_CALLS]", "")
-                        history[-1] = {"role": "assistant", "content": current_output or "⏳ Generating..."}
-                        yield history
-
-                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."}
+                    if isinstance(update, list):
+                        for msg in update:
+                            if hasattr(msg, 'content'):
+                                final_response += msg.content
+                    elif isinstance(update, str):
+                        final_response += update
+
+                    cleaned = final_response.strip().replace("[TOOL_CALLS]", "")
+                    history[-1] = {"role": "assistant", "content": cleaned or "❌ No response."}
+                    yield history
+
+                print("Final model response:\n", final_response)
+                history[-1] = {"role": "assistant", "content": final_response.strip() or "❌ No response."}
                 print(f"Model processing took: {time.time() - model_start:.2f}s")
                 yield history
 
@@ -232,4 +232,4 @@ if __name__ == "__main__":
         server_port=7860,
         show_error=True,
         share=True
-    )
+    )