add LoRA adapter dropdown to inference UI

app.py

@@ -234,8 +234,16 @@ def gradio_main():
     import gradio as gr
 
     # -- Generate tab handler --
+    def get_trained_loras():
+        loras = ["None (no LoRA)"]
+        if os.path.isdir(ADAPTER_DIR):
+            for d in os.listdir(ADAPTER_DIR):
+                if os.path.isdir(os.path.join(ADAPTER_DIR, d)):
+                    loras.append(d)
+        return loras
+
     def generate_music(caption, lyrics, instrumental, bpm, duration, seed,
-                       steps, progress=gr.Progress(track_tqdm=True)):
+                       steps, lora_select, progress=gr.Progress(track_tqdm=True)):
         if not _server_ok():
             return None, "ace-server not running. Check logs."
 
@@ -243,6 +251,7 @@ def gradio_main():
             lyrics = "[Instrumental]"
 
         actual_seed = None if seed is None or int(seed) < 0 else int(seed)
+        adapter = None if lora_select == "None (no LoRA)" else lora_select
 
         progress_map = {
             "lm_submit": (0.05, "Submitting LM job..."),
@@ -267,6 +276,7 @@ def gradio_main():
                 seed=actual_seed,
                 steps=steps,
                 output_format="mp3",
+                adapter=adapter,
                 progress_cb=gr_progress,
             )
             return audio_path, status
@@ -502,6 +512,7 @@ finally:
                         value=8, step=1, scale=1,
                     )
                     seed = gr.Number(label="Seed (-1=random)", value=-1, scale=1)
+                    lora_select = gr.Dropdown(label="LoRA", choices=get_trained_loras(), value="None (no LoRA)", scale=1)
 
                 with gr.Row(elem_classes="compact-row"):
                     gen_btn = gr.Button("Generate Music", variant="primary", scale=2)
@@ -510,7 +521,7 @@ finally:
                 gen_btn.click(
                     fn=generate_music,
                     inputs=[caption, lyrics, instrumental, bpm, duration,
-                            seed, steps],
+                            seed, steps, lora_select],
                     outputs=[audio_out, status],
                     api_name="generate",
                 )