Update app.py

app.py

@@ -6,6 +6,18 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
+_orig_handle = Gradio_UI.handle_agent_output_types
+
+def _patched_handle(step):
+    # If step is a FinalAnswerStep, extract the inner .final_answer
+    if step.__class__.__name__ == 'FinalAnswerStep':
+        return step.final_answer
+    # otherwise, call the original
+    return _orig_handle(step)
+
+Gradio_UI.handle_agent_output_types = _patched_handle
+
+
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool