Upload app.py

app.py

@@ -1456,6 +1456,124 @@ def identify_loras(*references):
     return "  \n".join(lines)
 
 
+# ------------------------------------------------------------------------------------------------------------------
+# Continuing a scene, and searching CivitAI
+# ------------------------------------------------------------------------------------------------------------------
+
+
+def last_frame_of(video_path) -> str:
+    """Write the final frame of a clip to a PNG and return its path, so it can be dropped straight into an image
+    slot. A few frames back from the very end on purpose: the last frame is often a duplicate or half-written, and a
+    soft still starts the next clip badly."""
+    if not video_path or not os.path.exists(str(video_path)):
+        raise gr.Error("Generate a video first - there is nothing to continue from.")
+
+    import av
+    from PIL import Image
+
+    frames = []
+    with av.open(str(video_path)) as container:
+        stream = container.streams.video[0]
+        stream.thread_type = "AUTO"
+        for frame in container.decode(stream):
+            frames.append(frame)
+            if len(frames) > 4:
+                frames.pop(0)
+
+    if not frames:
+        raise gr.Error("That video has no readable frames.")
+
+    chosen = frames[max(0, len(frames) - 3)]
+    directory = os.path.join(tempfile.gettempdir(), "continuations")
+    os.makedirs(directory, exist_ok=True)
+    path = os.path.join(directory, f"frame_{int(time.time() * 1000)}.png")
+    Image.fromarray(chosen.to_ndarray(format="rgb24")).save(path)
+    return path
+
+
+def stage_extension(video_path, queue, name_hint):
+    """Before a continuation runs: park the finished clip in the merge queue and hand its last frame back as the new
+    first reference."""
+    frame = last_frame_of(video_path)
+    merged, merged_file, queue, status = add_to_queue(video_path, queue, name_hint)
+    return frame, None, merged, merged_file, queue, status
+
+
+H3_BASE_MODELS = ["MiniMax H3", "(any base model)"]
+
+
+def civitai_search(query, base_model, want_nsfw, limit=20):
+    """Search CivitAI for lora. `/api/v1/models` embeds each model's versions, files and trigger words, so one call
+    gives everything a slot needs. Returns `(readable list, dropdown update, {label: url})`."""
+    import requests
+
+    query = (query or "").strip()
+    if not query:
+        return "Type something to search for.", gr.update(choices=[], value=None), {}
+
+    params = {"query": query, "types": "LORA", "limit": int(limit), "sort": "Most Downloaded"}
+    if base_model and base_model != "(any base model)":
+        params["baseModels"] = base_model
+    if want_nsfw:
+        params["nsfw"] = "true"
+
+    headers = {"User-Agent": "Mozilla/5.0"}
+    token = os.environ.get("CIVITAI_TOKEN", "").strip()
+    if token:
+        headers["Authorization"] = f"Bearer {token}"
+
+    try:
+        response = requests.get("https://civitai.com/api/v1/models", params=params, headers=headers, timeout=30)
+        response.raise_for_status()
+        items = response.json().get("items") or []
+    except Exception as error:  # noqa: BLE001
+        return f"Search failed: `{type(error).__name__}: {error}`", gr.update(choices=[], value=None), {}
+
+    if not items:
+        return ("Nothing found. Try fewer words, or set the base model to *(any base model)*.",
+                gr.update(choices=[], value=None), {})
+
+    mapping, lines, choices = {}, [], []
+    for item in items:
+        model_name = item.get("name") or "?"
+        creator = (item.get("creator") or {}).get("username") or "?"
+        downloads = (item.get("stats") or {}).get("downloadCount") or 0
+
+        for version in (item.get("modelVersions") or [])[:3]:
+            version_name = version.get("name") or ""
+            words = [w for w in (version.get("trainedWords") or []) if w]
+            for entry in (version.get("files") or []):
+                name = entry.get("name") or ""
+                if not name.lower().endswith(".safetensors"):
+                    continue
+                url = f"https://civitai.com/api/download/models/{version.get('id')}?fileId={entry.get('id')}"
+                size = float(entry.get("sizeKB") or 0) / 1024
+                label = f"{model_name} · {version_name} · {name}"[:150]
+                mapping[label] = url
+                choices.append(label)
+                line = f"**{model_name}** · {version_name} · {name}  \n<sub>{size:.0f} MB · {downloads} downloads · by {creator}"
+                if words:
+                    line += f" · triggers: {', '.join(words[:5])}"
+                lines.append(line + "</sub>")
+
+    if not choices:
+        return "Found models, but none with a `.safetensors` file.", gr.update(choices=[], value=None), {}
+
+    return "  \n".join(lines[:40]), gr.update(choices=choices, value=choices[0]), mapping
+
+
+def put_in_slot(label, mapping, slot):
+    """Write the chosen result's download link into one lora slot, leaving the others alone."""
+    updates = [gr.update() for _ in range(LORA_SLOTS)]
+    url = (mapping or {}).get(label)
+    if not url:
+        return [*updates, "Search and pick a file first."]
+
+    index = int(str(slot).split()[-1]) - 1
+    updates[index] = gr.update(value=url)
+    return [*updates, f"**Put into {slot}** — {label}"]
+
+
 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."""
@@ -1653,7 +1771,7 @@ CSS = """
     border-radius: 14px !important;
     box-shadow: 0 8px 22px rgba(29, 78, 216, .25);
 }
