Update app.py

app.py

@@ -4,12 +4,16 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from transformers import pipeline
 
 from Gradio_UI import GradioUI
 
 # Web search tool to find trending music
 search_tool = DuckDuckGoSearchTool()
 
+# Load the MusicGen model
+pipe = pipeline("text-to-audio", model="facebook/musicgen-large")
+
 # Custom tool for AI-assisted music generation
 @tool
 def generate_music(theme: str) -> str:
@@ -17,13 +21,19 @@ def generate_music(theme: str) -> str:
     Args:
         theme: The theme or mood of the music (e.g., 'moody jazz on a rainy night').
     """
-    response = requests.post(
-        "https://huggingface.co/facebook/musicgen-large", 
-        json={"prompt": theme}
-    )
-    if response.status_code == 200:
-        return f"Here’s your AI-generated music for '{theme}': {response.json().get('music_url', 'Music not available')}"
-    return "Failed to generate music. Try another theme!"
+    try:
+        # Generate music using the pipeline
+        output = pipe(theme)
+        
+        # Extract the generated audio file
+        audio_url = output[0]["audio"] if "audio" in output[0] else None
+
+        if audio_url:
+            return f"Here’s your AI-generated music for '{theme}': {audio_url}"
+        else:
+            return "Music generation failed. Try a different theme!"
+    except Exception as e:
+        return f"Error generating music: {str(e)}"
     
 # Tool to generate an album cover
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)