Upload app.py

app.py

@@ -215,6 +215,12 @@ if default_model in model_choices:
     model_choices.remove(default_model)
     model_choices.insert(0, default_model)
 
+# Handle the case where no models are found
+if not model_choices:
+    print("No model files found in the specified directory.")
+    model_choices = ["No models available"]
+    default_model = "No models available"
+
 # Set up dark grey theme
 theme = gr.themes.Monochrome(
     primary_hue="gray",
@@ -234,7 +240,7 @@ with gr.Blocks(theme=theme) as iface:
         """)
     
     with gr.Row():
-        model_dropdown = gr.Dropdown(choices=model_choices, label="Select Model", value=default_model if default_model in model_choices else model_choices[0])
+        model_dropdown = gr.Dropdown(choices=model_choices, label="Select Model", value=default_model)
     
     with gr.Row():
         prompt = gr.Textbox(label="Prompt")
@@ -250,15 +256,21 @@ with gr.Blocks(theme=theme) as iface:
     output_audio = gr.Audio(type="filepath")
 
     def on_model_change(model_name):
-        load_model(model_name)
+        if model_name != "No models available":
+            load_model(model_name)
+        else:
+            print("No valid model selected.")
 
     model_dropdown.change(on_model_change, inputs=[model_dropdown])
     generate_button.click(generate_music, inputs=[prompt, seed, cfg_scale, steps, duration], outputs=[output_status, output_audio])
 
-    # Load default model on startup
-    default_model_path = os.path.join(MODELS_DIR, default_model)
-    if os.path.exists(default_model_path):
-        iface.load(lambda: load_model(default_model), inputs=None, outputs=None)
+    # Load default model on startup only if it exists
+    if default_model != "No models available":
+        default_model_path = os.path.join(MODELS_DIR, default_model)
+        if os.path.exists(default_model_path):
+            iface.load(lambda: load_model(default_model), inputs=None, outputs=None)
+    else:
+        print("No default model available to load.")
 
 # Launch the interface
 iface.launch()