Update app.py

app.py

@@ -87,14 +87,22 @@ if __name__ == "__main__":
         return agent.update_parameters(seed=seed)
 
     # ✅ FIXED: retry must return, not yield
-    def handle_retry(history, retry_data, temperature, max_new_tokens, max_tokens, multi_agent, conversation, max_round):
-        update_seed()
-        new_history = history[:retry_data.index]
-        previous_prompt = history[retry_data.index]["content"]
-        return agent.run_gradio_chat(
-            new_history + [{"role": "user", "content": previous_prompt}],
-            temperature, max_new_tokens, max_tokens, multi_agent, conversation, max_round
-        )
+def handle_retry(history, retry_data, temperature, max_new_tokens, max_tokens, multi_agent, conversation, max_round):
+    update_seed()
+    new_history = history[:retry_data.index]
+    previous_prompt = history[retry_data.index]["content"]
+
+    # ✅ This MUST return, not yield
+    result = agent.run_gradio_chat(
+        new_history + [{"role": "user", "content": previous_prompt}],
+        temperature, max_new_tokens, max_tokens, multi_agent, conversation, max_round
+    )
+
+    # If your agent returns a generator, consume it into a list or string
+    if hasattr(result, "__iter__") and not isinstance(result, (str, dict, list)):
+        result = list(result)
+
+    return result
 
     # ===== Build Gradio Interface =====
     with gr.Blocks(css=css) as demo: