Update app.py

app.py

@@ -44,7 +44,7 @@ from smolagents import (
     ToolCallingAgent,
 )
 from smolagents.agent_types import AgentText, AgentImage, AgentAudio
-from smolagents.gradio_ui import pull_messages_from_step, handle_agent_output_types
+from smolagents.gradio_ui import pull_messages_from_step #agent type thing might not exist...
 
 AUTHORIZED_IMPORTS = [
     "requests",
@@ -71,6 +71,7 @@ AUTHORIZED_IMPORTS = [
     "datetime",
     "fractions",
     "csv",
+    "re", #allow regex
 ]
 load_dotenv(override=True)
 login(os.getenv("HF_TOKEN"))
@@ -167,7 +168,6 @@ def stream_to_gradio(
             yield message
 
     final_answer = step_log  # Last log is the run's final_answer
-    final_answer = handle_agent_output_types(final_answer)
 
     if isinstance(final_answer, AgentText):
         yield gr.ChatMessage(