Fix Gradio error

app.py

@@ -7,6 +7,18 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+import Gradio_UI
+_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
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type