zip download: LoRA adapter + generated captions bundled together

app.py

@@ -873,15 +873,29 @@ def gradio_main():
                     _log("[WARN] ace-server may not have restarted -- check logs")
                 adapter_safetensors = os.path.join(adapter_out, "adapter_model.safetensors")
                 if os.path.isfile(adapter_safetensors):
-                    tmp_out = tempfile.NamedTemporaryFile(
-                        suffix=".safetensors",
+                    import zipfile
+                    tmp_zip = tempfile.NamedTemporaryFile(
+                        suffix=".zip",
                         prefix=f"{lora_name}_",
                         delete=False,
                     )
-                    tmp_out.close()
-                    shutil.copy2(adapter_safetensors, tmp_out.name)
-                    _log(f"[OK] LoRA saved: {lora_name}")
-                    yield _log_text(), gr.Button(visible=True), gr.Button(visible=False), gr.File(value=tmp_out.name, visible=True)
+                    tmp_zip.close()
+                    with zipfile.ZipFile(tmp_zip.name, "w", zipfile.ZIP_DEFLATED) as zf:
+                        zf.write(adapter_safetensors, f"{lora_name}/adapter_model.safetensors")
+                        adapter_config = os.path.join(adapter_out, "adapter_config.json")
+                        if os.path.isfile(adapter_config):
+                            zf.write(adapter_config, f"{lora_name}/adapter_config.json")
+                        # Include generated captions if they exist
+                        caption_count = 0
+                        if os.path.isdir(audio_dir):
+                            for cf in sorted(os.listdir(audio_dir)):
+                                if cf.endswith(".json"):
+                                    zf.write(os.path.join(audio_dir, cf),
+                                             f"{lora_name}/captions/{cf}")
+                                    caption_count += 1
+                    _log(f"[OK] LoRA saved: {lora_name}" +
+                         (f" ({caption_count} captions included)" if caption_count else ""))
+                    yield _log_text(), gr.Button(visible=True), gr.Button(visible=False), gr.File(value=tmp_zip.name, visible=True)
                 else:
                     yield _log_text(), gr.Button(visible=True), gr.Button(visible=False), gr.File()
                 shutil.rmtree(work_dir, ignore_errors=True)