-#lora-identify, #profile-refresh, #seed-dice, #turbo-btn, #profile-save, #profile-load, #profile-delete {
+#search-btn, #search-put, #extend-btn, #lora-identify, #profile-refresh, #seed-dice, #turbo-btn, #profile-save, #profile-load, #profile-delete {
     min-height: 42px !important; border-radius: 12px !important; font-weight: 600 !important;
 }
 
@@ -1684,7 +1802,7 @@ THEME = gr.themes.Soft(primary_hue="blue", secondary_hue="cyan", neutral_hue="sl
 
 HERO = """
 <div id="h3-hero">
-  <h1>MiniMax-H3 &middot; reference &rarr; video + soundtrack &middot; Custom lora + CivitAI links, GPU cost, profiles, clip stitching</h1>
+  <h1>MiniMax-H3 &middot; reference &rarr; video + soundtrack &middot; Custom lora + CivitAI search, scene continuation, GPU cost, profiles, clip stitching</h1>
   <p>33B model generating video and a fully synchronized soundtrack (ambience, foley, speech) from your own subject,
   voice or camera move.
   <a href="https://huggingface.co/MiniMaxAI/MiniMax-H3" target="_blank" rel="noopener">model</a> &middot;
@@ -1692,7 +1810,7 @@ HERO = """
   <a href="https://huggingface.co/spaces/multimodalart/minimax-h3" target="_blank" rel="noopener">text / image to video</a></p>
   <div class="pills">
     <span>33B</span><span>joint video + audio</span><span>Turbo lora: 4&ndash;8 steps</span>
-    <span>ComfyUI lora accepted</span><span>up to 9 references</span><span>5 custom lora slots</span><span>named profiles</span><span>CivitAI links</span><span>kohya + LoKr auto-convert</span><span>GPU cost estimate</span><span>clip stitching with audio</span>
+    <span>ComfyUI lora accepted</span><span>up to 9 references</span><span>5 custom lora slots</span><span>named profiles</span><span>CivitAI search + links</span><span>scene continuation</span><span>kohya + LoKr auto-convert</span><span>GPU cost estimate</span><span>clip stitching with audio</span>
   </div>
 </div>
 """
@@ -1807,6 +1925,28 @@ with gr.Blocks(title="MiniMax-H3 - Custom lora + CivitAI, GPU cost, profiles, cl
             with gr.Tabs():
                 with gr.Tab(f"⭐ Custom lora ({LORA_SLOTS} slots)"):
                     gr.Markdown(LORA_HELP)
+                    with gr.Accordion("🔍 Search CivitAI", open=False):
+                        gr.Markdown(
+                            "Search CivitAI without leaving the Space, then drop a result straight into a slot. "
+                            "Each hit shows its size, downloads and trigger words."
+                        )
+                        with gr.Row():
+                            search_query = gr.Textbox(label="Search", placeholder="e.g. dance, rain, camera move",
+                                                      scale=3)
+                            search_base = gr.Dropdown(H3_BASE_MODELS, value=H3_BASE_MODELS[0], label="Base model",
+                                                      allow_custom_value=True, scale=2)
+                        with gr.Row():
+                            search_nsfw = gr.Checkbox(value=True, label="Include NSFW results")
+                            search_btn = gr.Button("🔍 Search", variant="secondary", elem_id="search-btn")
+                        search_results = gr.Markdown("No search yet.")
+                        search_pick = gr.Dropdown(choices=[], label="Pick a file")
+                        with gr.Row():
+                            search_slot = gr.Dropdown([f"lora {i}" for i in range(1, LORA_SLOTS + 1)],
+                                                      value="lora 1", label="Into slot", scale=2)
+                            search_put_btn = gr.Button("⬇️ Put it in", variant="primary", scale=1,
+                                                       elem_id="search-put")
+                        search_state = gr.State({})
+
                     lora_references, lora_scales = [], []
                     for slot in range(LORA_SLOTS):
                         with gr.Row():
@@ -1889,6 +2029,14 @@ with gr.Blocks(title="MiniMax-H3 - Custom lora + CivitAI, GPU cost, profiles, cl
         with gr.Column(scale=6):
             with gr.Group(elem_classes="panel"):
                 result = gr.Video(label="🎞️ Video + soundtrack", height=560)
+                extend_btn = gr.Button("➕ Continue the scene (one more clip, joined on)",
+                                       variant="secondary", elem_id="extend-btn")
+                gr.Markdown(
+                    "Takes the last frame of the clip above, makes it the first reference, generates again with "
+                    "the settings untouched, and joins the two into one file — soundtrack included. Press it again "
+                    "for a third clip. Each press costs one normal generation; the joining is free.",
+                    elem_classes="turbo-blurb",
+                )
             # An output, so it can be revealed only for a request that asked for a rewrite.
             with gr.Accordion("Upsampled prompt", open=False, visible=False) as upsampled_panel:
                 upsampled = gr.Textbox(show_label=False, lines=8, interactive=False)
@@ -1974,6 +2122,38 @@ with gr.Blocks(title="MiniMax-H3 - Custom lora + CivitAI, GPU cost, profiles, cl
         show_progress="hidden", api_name=False,
     )
 
+    # Continuing a scene: park the finished clip, hand its last frame to the first image slot, generate from it,
+    # then join the new clip on. `request` starts with the prompt and the first image, so the second step reads
+    # the still the first step just wrote.
+    extend_btn.click(
+        fn=stage_extension,
+        inputs=[result, merge_queue, merge_name],
+        outputs=[images[0], result, merged_video, merged_file, merge_queue, merge_status],
+        api_name=False,
+    ).then(
+        roll_seed, [randomize_seed, seed], seed, show_progress="hidden", api_name=False,
+    ).then(
+        generate, request, [result, upsampled, upsampled_panel], api_name=False,
+    ).then(
+        auto_queue, [result, auto_merge, merge_queue, merge_name],
+        [merged_video, merged_file, merge_queue, merge_status],
+        show_progress="hidden", api_name=False,
+    )
+
+    # Searching CivitAI, and dropping a result into a slot.
+    search_btn.click(
+        civitai_search, [search_query, search_base, search_nsfw],
+        [search_results, search_pick, search_state], api_name=False,
+    )
+    search_query.submit(
+        civitai_search, [search_query, search_base, search_nsfw],
+        [search_results, search_pick, search_state], api_name=False,
+    )
+    search_put_btn.click(
+        put_in_slot, [search_pick, search_state, search_slot],
+        [*lora_references, search_results], api_name=False,
+    )
+
     merge_add_btn.click(
         add_to_queue, [result, merge_queue, merge_name],
         [merged_video, merged_file, merge_queue, merge_status], api_name=False,