Update app.py

app.py

@@ -546,23 +546,112 @@ def parse_script_nodes(script_text: str) -> List[str]:
 
 
 def extract_last_frame(video_path: str, output_path: str) -> Optional[str]:
-    """提取视频最后一帧作为参考图""" 
+    """提取视频最后一帧（OpenCV优化版 - 精简）"""
+    import time
+    
     if cv2 is None:
+        LOGGER.warning("OpenCV 不可用")
         return None
     
-    cap = cv2.VideoCapture(video_path)
-    if not cap.isOpened():
+    if not os.path.exists(video_path):
+        LOGGER.error(f"视频文件不存在: {video_path}")
         return None
     
-    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-    cap.set(cv2.CAP_PROP_POS_FRAMES, max(total_frames - 1, 0))
-    ret, frame = cap.read()
-    cap.release()
+    # === 步骤1: 等待文件写入稳定 ===
+    max_wait = 30
+    check_interval = 1.0
+    stable_count = 0
+    required_stable = 3
+    last_size = 0
     
-    if ret:
-        cv2.imwrite(output_path, frame)
+    LOGGER.info("⏳ 等待文件写入完成...")
+    for i in range(int(max_wait / check_interval)):
+        try:
+            current_size = os.path.getsize(video_path)
+        except OSError:
+            time.sleep(check_interval)
+            continue
+        
+        if current_size == 0:
+            time.sleep(check_interval)
+            continue
+        
+        if current_size == last_size:
+            stable_count += 1
+            if stable_count >= required_stable:
+                LOGGER.info(f"✅ 文件稳定: {current_size / 1024 / 1024:.2f} MB")
+                break
+        else:
+            stable_count = 0
+        
+        last_size = current_size
+        time.sleep(check_interval)
+    
+    # 额外等待确保文件系统同步
+    time.sleep(2.0)
+    
+    # === 步骤2: OpenCV 读取视频 ===
+    capture = cv2.VideoCapture(video_path)
+    if not capture.isOpened():
+        LOGGER.error("OpenCV 无法打开视频")
+        return None
+    
+    try:
+        total_frames = int(capture.get(cv2.CAP_PROP_FRAME_COUNT) or 0)
+        fps = capture.get(cv2.CAP_PROP_FPS) or 30
+        
+        if total_frames <= 0:
+            LOGGER.error("视频帧数为 0")
+            return None
+        
+        LOGGER.info(f"📹 视频信息: {total_frames} 帧, {fps:.2f} FPS")
+        
+        # === 步骤3: 多候选帧策略（避免黑帧/损坏帧）===
+        candidates = [
+            total_frames - 1,              # 最后一帧
+            total_frames - 2,              # 倒数第2帧
+            total_frames - 5,              # 倒数第5帧
+            max(0, int(total_frames * 0.95))  # 95%位置
+        ]
+        
+        frame = None
+        used_index = -1
+        
+        for candidate_idx in candidates:
+            candidate_idx = max(0, min(candidate_idx, total_frames - 1))
+            capture.set(cv2.CAP_PROP_POS_FRAMES, candidate_idx)
+            success, temp_frame = capture.read()
+            
+            if success and temp_frame is not None and temp_frame.size > 0:
+                # 检查亮度（排除黑屏）
+                gray = cv2.cvtColor(temp_frame, cv2.COLOR_BGR2GRAY)
+                brightness = gray.mean()
+                
+                if brightness > 5:  # 亮度阈值
+                    frame = temp_frame
+                    used_index = candidate_idx
+                    LOGGER.info(f"✅ 提取第 {used_index}/{total_frames} 帧（亮度: {brightness:.1f}）")
+                    break
+        
+        # === 步骤4: 保存图片 ===
+        if frame is None:
+            LOGGER.error("所有候选帧均无效")
+            return None
+        
+        os.makedirs(os.path.dirname(output_path), exist_ok=True)
+        if not cv2.imwrite(output_path, frame):
+            LOGGER.error("保存图片失败")
+            return None
+        
+        file_size = os.path.getsize(output_path)
+        LOGGER.info(f"💾 参考帧已保存: {os.path.basename(output_path)} ({file_size / 1024:.1f} KB)")
         return output_path
-    return None
+        
+    except Exception as e:
+        LOGGER.error(f"提取帧时出错: {e}")
+        return None
+    finally:
+        capture.release()
 
 
 def stitch_videos(video_paths: List[str], output_path: str):
@@ -685,7 +774,7 @@ def run_video_generation_pipeline(
         if i < len(nodes) - 1:
             frame_path = os.path.join(output_dir, f"ref_{idx:02d}.png")
             last_frame_path = extract_last_frame(video_filename, frame_path)
-        
+
         yield generated_videos, None, f"✅ 分镜 {idx} 完成"
 
     # 拼接视频
@@ -727,39 +816,6 @@ def get_demo_path(filename):
     return filename if os.path.exists(filename) else None
 
 
-# # 构建 Gradio 界面
-# with gr.Blocks(title="AI 剧本视频工厂") as demo:
-#     gr.Markdown("# 🎬 ScriptAgent & Sora/Veo 视频生成工坊")
-    
-#     with gr.Tabs():
-#         # --- TAB 1: 剧本创作 ---
-#         with gr.Tab("📝 第一步：剧本创作"):
-#             with gr.Row():
-#                 with gr.Column():
-#                     llm_input = gr.Textbox(
-#                         label="剧情输入", 
-#                         placeholder="主角：你在做什么？...", 
-#                         lines=6
-#                     )
-#                     llm_btn = gr.Button("生成/续写剧本", variant="primary")
-                
-#                 with gr.Column():
-#                     llm_output = gr.Textbox(
-#                         label="生成的剧本", 
-#                         lines=10, 
-#                         interactive=True
-#                     )
-#                     to_video_btn = gr.Button("⬇️ 发送到视频生成", variant="secondary")
-            
-#             gr.Examples(
-#                 [[
-#                     "主角：你在做什么？指引图是不能随便篡改的。\n"
-#                     "蒋前：篡改指引图？不不不，你误会了。\n"
-#                     "蒋前：我才加入风物家没多久，哪有这个本事能篡改它..."
-#                 ]], 
-#                 inputs=llm_input
-#             )
-
 with gr.Blocks(title="AI 剧本视频工厂") as demo:
     gr.Markdown("# 🎬 ScriptAgent & Sora/Veo 视频生成工坊 ")
     
@@ -1024,4 +1080,4 @@ print(response.choices[0].message.content)'''
 
 if __name__ == "__main__":
     demo.queue()
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)