Update app.py

app.py

@@ -11,13 +11,12 @@
 
 
 import sys, types
-try:
-    import torchvision.transforms.functional_tensor as _ft  # noqa: F401
-except Exception:
-    from torchvision.transforms import functional as _F
-    _mod = types.ModuleType("torchvision.transforms.functional_tensor")
-    _mod.rgb_to_grayscale = _F.rgb_to_grayscale
-    sys.modules["torchvision.transforms.functional_tensor"] = _mod
+def _rgb_to_grayscale_np(arr: np.ndarray) -> np.ndarray:
+    # arr: HxWx3 uint8
+    r, g, b = arr[...,0], arr[...,1], arr[...,2]
+    gray = (0.2989*r + 0.5870*g + 0.1140*b).astype(arr.dtype)
+    return np.stack([gray, gray, gray], axis=-1)
+
 
 # ────────────────────────────────────────────────────────
 # Standard imports
@@ -73,7 +72,28 @@ def sample_paths(paths: List[Path] | List[str], n: int = 30) -> List[str]:
             out.append(str(paths[i]))
             seen.add(i)
     return out
-    
+
+import base64
+
+def load_logo_base64(path: str) -> str:
+    with open(path, "rb") as f:
+        return base64.b64encode(f.read()).decode("utf-8")
+
+# Preload your Bifröst logo
+LOGO_B64 = load_logo_base64(os.path.join("assets", "bifrost_logo.png"))
+
+def render_logo_html(px: int = 96) -> str:
+    return f"""
+    <div style="display:flex;align-items:center;gap:16px;">
+      <img src="data:image/png;base64,{LOGO_B64}" style="height:{px}px;width:auto;" />
+      <div>
+        <div style="font-size:1.6rem;font-weight:800;">Bifröst Beam</div>
+        <div style="opacity:0.8;">Video → Frames → Upscale (Nordic Inspired)</div>
+      </div>
+    </div>
+    <hr>
+    """
+
 # Flag so UI can know if realesrgan is importable
 HAVE_REALESRGAN = True
 
