Update app.py

app.py

@@ -12,7 +12,10 @@ from Gradio_UI import GradioUI
 search_tool = DuckDuckGoSearchTool()
 
 # Load the MusicGen model
-pipe = pipeline("text-to-audio", model="facebook/musicgen-small")
+try:
+    pipe = pipeline("text-to-audio", model="facebook/musicgen-small")
+except Exception as e:
+    print(f"Error loading MusicGen model: {str(e)}")
 
 # Custom tool for AI-assisted music generation
 @tool
@@ -22,16 +25,23 @@ def generate_music(theme: str) -> str:
         theme: The theme or mood of the music (e.g., 'moody jazz on a rainy night').
     """
     try:
+        if not theme:
+            return "Error: Theme cannot be empty!"
+
         # Generate music using the pipeline
-        output = pipe(theme)
+        output = pipe(theme, guidance_scale=1.0)  # Set guidance_scale=1.0 to prevent errors
         
+        if not output or not isinstance(output, list) or len(output) == 0:
+            return "Music generation failed: No output received from the model."
+
         # Extract the generated audio file
-        audio_url = output[0]["audio"] if "audio" in output[0] else None
+        audio_url = output[0].get("audio", None)  # Safer way to access audio data
 
         if audio_url:
-            return f"Here’s your AI-generated music for '{theme}': {audio_url}"
+            return f"Here’s your AI-generated music for '{theme}': [Click to listen]({audio_url})"
         else:
-            return "Music generation failed. Try a different theme!"
+            return "Music generation failed: No audio URL found."
+    
     except Exception as e:
         return f"Error generating music: {str(e)}"
     
@@ -39,8 +49,12 @@ def generate_music(theme: str) -> str:
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 # Load prompts
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
+try:
+    with open("prompts.yaml", 'r') as stream:
+        prompt_templates = yaml.safe_load(stream)
+except FileNotFoundError:
+    print("Warning: prompts.yaml not found. Using default prompts.")
+    prompt_templates = {}
 
 # Define model
 model = HfApiModel(