added download functionality

app.py

@@ -8,6 +8,7 @@ from tools.final_answer import FinalAnswerTool
 from huggingface_hub import InferenceClient
 from pydub.generators import WhiteNoise
 from pydub import AudioSegment
+from gradio import Files
 
 from Gradio_UI import GradioUI
 import os
@@ -29,18 +30,19 @@ def generate_sound(sound_type: str, duration: int) -> str:
         duration: Duration of the sound in seconds.
 
     Returns:
-        Path to the generated audio file.
+        Download link
     """
     try:
-        duration_ms = duration * 1000  # Convert to milliseconds
+        duration_ms = duration * 1000
         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"
+        output_path = f"rain_{duration}s.wav"
         sound.export(output_path, format="wav")
-        return output_path
+
+        return Files(output_path)  # This makes the file downloadable in Gradio
 
     except Exception as e:
         return f"Error generating sound: {str(e)}"
@@ -85,6 +87,7 @@ agent = CodeAgent(
     name=None,
     description=None,
     prompt_templates=prompt_templates
+    output_modality="file"
 )
 
 # Start the UI with processing time display