Update app.py

app.py

@@ -22,27 +22,26 @@ else:
 # ✅ Sound generation tool with structured docstring for smolagents
 @tool
 def generate_sound(sound_type: str, duration: int) -> str:
-    """
-    Generates a simple sound based on the specified type and duration.
-
-    Parameters:
-        sound_type (str): The type of sound to generate. Valid values are 'rain', 'white_noise'.
-        duration (int): The duration of the sound in seconds (integer).
+    """Generates a simple sound based on the specified type and duration.
+    
+    Args:
+        sound_type: Type of sound to generate (e.g., 'rain', 'white_noise').
+        duration: Duration of the sound in seconds.
 
     Returns:
-        str: Path to the generated sound file (e.g., '/home/user/app/rain_3s.wav').
+        Path to the generated audio file.
     """
     try:
-        duration_ms = duration * 1000
-        output_path = f"/home/user/app/{sound_type}_{duration}s.wav"
-
+        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"Unsupported sound type: {sound_type}"
 
+        output_path = f"/tmp/{sound_type}_{duration}s.wav"
         sound.export(output_path, format="wav")
-        return output_path  # ✅ Returns a valid file path for Gradio
+        return output_path
 
     except Exception as e:
         return f"Error generating sound: {str(e)}"