Update Gradio_UI.py

Gradio_UI.py

@@ -23,6 +23,7 @@ from smolagents.agent_types import AgentAudio, AgentImage, AgentText, handle_age
 from smolagents.agents import ActionStep, MultiStepAgent
 from smolagents.memory import MemoryStep
 from smolagents.utils import _is_package_available
+from smolagents.agents import FinalAnswerStep
 
 
 def pull_messages_from_step(
@@ -154,6 +155,10 @@ def stream_to_gradio(
             yield message
 
     final_answer = step_log  # Last log is the run's final_answer
+
+    if isinstance(final_answer, FinalAnswerStep):
+        final_answer = final_answer.final_answer
+         
     final_answer = handle_agent_output_types(final_answer)
 
     if isinstance(final_answer, AgentText):