1

app.py

@@ -60,7 +60,6 @@ PAPER = """
   year      = {2024}
 }
 """
-
 BADGES_HTML = """
 <div style="display:flex;gap:12px;align-items:center;flex-wrap:wrap;">
   <a href="https://github.com/yyang181/colormnet" target="_blank" title="Open GitHub Repo">
@@ -74,6 +73,51 @@ BADGES_HTML = """
 </div>
 """
 
+# ----------------- REFERENCE FRAME GUIDE (NO CROPPING) -----------------
+REF_GUIDE_MD = r"""
+## 参考帧制作指南 / Reference Frame Guide
+
+**目的 / Goal**  
+为模型提供一张与你的视频关键帧在**姿态、光照、构图**尽量接近的**彩色参考图**，用来指导整段视频的着色风格与主体颜色。
+
+---
+
+### 中文步骤
+1. **挑帧**：从视频里挑一帧（或相近角度的照片），尽量与要着色的镜头在**姿态 / 光照 / 场景**一致。  
+2. **上色方式**：若你只有黑白参考图、但需要彩色参考，可用 **通义千问·图像编辑（Qwen-Image）**：  
+   - 打开：<https://chat.qwen.ai/> → 选择**图像编辑**  
+   - 上传你的黑白参考图  
+   - 在提示词里输入：  
+     **「帮我给这张照片上色，只修改颜色，不要修改内容」**  
+   - 可按需多次编辑（如补充「衣服为复古蓝、肤色自然、不要锐化」）  
+3. **保存格式**：PNG/JPG 均可；推荐分辨率 ≥ **480px**（短边）。  
+4. **文件放置**：本应用会自动放置为 `ref/<视频名>/ref.png`。  
+
+**注意事项（Do/Don’t）**  
+- ✅ 主体清晰、颜色干净，不要过曝或强滤镜。  
+- ✅ 关键区域（衣服、皮肤、头发、天空等）颜色与目标风格一致。  
+- ❌ 不要更改几何结构（如人脸形状/姿态），**只修改颜色**。  
+- ❌ 避免文字、贴纸、重度风格化滤镜。
+
+---
+
+### English Steps
+1. **Pick a frame** (or a similar photo) that matches the target shot in **pose / lighting / composition**.  
+2. **Colorizing if your reference is B&W** — use **Qwen-Image (Image Editing)**:  
+   - Open <https://chat.qwen.ai/> → **Image Editing**  
+   - Upload your B&W reference  
+   - Prompt: **“Help me colorize this photo; only change colors, do not alter the content.”**  
+   - Iterate if needed (e.g., “vintage blue jacket, natural skin tone; avoid sharpening”).  
+3. **Format**: PNG/JPG; recommended short side ≥ **480px**.  
+4. **File placement**: The app will place it as `ref/<video_stem>/ref.png`.
+
+**Do / Don’t**
+- ✅ Clean subject and palette; avoid overexposure/harsh filters.  
+- ✅ Ensure key regions (clothes/skin/hair/sky) match the intended colors.  
+- ❌ Do not change geometry/structure — **colors only**.  
+- ❌ Avoid text/stickers/heavy stylization filters.
+"""
+
 # ----------------- TEMP WORKDIR -----------------
 TEMP_ROOT = path.join(os.getcwd(), "_colormnet_tmp")
 INPUT_DIR = "input_video"
@@ -298,7 +342,7 @@ def gradio_infer(
 
     # 8) 同进程调用 test.py
     try:
-        import test  # 确保 test.py 同目录且有 run_cli 函数
+        import test  # 确保 test.py 同目录且提供 run_cli(args_list)
     except Exception as e:
         return None, f"导入 test.py 失败 / Failed to import test.py：\n{e}"
 
@@ -321,15 +365,11 @@ def gradio_infer(
         log = f"Args: {' '.join(args_list)}\n\n{buf.getvalue()}\n\nERROR: {e}"
         return None, log
 
-    # 在合成 mp4 之前：清空 CUDA
+    # 在合成 mp4 之前：清空 CUDA（防止显存占用）
     try:
         torch.cuda.synchronize()
     except Exception:
         pass
-    try:
-        del network, processor, loader, vid_reader, data, rgb, msk, prob
-    except Exception:
-        pass
     try:
         torch.cuda.empty_cache()
     except Exception:
@@ -339,7 +379,7 @@ def gradio_infer(
     out_frames = path.join(output_root, video_stem)
     if not path.isdir(out_frames):
         return None, f"未找到输出帧目录 / Output frame dir not found：{out_frames}\n\n{log}"
-    final_mp4 = path.join(TEMP_ROOT, f"{video_stem}.mp4")
+    final_mp4 = path.abspath(path.join(TEMP_ROOT, f"{video_stem}.mp4"))
     try:
         encode_frames_to_video(out_frames, final_mp4, fps=fps)
     except Exception as e:
@@ -350,10 +390,14 @@ def gradio_infer(
 # ----------------- UI -----------------
 with gr.Blocks() as demo:
     gr.Markdown(f"# {TITLE}")
-    gr.HTML(BADGES_HTML)     # 页头徽章
+    gr.HTML(BADGES_HTML)
     gr.Markdown(PAPER)
     gr.Markdown(DESC)
 
+    # 参考帧制作指南（中英双语，无裁剪步骤）
+    with gr.Accordion("参考帧制作指南 / Reference Frame Guide", open=False):
+        gr.Markdown(REF_GUIDE_MD)
+
     debug_shapes = gr.Checkbox(label="调试日志 / Debug Logs（仅用于显示更完整日志 / show verbose logs）", value=False)
 
     with gr.Row():
@@ -390,7 +434,7 @@ with gr.Blocks() as demo:
 
     run_btn = gr.Button("开始着色 / Start Coloring（同进程调用 test.py / in-process）")
     with gr.Row():
-        out_video = gr.Video(label="输出视频（着色结果） / Output (Colorized)")
+        out_video = gr.Video(label="输出视频（着色结果） / Output (Colorized)", autoplay=True)
         status = gr.Textbox(label="状态 / 日志输出（test.py stdout/stderr） / Status & Logs", interactive=False, lines=16)
 
     run_btn.click(
@@ -406,7 +450,6 @@ with gr.Blocks() as demo:
         outputs=[out_video, status]
     )
 
-    # 页脚徽章
     gr.HTML("<hr/>")
     gr.HTML(BADGES_HTML)