Added error handling

app.py

@@ -2,41 +2,46 @@ import gradio as gr
 import replicate
 
 def generate_music(genre, instruments, tempo, mood, duration, key_scale):
-    # Creating a more detailed prompt based on the additional options
     prompt = f"Generate a {genre} track featuring {instruments}, in a {mood} mood, with a consistent melody throughout, with a tempo of {tempo} BPM, in the key of {key_scale}."
-    # Assuming replicate.run() is a placeholder for actual music generation API call
-    # Please replace with the actual function call to generate music
-    output = replicate.run(
-        "bamartin1618/weather-channel:77dae70c23138abd20014ba686f62da780c5aed738e00085ce698eb5e344c65a",
-        input={
-            "top_k": 250,
-            "top_p": 1,
-            "prompt": prompt,
-            "duration": duration,
-            "temperature": 1,
-            "continuation": False,
-            "output_format": "wav",
-            "continuation_start": 0,
-            "multi_band_diffusion": False,
-            "normalization_strategy": "loudness",
-            "classifier_free_guidance": 6
-        }
-    )
-    
+    try:
+        # Assuming replicate.run() is a placeholder for actual music generation API call
+        output = replicate.run(
+            "bamartin1618/weather-channel:77dae70c23138abd20014ba686f62da780c5aed738e00085ce698eb5e344c65a",
+            input={
+                "top_k": 250,
+                "top_p": 1,
+                "prompt": prompt,
+                "duration": duration,
+                "temperature": 1,
+                "continuation": False,
+                "output_format": "wav",
+                "continuation_start": 0,
+                "multi_band_diffusion": False,
+                "normalization_strategy": "loudness",
+                "classifier_free_guidance": 6
+            }
+        )
+    except Exception as e:  # Consider specifying more specific exception types
+        return f'An error occurred during music generation: {str(e)}'
+
     if output:
         return f'<a href="{output}" target="_blank">Click here to listen to the generated music</a>'
     else:
         return 'Music generation failed or no URL returned.'
 
 def get_audio_path(selection):
-    # Map selection to your audio file URLs
     audio_files = {
         "Smooth Jazz Evening": "smooth_jazz_evening.wav",
         "Uplifting Soft Rock": "uplifting_soft_rock.wav",
         "Mellow Jazz Morning": "mellow_jazz_morning.wav",
         "Soft Rock Serenade": "soft_rock_serenade.wav",
     }
-    return audio_files[selection]
+    # Check if the selection exists in the dictionary to avoid KeyError
+    if selection in audio_files:
+        return audio_files[selection]
+    else:
+        # Handle unexpected selection
+        return 'Error: Selected audio file does not exist.'
 
 with gr.Blocks() as app:
     with gr.Tab("Generate Music"):