Upload app.py

app.py

@@ -7,6 +7,7 @@ gradio API for every request; `reference_encoder` stays here, next to the autoen
 
 from __future__ import annotations
 
+import json
 import os
 import tempfile
 import time
@@ -619,6 +620,58 @@ def generate(
     return path, refined, gr.update(visible=bool(refined))
 
 
+# ----------------------------------------------------------------------------------------------------------------
+# Settings file
+# ----------------------------------------------------------------------------------------------------------------
+# Everything typed rather than uploaded, so a session can be picked up where it was left off. The references
+# themselves are deliberately left out: gradio hands them over as paths into a per-session temporary directory that
+# is gone by the next visit, so a saved path would restore as a dead file rather than as the image.
+
+SETTINGS_VERSION = 1
+SETTINGS_KEYS = (
+    ["prompt", "upsample", "canvas", "match", "duration", "steps", "seed"]
+    + [f"lora_{slot + 1}" for slot in range(LORA_SLOTS)]
+    + [f"lora_{slot + 1}_scale" for slot in range(LORA_SLOTS)]
+)
+
+
+def save_settings(*values):
+    """Write the current controls to a `.json` and reveal it for download."""
+    payload = {"version": SETTINGS_VERSION, "saved": time.strftime("%Y-%m-%d %H:%M:%S")}
+    payload.update(dict(zip(SETTINGS_KEYS, values)))
+
+    directory = os.path.join(tempfile.gettempdir(), "h3-settings")
+    os.makedirs(directory, exist_ok=True)
+    path = os.path.join(directory, f"h3-settings-{int(time.time())}.json")
+    with open(path, "w", encoding="utf-8") as handle:
+        json.dump(payload, handle, ensure_ascii=False, indent=2, default=str)
+    return gr.update(value=path, visible=True)
+
+
+def load_settings(path):
+    """Restore the controls from a `.json`. A key the file does not carry leaves its control alone, so a settings
+    file written by an older version of this Space still loads."""
+    if not path:
+        return [gr.update() for _ in SETTINGS_KEYS]
+    try:
+        with open(path, encoding="utf-8") as handle:
+            payload = json.load(handle)
+    except Exception as error:
+        raise gr.Error(f"Файлът с настройки не се чете: `{type(error).__name__}: {error}`")
+    if not isinstance(payload, dict):
+        raise gr.Error("Това не е файл с настройки на този Space.")
+
+    updates = []
+    for key in SETTINGS_KEYS:
+        value = payload.get(key)
+        # An unknown canvas label would be rejected by the conditioner, which is the wrong place to find out.
+        if value is None or (key == "canvas" and value not in CANVASES):
+            updates.append(gr.update())
+        else:
+            updates.append(gr.update(value=value))
+    return updates
+
+
 def _fill_lora_slots(files, *current):
     """Drop `.safetensors` files on the uploader and their paths land in the first free slots, so a local adapter
     needs no typing at all."""
@@ -651,6 +704,11 @@ LORA_HELP = """Each slot takes a Hugging Face repo (`owner/repo`), a file inside
 switches a slot off without clearing it. Adapters have to be trained against the `transformer_ref/` partition.
 """
 
+SETTINGS_HELP = """Saves the prompt, the canvas, the sliders and the LoRA slots — everything typed rather than
+uploaded. Images, audio and video are not saved: gradio keeps them in a temporary folder that is gone by the next
+visit, so a saved path would come back as a dead file.
+"""
+
 CSS = """
 .main.fillable { max-width: 1250px !important; }
 .dark .gradio-container { color: var(--body-text-color); }
@@ -726,6 +784,14 @@ with gr.Blocks(title="MiniMax-H3 Reference") as demo:
                 steps = gr.Slider(label="Steps", minimum=10, maximum=40, step=1, value=28)
                 seed = gr.Number(label="Seed", value=42, precision=0)
 
+            with gr.Accordion("Settings file", open=False):
+                gr.Markdown(SETTINGS_HELP)
+                save = gr.Button("Save settings to .json", size="sm")
+                settings_download = gr.File(label="Your settings", visible=False, interactive=False)
+                settings_upload = gr.File(
+                    label="Load a settings .json", file_types=[".json"], type="filepath"
+                )
+
         with gr.Column():
             result = gr.Video(label="Video + soundtrack")
             # An output, so it can be revealed only for a request that asked for a rewrite.
@@ -753,6 +819,11 @@ with gr.Blocks(title="MiniMax-H3 Reference") as demo:
     lora_inputs = [field for pair in zip(lora_references, lora_scales) for field in pair]
     lora_upload.upload(_fill_lora_slots, [lora_upload, *lora_references], lora_references, api_name=False)
 
+    # Same order as `SETTINGS_KEYS`.
+    settings_fields = [prompt, upsample, canvas, match, duration, steps, seed, *lora_references, *lora_scales]
+    save.click(save_settings, settings_fields, settings_download, api_name=False)
+    settings_upload.upload(load_settings, settings_upload, settings_fields, api_name=False)
+
     # Same order as `generate`'s signature: the five leading columns first, then the remaining image slots, then the
     # LoRA fields the `*lora_fields` tail collects.
     request = [