Update app.py

app.py

@@ -16,6 +16,7 @@ from Code_Functions import speak_text
 
 
 from smolagents.agent_types import AgentText
+from smolagents.agent_types import AgentAudio
 
 @tool
 def lookup_definition(query: str) -> AgentText:
@@ -54,6 +55,25 @@ def lookup_definition(query: str) -> AgentText:
 
 
 
+
+AUDIO_OUTPUT_PATH = "/tmp/response.mp3"
+
+@tool
+def text_to_speech(text: str) -> AgentAudio:
+    """Converts input text to speech and returns an AgentAudio instance with the audio file path.
+
+    Args:
+        query: The text to convert to speech.
+    
+    Returns:
+        An audio response.
+    
+    """
+    tts = gTTS(text=text, lang='en')
+    tts.save(AUDIO_OUTPUT_PATH)
+    return AgentAudio(AUDIO_OUTPUT_PATH)
+
+
 # # Define the audio output path
 # AUDIO_OUTPUT_PATH = "/tmp/response.mp3"
 
@@ -93,6 +113,8 @@ def lookup_definition(query: str) -> AgentText:
 
 
 lookup_definition_tool = lookup_definition
+text_to_speech_tool = text_to_speech
+
 final_answer = FinalAnswerTool()
 web_search_tool = DuckDuckGoSearchTool()
 visit_webpage_tool = VisitWebpageTool()
@@ -118,7 +140,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[visit_webpage_tool, web_search_tool, final_answer, image_generation_tool, lookup_definition_tool], ## add your tools here (don't remove final answer)
+    tools=[visit_webpage_tool, web_search_tool, final_answer, image_generation_tool, lookup_definition_tool, text_to_speech_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,