updated musicgen api handler

app.py

@@ -390,16 +390,16 @@ def extend_with_musicgen_api(audio_path, prompt_duration, musicgen_model, output
         client = Client("thepatch/micro-slot-machine")
         
         print(f"🎼 MusicGen extension:")
-        print(f"   Prompt duration: {prompt_duration}s")
+        print(f"   Prompt duration: {prompt_duration} (type: {type(prompt_duration)})")
         print(f"   Model: {musicgen_model}")
-        print(f"   Output duration: {output_duration}s")
+        print(f"   Output duration: {output_duration} (type: {type(output_duration)})")
         
         # Call the continue_music API
         result = client.predict(
             input_audio_path=handle_file(audio_path),
-            prompt_duration=str(prompt_duration),
+            prompt_duration=prompt_duration,  # Already a string from dropdown
             musicgen_model=musicgen_model,
-            output_duration=output_duration,
+            output_duration=float(output_duration),  # Ensure it's a float
             api_name="/continue_music"
         )
         
@@ -644,16 +644,16 @@ with gr.Blocks(title="stable-melodyflow") as iface:
         with gr.Column():
             musicgen_prompt_duration = gr.Dropdown(
                 label="prompt duration (seconds)",
-                choices=[3, 5, 7, 10],
-                value=5,
+                choices=["3", "5", "7", "10"],  # Strings to match API requirements
+                value="5",
                 info="how much of the end to use as prompt for continuation"
             )
             musicgen_output_duration = gr.Slider(
                 label="extension duration (seconds)",
-                minimum=10,
-                maximum=30,
-                step=1,
-                value=20,
+                minimum=10.0,
+                maximum=30.0,
+                step=1.0,
+                value=20.0,
                 info="how much new audio to generate"
             )
             
@@ -662,10 +662,11 @@ with gr.Blocks(title="stable-melodyflow") as iface:
                 label="musicgen model",
                 choices=[
                     "thepatch/vanya_ai_dnb_0.1 (small)",
-                    "thepatch/bleeps-medium (medium)"
+                    "thepatch/bleeps-medium (medium)",
+                    "thepatch/hoenn_lofi (large)"
                 ],
                 value="thepatch/vanya_ai_dnb_0.1 (small)",
-                info="drum & bass focused vs experimental bleeps"
+                info="various musicgen fine-tunes for different styles"
             )
     
     extend_btn = gr.Button("extend with musicgen", variant="primary", size="lg")