@@ -856,94 +876,94 @@ def step3_encode(frames_dir_state: str | None, prefix_state: str | None, orig_vi
 
 # ───────────────── Quick Mode — one click: All frames → Upscale ×4 → MP4 (audio)
 
-def quick_mode(video: gr.File | None, start_time: str, end_time: str, resize_long: int, prefix_in: str, prog_html: str):
-    if not video or not video.name:
-        return None, None, None, "Upload a video.", prog_html
-    if not (FFMPEG and FFPROBE and HAVE_REALESRGAN):
-        return None, None, None, "Missing deps (ffmpeg/ffprobe/realesrgan). See requirements.txt.", prog_html
+#def quick_mode(video: gr.File | None, start_time: str, end_time: str, resize_long: int, prefix_in: str, prog_html: str):
+#    if not video or not video.name:
+#        return None, None, None, "Upload a video.", prog_html
+#    if not (FFMPEG and FFPROBE and HAVE_REALESRGAN):
+#        return None, None, None, "Missing deps (ffmpeg/ffprobe/realesrgan). See requirements.txt.", prog_html
 
-    info = parse_video_info(ffprobe_json(video.name))
-    in_fps = info.get("fps") or 30.0
-    prefix = sanitize_prefix(prefix_in) or Path(video.name).stem
+#    info = parse_video_info(ffprobe_json(video.name))
+#    in_fps = info.get("fps") or 30.0
+#    prefix = sanitize_prefix(prefix_in) or Path(video.name).stem
 
-    work = Path(tempfile.mkdtemp(prefix="quick_"))
-    raw_dir = work / "frames_raw"; raw_dir.mkdir(parents=True, exist_ok=True)
-    up_dir = work / "upscaled";   up_dir.mkdir(parents=True, exist_ok=True)
+#    work = Path(tempfile.mkdtemp(prefix="quick_"))
+#    raw_dir = work / "frames_raw"; raw_dir.mkdir(parents=True, exist_ok=True)
+#    up_dir = work / "upscaled";   up_dir.mkdir(parents=True, exist_ok=True)
 
     # Extract all frames
-    extract_cmd = build_ffmpeg_extract(
-        input_path=video.name,
-        mode="All frames",
-        every_seconds=1.0,
-        nth_frame=1,
-        exact_fps=in_fps,
-        start_time=(start_time or "").strip(),
-        end_time=(end_time or "").strip(),
-        long_side=resize_long,
-        out_format="jpg",
-        jpg_quality=3,
-        png_level=2,
-        scene_detect=False,
-        scene_thresh=0.3,
-        out_pattern=str(raw_dir / f"{prefix}_%05d.jpg"),
-    )
-    proc = subprocess.Popen(extract_cmd, stderr=subprocess.PIPE, stdout=subprocess.DEVNULL, text=True, bufsize=1)
-    est = estimate_output_count("All frames", info.get("duration"), in_fps, 1.0, 1, in_fps)
-    created = 0
-    while True:
-        line = proc.stderr.readline()
-        if not line and proc.poll() is not None:
-            break
-        if int(time.time()*10) % 3 == 0:
-            created = len(list(raw_dir.glob(f"{prefix}_*.jpg")))
-            pct = min(100.0, (created / est) * 100.0) if est else 0
-            prog_html = render_progress(pct, f"Phase 1/3: Extracting {created}/{est or '?'}")
-    proc.wait()
-
-    frames = sorted(raw_dir.glob(f"{prefix}_*.jpg"))
-    if not frames:
-        return None, None, None, "No frames extracted in Quick Mode.", prog_html
+#    extract_cmd = build_ffmpeg_extract(
+#        input_path=video.name,
+#        mode="All frames",
+#        every_seconds=1.0,
+#        nth_frame=1,
+#        exact_fps=in_fps,
+#        start_time=(start_time or "").strip(),
+#        end_time=(end_time or "").strip(),
+#        long_side=resize_long,
+#        out_format="jpg",
+#        jpg_quality=3,
+#        png_level=2,
+#        scene_detect=False,
+#        scene_thresh=0.3,
+#        out_pattern=str(raw_dir / f"{prefix}_%05d.jpg"),
+#    )
+#    proc = subprocess.Popen(extract_cmd, stderr=subprocess.PIPE, stdout=subprocess.DEVNULL, text=True, bufsize=1)
+#    est = estimate_output_count("All frames", info.get("duration"), in_fps, 1.0, 1, in_fps)
+#    created = 0
+#    while True:
+#        line = proc.stderr.readline()
+#        if not line and proc.poll() is not None:
+#            break
+#        if int(time.time()*10) % 3 == 0:
+#            created = len(list(raw_dir.glob(f"{prefix}_*.jpg")))
+#            pct = min(100.0, (created / est) * 100.0) if est else 0
+#            prog_html = render_progress(pct, f"Phase 1/3: Extracting {created}/{est or '?'}")
+#    proc.wait()
+
+#    frames = sorted(raw_dir.glob(f"{prefix}_*.jpg"))
+#    if not frames:
+#        return None, None, None, "No frames extracted in Quick Mode.", prog_html
 
     # Upscale x4
-    device = "cuda" if os.environ.get("CUDA_VISIBLE_DEVICES") else "cpu"
-    upsampler = get_realesrganer("x4plus", 4, 0, (device=="cuda"), device=device)
-
-    total = len(frames)
-    done = 0
-    for fp in frames:
-        img = Image.open(fp).convert("RGB")
-        output, _ = upsampler.enhance(np.array(img), outscale=4)
-        Image.fromarray(output).save(up_dir / (Path(fp).stem + ".jpg"), quality=95)
-        done += 1
-        pct = (done/total)*100 if total else 0
-        prog_html = render_progress(pct, f"Phase 2/3: Upscaling {done}/{total}")
+#    device = "cuda" if os.environ.get("CUDA_VISIBLE_DEVICES") else "cpu"
+#    upsampler = get_realesrganer("x4plus", 4, 0, (device=="cuda"), device=device)
+
+#    total = len(frames)
+#    done = 0
+#    for fp in frames:
+#        img = Image.open(fp).convert("RGB")
+#        output, _ = upsampler.enhance(np.array(img), outscale=4)
+#        Image.fromarray(output).save(up_dir / (Path(fp).stem + ".jpg"), quality=95)
+#        done += 1
+#        pct = (done/total)*100 if total else 0
+#        prog_html = render_progress(pct, f"Phase 2/3: Upscaling {done}/{total}")
 
     # Encode MP4 with audio
-    encode_cmd = build_ffmpeg_encode(str(up_dir), prefix, in_fps, "h264", True, video.name)
-    proc2 = subprocess.Popen(encode_cmd, stderr=subprocess.PIPE, stdout=subprocess.DEVNULL, text=True, bufsize=1, cwd=str(up_dir))
-    while True:
-        line = proc2.stderr.readline()
-        if not line and proc2.poll() is not None:
-            break
-        if int(time.time()*10) % 5 == 0:
-            prog_html = render_progress(50.0, "Phase 3/3: Encoding…")
-    proc2.wait()
-
-    out_file = Path(up_dir) / "output.mp4"
-    if not out_file.exists():
-        return None, None, None, "Encoding failed in Quick Mode.", prog_html
+#    encode_cmd = build_ffmpeg_encode(str(up_dir), prefix, in_fps, "h264", True, video.name)
+#    proc2 = subprocess.Popen(encode_cmd, stderr=subprocess.PIPE, stdout=subprocess.DEVNULL, text=True, bufsize=1, cwd=str(up_dir))
+#    while True:
+#        line = proc2.stderr.readline()
+#        if not line and proc2.poll() is not None:
+#            break
+#        if int(time.time()*10) % 5 == 0:
+#            prog_html = render_progress(50.0, "Phase 3/3: Encoding…")
+#    proc2.wait()
+
+#    out_file = Path(up_dir) / "output.mp4"
+#    if not out_file.exists():
+#        return None, None, None, "Encoding failed in Quick Mode.", prog_html
 
     # Intermediates
-    zip_frames = work / "frames.zip"
-    with zipfile.ZipFile(zip_frames, "w", zipfile.ZIP_DEFLATED) as zf:
-        for p in frames:
-            zf.write(p, p.name)
-    zip_up = work / "upscaled.zip"
-    with zipfile.ZipFile(zip_up, "w", zipfile.ZIP_DEFLATED) as zf:
-        for p in sorted(up_dir.glob("*.jpg"), key=_natural_key):
-            zf.write(p, p.name)
+#    zip_frames = work / "frames.zip"
+#    with zipfile.ZipFile(zip_frames, "w", zipfile.ZIP_DEFLATED) as zf:
+#        for p in frames:
+#            zf.write(p, p.name)
+#    zip_up = work / "upscaled.zip"
+#    with zipfile.ZipFile(zip_up, "w", zipfile.ZIP_DEFLATED) as zf:
+#        for p in sorted(up_dir.glob("*.jpg"), key=_natural_key):
+#            zf.write(p, p.name)
 
-    return str(out_file), str(zip_frames), str(zip_up), "Quick Mode complete.", render_progress(100.0, "All done")
+#    return str(out_file), str(zip_frames), str(zip_up), "Quick Mode complete.", render_progress(100.0, "All done")
 
 # ───────────────── UI
 
@@ -952,10 +972,9 @@ def build_ui():
     .cf-title { font-size: 1.6rem; font-weight: 800; }
     .cmdbox textarea { font-family: ui-monospace, Menlo, monospace; font-size: 12px; }
     """) as demo:
-        gr.Markdown("""
-        <div class=\"cf-title\">Video → Frames → Upscale → Re-encode</div>
-        Three-step workflow + Quick Mode. Step 2/3 now accept your own uploaded files as inputs.
-        """)
+        gr.HTML(render_logo_html(96))
+        gr.Markdown("Three-step workflow. Video → Frames → Upscale → Re-encode")
+
 
         # Shared states (from Step 1)
         frames_state = gr.State([])        # list[str]