Update app.py

app.py

@@ -109,6 +109,7 @@ def sample_paths(paths: List[Path] | List[str], n: int = 30) -> List[str]:
 
 import base64
 
+APP_DIR = os.getcwd()
 def load_logo_base64(path: str) -> str:
     with open(path, "rb") as f:
         return base64.b64encode(f.read()).decode("utf-8")
@@ -353,7 +354,21 @@ def build_ffmpeg_extract(
     return cmd
 
     frames = sorted(raw_dir.glob(f"{prefix}_*.{out_format}"), key=_natural_key)
-    gallery = sample_paths(frames, 30)
+
+    # Adaptive preview:
+    if len(frames) <= 100:
+        gallery = [str(p) for p in frames]   # show all
+    else:
+        gallery = sample_paths(frames, 100)  # evenly sample 100
+    
+    zip_path = work / "frames.zip"
+    with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
+        for p in frames:
+            zf.write(p, p.name)
+    
+    details = f"Frames extracted: {len(frames)} | Saved to: {raw_dir}"
+    return gallery, str(zip_path), details, cmd_preview, render_progress(100.0, f"Extracted {len(frames)} frames"), [str(p) for p in frames], str(raw_dir), prefix
+
 
 def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
     if img is None:
@@ -555,7 +570,7 @@ def step1_extract(
 
     frames = sorted(raw_dir.glob(f"{prefix}_*.{out_format}"), key=_natural_key)
 
-    gallery = sample_paths(frames, 30)
+    [str(p) for p in frames]   # show all extracted frames
     zip_path = work / "frames.zip"
     with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
         for p in frames:
@@ -1039,7 +1054,7 @@ def build_ui():
                 with gr.Row():
                     btn_extract = gr.Button("Step 1: Extract Frames", variant="primary")
                 prog1 = gr.HTML(render_progress(0.0, "Idle"))
-                gallery = gr.Gallery(label="Preview (30 sampled)", columns=6, height=480)
+                gallery = gr.Gallery(label="Preview (all ≤100, else sample 100)", columns=6, height=480)
                 zip_out = gr.File(label="Download frames ZIP")
                 details1 = gr.Markdown("Ready.")
                 with gr.Accordion("Show FFmpeg command", open=False):