Update app.py

app.py

@@ -20,8 +20,8 @@ def render_logo_html(px: int = 96) -> str:
     <div style="display:flex;align-items:center;gap:16px;">
       {img}
       <div>
-        <div style="font-size:1.6rem;font-weight:800;">Bifröst · Re-encode Video</div>
-        <div style="opacity:0.8;">Frames → Video, optional audio mux</div>
+        <div style="font-size:1.6rem;font-weight:800;">Valknut · Re-encode Video</div>
+        <div style="opacity:0.8;">Frames → Video, optional audio mix</div>
       </div>
     </div>
     <hr>
@@ -69,24 +69,34 @@ def prepare_frames_from_upload(files: List[gr.File] | None, prefix: str = "enc")
         counter += 1
     return str(frames_dir), prefix
 
-def build_ffmpeg_encode(frames_dir: str, prefix: str, fps: float, fmt: str, include_audio: bool, orig_video: str | None) -> list[str]:
-    pattern_jpg = Path(frames_dir) / f"{prefix}_%05d.jpg"
-    pattern_png = Path(frames_dir) / f"{prefix}_%05d.png"
-    pattern = str(pattern_jpg if pattern_jpg.exists() else pattern_png)
-    if not Path(pattern.replace("%05d", "00001")).with_suffix(Path(pattern).suffix).exists():
-        # if first frame isn't 00001, we still let ffmpeg start_number 1; files must be named *_00001.*
-        pass
-    args = [FFMPEG, "-y", "-start_number", "1", "-framerate", f"{fps:.6f}", "-i", pattern]
+def build_ffmpeg_encode(frames_dir: str, prefix: str, fps: float, fmt: str,
+                        include_audio: bool, orig_video: str | None) -> list[str]:
+    jpgs = sorted(Path(frames_dir).glob(f"{prefix}_*.jpg"))
+    pngs = sorted(Path(frames_dir).glob(f"{prefix}_*.png"))
+    imgs = jpgs if jpgs else pngs
+    if not imgs:
+        return []
+
+    first_frame = imgs[0].name
+    pattern = str(imgs[0].with_name(f"{prefix}_%05d{imgs[0].suffix}"))
+    start_num = int(Path(first_frame).stem.split("_")[-1])
+
+    args = [FFMPEG, "-y", "-start_number", str(start_num),
+            "-framerate", f"{fps:.6f}", "-i", pattern]
+
     if include_audio and orig_video:
         args += ["-i", orig_video, "-map", "0:v:0", "-map", "1:a:0", "-shortest"]
+
     if fmt == "h265":
         vcodec = ["-c:v", "libx265"]
     elif fmt == "vp9":
         vcodec = ["-c:v", "libvpx-vp9"]
     else:
         vcodec = ["-c:v", "libx264"]
-    args += vcodec + ["-pix_fmt", "yuv420p", "-crf", "18", "output.mp4" if fmt in ("h264", "h265") else "output.webm"]
-    return args
+
+    out_name = "output.mp4" if fmt in ("h264", "h265") else "output.webm"
+    return args + vcodec + ["-pix_fmt", "yuv420p", "-crf", "18", out_name]
+
 
 def step3_encode(
     uploaded_frames: List[gr.File] | None,
@@ -109,16 +119,21 @@ def step3_encode(
 
     # Add progress pipe and run in frames_dir for clean output pathing
     cmd = [cmd[0], "-progress", "pipe:2"] + cmd[1:]
-    proc = subprocess.Popen(cmd, stderr=subprocess.PIPE, stdout=subprocess.DEVNULL, text=True, bufsize=1, cwd=frames_dir)
-
+    proc = subprocess.Popen(
+        cmd, stderr=subprocess.PIPE, stdout=subprocess.DEVNULL,
+        text=True, bufsize=1, cwd=frames_dir
+    )
+    
     # Approximate by counting files
     total_frames = len(list(Path(frames_dir).glob(f"{prefix}_*.jpg"))) + len(list(Path(frames_dir).glob(f"{prefix}_*.png")))
     current = 0
-
+    errors = []   # ✅ collect errors while streaming
+    
     while True:
         line = proc.stderr.readline()
         if not line and proc.poll() is not None:
             break
+    
         if "frame=" in line:
             try:
                 current = int(line.strip().split("=")[-1])
@@ -129,18 +144,20 @@ def step3_encode(
                 yield None, f"Encoding… {current}/{total_frames} frames", render_progress(pct, f"Encoding {pct:.0f}%")
             else:
                 yield None, "Encoding…", render_progress(50.0, "Encoding…")
-
+    
+        # ✅ collect suspicious lines (errors, missing files, etc.)
+        if "Error" in line or "No such file" in line:
+            errors.append(line.strip())
+    
     ret = proc.wait()
     out_file = Path(frames_dir) / ("output.mp4" if fmt in ("h264", "h265") else "output.webm")
+    
     if ret != 0 or not out_file.exists():
-        err = ""
-        try:
-            err = proc.stderr.read() if proc.stderr else ""
-        except Exception:
-            pass
+        err = "\n".join(errors) or "Unknown ffmpeg error."
         yield None, f"Encoding failed.\n\n{err}", render_progress(0.0, "Failed")
         return
 
+
     yield str(out_file), f"Video created: {out_file.name}", render_progress(100.0, "Encoding complete")
 
 def build_ui():