Update app.py

app.py

@@ -80,46 +80,47 @@ def text_to_speech(text: str) -> AgentAudio:
     return AgentAudio(AUDIO_OUTPUT_PATH)
 
 
-# # Define the audio output path
-# AUDIO_OUTPUT_PATH = "/tmp/response.mp3"
-
-# def speak_text(text):
-#     """Convert text to speech using gTTS and save as an MP3 file."""
-#     tts = gTTS(text=text, lang='en')
-#     tts.save(AUDIO_OUTPUT_PATH)
-#     return AUDIO_OUTPUT_PATH  # Return the file path for Gradio Audio component
-
-
-
-# @tool
-# def search_dad_jokes(term: str) -> str:
-#     """A tool that searches for dad jokes containing a specific term.
-#     Args:
-#         term: The keyword to search for in dad jokes.
-#     """
-#     try:
-#         headers = {
-#             "Accept": "application/json",
-#             "User-Agent": "YourAppName (https://yourappurl.com)"
-#         }
-#         response = requests.get(f"https://icanhazdadjoke.com/search?term={term}", headers=headers)
-#         data = response.json()
-#         if data['results']:
-#             jokes = [joke['joke'] for joke in data['results']]
-#             response_text = f"Found {len(jokes)} jokes:\n" + "\n\n".join(jokes)
-#         else:
-#             response_text = f"No jokes found for the term '{term}'."
-#     except Exception as e:
-#         response_text = f"Error searching for jokes: {str(e)}"
-
-#     # Generate audio using gTTS
-#     audio_file = speak_text(response_text)
-
-#     return response_text, audio_file  # Return text and audio file path
+# Define the audio output path
+AUDIO_OUTPUT_PATH = "/tmp/response.mp3"
+
+def speak_text(text):
+    """Convert text to speech using gTTS and save as an MP3 file."""
+    tts = gTTS(text=text, lang='en')
+    tts.save(AUDIO_OUTPUT_PATH)
+    return AUDIO_OUTPUT_PATH  # Return the file path for Gradio Audio component
+
+
+
+@tool
+def search_dad_jokes(term: str) -> str:
+    """A tool that searches for dad jokes containing a specific term.
+    Args:
+        term: The keyword to search for in dad jokes.
+    """
+    try:
+        headers = {
+            "Accept": "application/json",
+            "User-Agent": "YourAppName (https://yourappurl.com)"
+        }
+        response = requests.get(f"https://icanhazdadjoke.com/search?term={term}", headers=headers)
+        data = response.json()
+        if data['results']:
+            jokes = [joke['joke'] for joke in data['results']]
+            response_text = f"Found {len(jokes)} jokes:\n" + "\n\n".join(jokes)
+        else:
+            response_text = f"No jokes found for the term '{term}'."
+    except Exception as e:
+        response_text = f"Error searching for jokes: {str(e)}"
+
+    # Generate audio using gTTS
+    audio_file = speak_text(response_text)
+
+    return response_text, audio_file  # Return text and audio file path
 
 
 lookup_definition_tool = lookup_definition
 text_to_speech_tool = text_to_speech
+dad_jokes_tool = search_dad_jokes
 
 final_answer = FinalAnswerTool()
 web_search_tool = DuckDuckGoSearchTool()
@@ -146,7 +147,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, text_to_speech_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, dad_jokes_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -156,29 +157,29 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
     )
 
-# #Gradio interface with text and audio output
-# def gradio_search_jokes(word):
-#     """Wrapper function for Gradio to call search_dad_jokes and generate audio."""
-#     response_text, audio_file = search_dad_jokes(word)  # Ensure search_dad_jokes returns (text, file path)
-#     return response_text, audio_file
+#Gradio interface with text and audio output
+def gradio_search_jokes(word):
+    """Wrapper function for Gradio to call search_dad_jokes and generate audio."""
+    response_text, audio_file = search_dad_jokes(word)  # Ensure search_dad_jokes returns (text, file path)
+    return response_text, audio_file
 
 
 
-# #Define the Gradio UI
-# with gr.Blocks() as demo:
-#    gr.Markdown("### Dad Jokes Finder with AI & Text-to-Speech 🎙️")
+#Define the Gradio UI
+with gr.Blocks() as demo:
+   gr.Markdown("### Dad Jokes Finder with AI & Text-to-Speech 🎙️")
    
-#    with gr.Row():
-#        input_box = gr.Textbox(label="Enter a word")
-#        output_text = gr.Textbox(label="Jokes Found")
-#        output_audio = gr.Audio(label="Audio Pronunciation", type="filepath")
+   with gr.Row():
+       input_box = gr.Textbox(label="Enter a word")
+       output_text = gr.Textbox(label="Jokes Found")
+       output_audio = gr.Audio(label="Audio Pronunciation", type="filepath")
    
-#    btn = gr.Button("Get Jokes")
-#    btn.click(gradio_search_jokes, inputs=input_box, outputs=[output_text, output_audio])
+   btn = gr.Button("Get Jokes")
+   btn.click(gradio_search_jokes, inputs=input_box, outputs=[output_text, output_audio])
 
-# demo.launch()
+demo.launch()
 
 
-GradioUI(agent).launch()
+# GradioUI(agent).launch()