Custom debug

Gradio_UI.py

@@ -25,6 +25,9 @@ from smolagents.agents import ActionStep, MultiStepAgent
 from smolagents.memory import MemoryStep
 from smolagents.utils import _is_package_available
 
+def debug_print(x):
+    print("DEBUG_FROM_GRADIO:", x)
+    return x
 
 def pull_messages_from_step(
     step_log: MemoryStep,
@@ -185,9 +188,9 @@ def stream_to_gradio(
         # mais pour l’instant on laisse image_path = None.
 
     elif isinstance(final_answer, AgentImage):
-        print("DEBUG AgentImage type:", type(final_answer))
-        print("DEBUG AgentImage to_string:", final_answer.to_string())
-        print("DEBUG AgentImage to_raw:", type(final_answer.to_raw()))
+        debug_print("DEBUG AgentImage type:", type(final_answer))
+        debug_print("DEBUG AgentImage to_string:", final_answer.to_string())
+        debug_print("DEBUG AgentImage to_raw:", type(final_answer.to_raw()))
         # On convertit l'AgentImage en fichier sur disque
         image_path = save_agent_image_to_tmp(final_answer)