well that was easier

app.py

@@ -140,7 +140,7 @@ def generate_music(wav_filename, prompt_duration, musicgen_model, output_duratio
 
 # ========== MELODYFLOW FUNCTIONS (Via Facebook Space) ==========
 
-def transform_with_melodyflow_api(audio_path, variation, custom_prompt="", solver="euler", steps=128):
+def transform_with_melodyflow_api(audio_path, variation, custom_prompt="", solver="euler", flowstep=0.12):
     """Transform audio using Facebook/MelodyFlow space API"""
     if audio_path is None:
         return None, "❌ No audio file provided"
@@ -152,10 +152,13 @@ def transform_with_melodyflow_api(audio_path, variation, custom_prompt="", solve
         # Determine the prompt to use
         if custom_prompt.strip():
             prompt_text = custom_prompt.strip()
-            status_msg = f"✅ Transformed with custom prompt: '{prompt_text}'"
+            status_msg = f"✅ Transformed with custom prompt: '{prompt_text}' (flowstep: {flowstep})"
         else:
             prompt_text = VARIATION_PROMPTS.get(variation, f"transform this audio to {variation} style")
-            status_msg = f"✅ Transformed with {variation} style"
+            status_msg = f"✅ Transformed with {variation} style (flowstep: {flowstep})"
+        
+        # Set steps based on solver (from your original logic)
+        steps = 64 if solver == "midpoint" else 128
         
         # Call the MelodyFlow API
         result = client.predict(
@@ -163,8 +166,8 @@ def transform_with_melodyflow_api(audio_path, variation, custom_prompt="", solve
             text=prompt_text,
             solver=solver,
             steps=steps,
-            target_flowstep=0,  # Default flowstep
-            regularize=False,
+            target_flowstep=flowstep,  # This is the key parameter!
+            regularize=solver == "euler",  # Regularize for euler, not for midpoint
             regularization_strength=0.2,
             duration=30,  # Max duration
             melody=handle_file(audio_path),
@@ -248,12 +251,13 @@ with gr.Blocks() as iface:
             choices=["euler", "midpoint"],
             value="euler"
         )
-        transform_steps = gr.Slider(
-            label="Steps",
-            minimum=64,
-            maximum=256,
-            step=32,
-            value=128
+        transform_flowstep = gr.Slider(
+            label="Transform Intensity (Flowstep)",
+            minimum=0.0,
+            maximum=0.15,
+            step=0.01,
+            value=0.12,
+            info="Lower values = more dramatic transformation"
         )
         transform_button = gr.Button("🎛️ Transform Audio", variant="secondary", size="lg")
     
@@ -302,7 +306,7 @@ with gr.Blocks() as iface:
     # Step 2: Transform (using Facebook MelodyFlow API)
     transform_button.click(
         transform_with_melodyflow_api,
-        inputs=[main_audio, transform_variation, transform_prompt, transform_solver, transform_steps],
+        inputs=[main_audio, transform_variation, transform_prompt, transform_solver, transform_flowstep],
         outputs=[main_audio, transform_status]
     )