Update app.py

app.py

@@ -1,8 +1,8 @@
-import os, uuid, time, json, shutil, mimetypes, subprocess, requests, gradio as gr
+import os, uuid, time, shutil, mimetypes, subprocess, requests, gradio as gr
 from datetime import datetime
 from typing import Optional, List
 
-# -------- config --------
+# --- config ---
 ENDPOINT = "https://moonmath-ai-dev--moonmath-i2v-backend-moonmathinference-run.modal.run"
 FFMPEG = "ffmpeg"
 OUT, TMP = "outputs", "tmp"
@@ -83,7 +83,7 @@ def call_backend(
     if ctype.startswith("video/"):   # raw video bytes
         return save_video_bytes(r.content, ctype)
 
-    # expect JSON with { "video_url": ... } or direct mp4 URL
+    # expect JSON with { "video_url": ... }
     try:
         payload = r.json()
         url = payload.get("video_url")
@@ -96,37 +96,43 @@ def call_backend(
                 if chunk: f.write(chunk)
         return path
     except Exception:
-        # if backend returns direct bytes but mislabeled JSON, fallback
+        # fallback if backend mislabels content
         return save_video_bytes(r.content, ctype or "video/mp4")
 
+# ================= UI =================
 with gr.Blocks() as demo:
     gr.Markdown("## Continuous Video — chain last frame → next first frame")
 
     # state
     used_videos   = gr.State([])      # list[str]
-    last_seed_img = gr.State(None)    # str path (PNG) to send as image_bytes
-    current_video = gr.State(None)    # str path
+    last_seed_img = gr.State(None)    # PNG path (becomes image_bytes)
+    current_video = gr.State(None)    # latest generated video
 
     with gr.Row():
         prompt = gr.Textbox(label="Prompt", placeholder="Describe your shot…", lines=2)
     with gr.Row():
-        start_file = gr.File(label="Initial start image or video (only needed for the very first clip)", file_types=["image","video"])
+        start_file = gr.File(
+            label="Initial start image or video (only for the very first clip)",
+            file_types=["image","video"]
+        )
     with gr.Row():
         negative = gr.Textbox(label="Negative prompt (optional)", placeholder="What to avoid…")
     with gr.Row():
         fps   = gr.Slider(1,60, value=24, step=1, label="Frames per second")
         vlen  = gr.Slider(1,12, value=4,  step=1, label="Video length (seconds)")
-        steps = gr.Slider(1,100, value=30, step=1, label="Num inference steps (optional)", interactive=True)
-        seed  = gr.Number(label="Seed (optional)", precision=0, value=None)
+        steps = gr.Slider(1,100, value=30, step=1, label="Num inference steps (optional)")
+        seed  = gr.Number(label="Seed (optional)", precision=0)
 
     video_out = gr.Video(label="Output")
+    status_md = gr.Markdown("")  # shows ✅ Saved… after Chain
     with gr.Row():
         gen_btn  = gr.Button("Generate", variant="primary")
-        use_btn  = gr.Button("Chain")
+        chain_btn= gr.Button("Chain (save & clear)")
         dl_btn   = gr.Button("Download")
 
     files_out = gr.Files(label="Downloads")
 
+    # ---- handlers ----
     def on_generate(prompt_txt, start_file_obj, neg, fps_val, vlen_val, steps_val, seed_val, seed_img):
         if not prompt_txt or not prompt_txt.strip():
             raise gr.Error("Prompt is required.")
@@ -147,37 +153,51 @@ with gr.Blocks() as demo:
             )
         except Exception as e:
             raise gr.Error(str(e))
-        return out, out  # preview path, current_video state
+        return out, out, gr.update(value="")  # show video, set state, clear status
 
     gen_btn.click(
         on_generate,
         inputs=[prompt, start_file, negative, fps, vlen, steps, seed, last_seed_img],
-        outputs=[video_out, current_video]
+        outputs=[video_out, current_video, status_md]
     )
 
-    def on_use(curr, used):
+    def on_chain(curr, used):
+        """
+        Save current video, steal last frame for next seed,
+        and CLEAR: preview, prompt, negative, start_file.
+        Also show a visible 'saved' banner.
+        """
         if not curr or not os.path.exists(curr):
             raise gr.Error("Generate a video first.")
-        if curr not in used: used = used + [curr]
-        try:
-            seed_img = extract_last_frame(curr)
-        except Exception as e:
-            raise gr.Error(f"Could not extract last frame: {e}")
-        return used, seed_img
-
-    use_btn.click(
-        on_use,
+        new_used = used[:] if used else []
+        if curr not in new_used: new_used.append(curr)
+        # last frame for next seed
+        seed_img = extract_last_frame(curr)
+        saved_idx = len(new_used)
+        saved_msg = f"✅ **Saved clip #{saved_idx}** — last frame captured. Ready for your next prompt."
+
+        return (
+            new_used,            # used_videos
+            seed_img,            # last_seed_img (image_bytes for next gen)
+            gr.update(value=None), # CLEAR prompt
+            gr.update(value=None), # CLEAR negative prompt
+            gr.update(value=None), # CLEAR start_file
+            gr.update(value=None), # CLEAR video preview
+            gr.update(value=saved_msg)  # status banner
+        )
+
+    chain_btn.click(
+        on_chain,
         inputs=[current_video, used_videos],
-        outputs=[used_videos, last_seed_img]
+        outputs=[used_videos, last_seed_img, prompt, negative, start_file, video_out, status_md]
     )
 
     def on_download(used):
-        if not used: raise gr.Error("Nothing to download. Click Use after generating clips.")
+        if not used: raise gr.Error("Nothing to download. Click Chain after generating clips.")
         files = []
         try:
             merged = concat_videos(used); files.append(merged)
         except Exception as e:
-            # still allow ZIP even if concat fails
             print("concat error:", e)
         try:
             zipped = zip_used(used); files.append(zipped)