first

app.py

@@ -6,18 +6,6 @@ import yaml
 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 !