Update Gradio_UI.py

Gradio_UI.py

@@ -141,11 +141,14 @@ def stream_to_gradio(
             # Extract the actual content from the final answer
             content = str(final)
             # Remove the wrapper if it exists (e.g., "FinalAnswerStep(final_answer='...')")
-            if "final_answer=" in content:
+            if "final_answer=" in content or "FinalAnswerStep" in content:
                 import re
-                match = re.search(r"final_answer=['\"](.+)['\"]", content, re.DOTALL)
+                match = re.search(r"final_answer=['\"](.+?)['\"](?:\)|$)", content, re.DOTALL)
                 if match:
                     content = match.group(1)
+                elif "final_answer='" in content:
+                    # Alternative extraction if regex fails
+                    content = content.split("final_answer='", 1)[1].rsplit("')", 1)[0]
                     # Unescape newlines and other escape sequences
                     content = content.encode().decode('unicode_escape')