fixed tool

app.py

@@ -6,6 +6,8 @@ import yaml
 import time
 from tools.final_answer import FinalAnswerTool
 from huggingface_hub import InferenceClient
+from pydub.generators import WhiteNoise
+from pydub import AudioSegment
 
 from Gradio_UI import GradioUI
 import os
@@ -20,18 +22,26 @@ else:
 # Sound generation tool
 @tool
 def generate_sound(sound_type: str, duration: int) -> str:
-    """A tool that generates a sound based on the given type and duration.
+    """Generates a simple sound based on the specified type and duration.
+    
     Args:
-        sound_type: A string specifying the type of sound (e.g., 'rain', 'ocean waves', 'fire crackling').
-        duration: An integer representing the duration of the sound in seconds.
+        sound_type: Type of sound to generate (e.g., 'rain', 'white_noise').
+        duration: Duration of the sound in seconds.
+
+    Returns:
+        Path to the generated audio file.
     """
     try:
-        # Simulate API call to generate sound
-        response = requests.get(f"https://api.example.com/generate_sound?sound={sound_type}&duration={duration}")
-        if response.status_code == 200:
-            return f"Generated {sound_type} sound for {duration} seconds. Link: {response.json().get('sound_url', 'N/A')}"
+        duration_ms = duration * 1000  # Convert to milliseconds
+        if sound_type == "rain":
+            sound = WhiteNoise().to_audio_segment(duration=duration_ms).low_pass_filter(5000)
         else:
-            return f"Failed to generate sound: {response.text}"
+            return f"Unsupported sound type: {sound_type}"
+
+        output_path = f"/tmp/{sound_type}_{duration}s.wav"
+        sound.export(output_path, format="wav")
+        return output_path
+
     except Exception as e:
         return f"Error generating sound: {str(e)}"
 
@@ -67,7 +77,7 @@ with open("prompts.yaml", 'r') as stream:
 
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, generate_sound],  # Added generate_sound tool
+    tools=[final_answer, generate_sound],  # Add your tool
     max_steps=6,
     verbosity_level=1,
     grammar=None,