Update app.py

app.py

@@ -20,7 +20,7 @@ from smolagents.agent_types import (
     AgentText,
     handle_agent_output_types,
 )
-from smolagents.gradio_ui import pull_messages_from_step
+from smolagents.gradio_ui import stream_to_gradio
 
 from scripts.text_inspector_tool import TextInspectorTool
 from scripts.text_web_browser import (
@@ -160,9 +160,6 @@ AUTHORIZED_IMPORTS = [
 load_dotenv(override=True)
 login(os.getenv("HF_TOKEN"))
 
-
-print("TOKKKK", os.getenv("HF_TOKEN")[-10:])
-
 append_answer_lock = threading.Lock()
 
 custom_role_conversions = {"tool-call": "assistant", "tool-response": "user"}
@@ -218,44 +215,6 @@ def create_agent():
 document_inspection_tool = TextInspectorTool(model, 20000)
 
 
-def stream_to_gradio(
-    agent,
-    task: str,
-    reset_agent_memory: bool = False,
-    additional_args: Optional[dict] = None,
-):
-    """Runs an agent with the given task and streams the messages from the agent as gradio ChatMessages."""
-    for step_log in agent.run(
-        task, stream=True, reset=reset_agent_memory, additional_args=additional_args
-    ):
-        for message in pull_messages_from_step(
-            step_log,
-        ):
-            yield message
-
-    final_answer = handle_agent_output_types(step_log.content)  # Last log is the run's final_answer
-
-    if isinstance(final_answer, AgentText):
-        yield gr.ChatMessage(
-            role="assistant",
-            content=f"**Final answer:**\n{final_answer.to_string()}\n",
-        )
-    elif isinstance(final_answer, AgentImage):
-        yield gr.ChatMessage(
-            role="assistant",
-            content={"path": final_answer.to_string(), "mime_type": "image/png"},
-        )
-    elif isinstance(final_answer, AgentAudio):
-        yield gr.ChatMessage(
-            role="assistant",
-            content={"path": final_answer.to_string(), "mime_type": "audio/wav"},
-        )
-    else:
-        yield gr.ChatMessage(
-            role="assistant", content=f"**Final answer:** {str(final_answer)}"
-        )
-
-
 class GradioUI:
     """A one-line interface to launch your agent in Gradio"""