Update app.py

app.py

@@ -7,7 +7,9 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-import kokoro
+from kokoro import KPipeline
+import soundfile as sf
+import os
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -20,25 +22,25 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+
+# Initialize the Kokoro pipeline
+pipeline = KPipeline(lang_code='a')  # 'a' stands for American English
+
 @tool
-def text_to_speech_kokoro(text: str, voice: str = 'af_heart', lang_code: str = 'a') -> str:
-    """A tool that converts text to speech using the Kokoro-82M model.
+def text_to_speech_kokoro(text: str, voice: str = 'af_heart', speed: float = 1.0) -> str:
+    """Convert text to speech using the Kokoro-82M model.
+    
     Args:
         text: The text to be converted to speech.
         voice: The voice to use for speech synthesis (default is 'af_heart').
-        lang_code: The language code corresponding to the voice (default is 'a' for American English).
+        speed: The speed of the speech (default is 1.0).
+    
+    Returns:
+        The filename of the generated audio file.
     """
     try:
-        # Import necessary libraries
-        from kokoro import KPipeline
-        import soundfile as sf
-        import os
-
-        # Initialize the Kokoro pipeline
-        pipeline = KPipeline(lang_code=lang_code)
-
         # Generate speech audio
-        generator = pipeline(text, voice=voice, speed=1, split_pattern=r'\n+')
+        generator = pipeline(text, voice=voice, speed=speed, split_pattern=r'\n+')
         audio_files = []
 
         # Save each audio segment to a file
@@ -52,7 +54,6 @@ def text_to_speech_kokoro(text: str, voice: str = 'af_heart', lang_code: str = '
         return f"Error generating speech: {str(e)}"
 
 
-
 @tool
 def search_dad_jokes(term: str) -> str:
     """A tool that searches for dad jokes containing a specific term.