uasername commited on
Commit
f08fb67
·
verified ·
1 Parent(s): 35cb1a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -1
app.py CHANGED
@@ -16,6 +16,7 @@ from Code_Functions import speak_text
16
 
17
 
18
  from smolagents.agent_types import AgentText
 
19
 
20
  @tool
21
  def lookup_definition(query: str) -> AgentText:
@@ -54,6 +55,25 @@ def lookup_definition(query: str) -> AgentText:
54
 
55
 
56
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
57
  # # Define the audio output path
58
  # AUDIO_OUTPUT_PATH = "/tmp/response.mp3"
59
 
@@ -93,6 +113,8 @@ def lookup_definition(query: str) -> AgentText:
93
 
94
 
95
  lookup_definition_tool = lookup_definition
 
 
96
  final_answer = FinalAnswerTool()
97
  web_search_tool = DuckDuckGoSearchTool()
98
  visit_webpage_tool = VisitWebpageTool()
@@ -118,7 +140,7 @@ with open("prompts.yaml", 'r') as stream:
118
 
119
  agent = CodeAgent(
120
  model=model,
121
- tools=[visit_webpage_tool, web_search_tool, final_answer, image_generation_tool, lookup_definition_tool], ## add your tools here (don't remove final answer)
122
  max_steps=6,
123
  verbosity_level=1,
124
  grammar=None,
 
16
 
17
 
18
  from smolagents.agent_types import AgentText
19
+ from smolagents.agent_types import AgentAudio
20
 
21
  @tool
22
  def lookup_definition(query: str) -> AgentText:
 
55
 
56
 
57
 
58
+
59
+ AUDIO_OUTPUT_PATH = "/tmp/response.mp3"
60
+
61
+ @tool
62
+ def text_to_speech(text: str) -> AgentAudio:
63
+ """Converts input text to speech and returns an AgentAudio instance with the audio file path.
64
+
65
+ Args:
66
+ query: The text to convert to speech.
67
+
68
+ Returns:
69
+ An audio response.
70
+
71
+ """
72
+ tts = gTTS(text=text, lang='en')
73
+ tts.save(AUDIO_OUTPUT_PATH)
74
+ return AgentAudio(AUDIO_OUTPUT_PATH)
75
+
76
+
77
  # # Define the audio output path
78
  # AUDIO_OUTPUT_PATH = "/tmp/response.mp3"
79
 
 
113
 
114
 
115
  lookup_definition_tool = lookup_definition
116
+ text_to_speech_tool = text_to_speech
117
+
118
  final_answer = FinalAnswerTool()
119
  web_search_tool = DuckDuckGoSearchTool()
120
  visit_webpage_tool = VisitWebpageTool()
 
140
 
141
  agent = CodeAgent(
142
  model=model,
143
+ 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)
144
  max_steps=6,
145
  verbosity_level=1,
146
  grammar=None,