Update Gradio_UI.py

Gradio_UI.py

@@ -137,17 +137,23 @@ def stream_to_gradio(
         final = handle_agent_output_types(step_log)
         if isinstance(final, AgentText):
             content = final.to_string()
-            yield gr.ChatMessage(
-                role="assistant",
-                content=content,
-                metadata={"react": True, "status": "done"}
-            )
         else:
-            yield gr.ChatMessage(
-                role="assistant",
-                content=f"**Final Answer:** {str(final)}",
-                metadata={"status": "done"}
-            )
+            # 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:
+                import re
+                match = re.search(r"final_answer=['\"](.+)['\"]", content, re.DOTALL)
+                if match:
+                    content = match.group(1)
+                    # Unescape newlines and other escape sequences
+                    content = content.encode().decode('unicode_escape')
+        
+        yield gr.ChatMessage(
+            role="assistant",
+            content=content,
+            metadata={"status": "done"}
+        )
     
     except Exception as e:
         # Handle errors gracefully