feat(bilibili-analyzer): enhance prepare script with robust tool detection and internationalization

- Add UTF-8 encoding support for Windows console output
- Implement find_yt_dlp() function to detect yt-dlp as command or Python module
- Implement find_ffmpeg() function with fallback to common Windows installation paths
- Replace hardcoded tool commands with dynamic detection and helpful error messages
- Translate all user-facing strings from Chinese to English for broader accessibility
- Add improved error handling with detailed installation instructions for missing dependencies
- Add progress indicators and formatted output with visual separators
- Add -y flag to ffmpeg to automatically overwrite existing files
- Improve exception handling to catch and report unexpected errors gracefully
- Enhance help text and examples with clearer descriptions
- Add UTF-8 encoding declaration at file header

Author: 239573049

skills/tools/bilibili-analyzer/scripts/prepare.py

@@ -1,132 +1,216 @@
 #!/usr/bin/env python3
+# -*- coding: utf-8 -*-
 """
 Bilibili Video Downloader and Frame Extractor
-下载B站视频并拆解成帧图片
+Download Bilibili videos and extract frames
 """
 
 import os
 import sys
 import subprocess
 import argparse
+import shutil
+
+# Fix Windows console encoding
+if sys.platform == "win32":
+    sys.stdout.reconfigure(encoding='utf-8', errors='replace')
+    sys.stderr.reconfigure(encoding='utf-8', errors='replace')
+
+
+def find_yt_dlp():
+    """Find yt-dlp executable or module
+
+    Returns:
+        list: Command to run yt-dlp, or None if not found
+    """
+    # Try direct command first
+    if shutil.which("yt-dlp"):
+        return ["yt-dlp"]
+
+    # Try as Python module
+    try:
+        result = subprocess.run(
+            [sys.executable, "-m", "yt_dlp", "--version"],
+            capture_output=True,
+            timeout=10
+        )
+        if result.returncode == 0:
+            return [sys.executable, "-m", "yt_dlp"]
+    except Exception:
+        pass
+
+    return None
+
+
+def find_ffmpeg():
+    """Find ffmpeg executable
+
+    Returns:
+        str: Path to ffmpeg, or None if not found
+    """
+    if shutil.which("ffmpeg"):
+        return "ffmpeg"
+
+    # Common Windows paths
+    common_paths = [
+        r"C:\ffmpeg\bin\ffmpeg.exe",
+        r"C:\Program Files\ffmpeg\bin\ffmpeg.exe",
+        r"C:\tools\ffmpeg\bin\ffmpeg.exe",
+    ]
+    for path in common_paths:
+        if os.path.exists(path):
+            return path
+
+    return None
 
 
 def download_video(url: str, output_path: str = "video.mp4") -> bool:
-    """下载B站视频
+    """Download Bilibili video
 
     Args:
-        url: B站视频URL
-        output_path: 输出文件路径
+        url: Bilibili video URL
+        output_path: Output file path
 
     Returns:
-        是否下载成功
+        Whether download succeeded
     """
-    print(f"[INFO] 正在下载视频: {url}")
+    print(f"[INFO] Downloading video: {url}")
 
-    cmd = [
-        "yt-dlp",
+    yt_dlp_cmd = find_yt_dlp()
+    if not yt_dlp_cmd:
+        print("[ERROR] yt-dlp not found!")
+        print("        Install with: pip install yt-dlp")
+        print(f"        Current Python: {sys.executable}")
+        return False
+
+    cmd = yt_dlp_cmd + [
         "-f", "bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best",
         "-o", output_path,
+        "--no-warnings",
         url
     ]
 
     try:
+        print(f"[INFO] Running: {' '.join(cmd[:3])}...")
         result = subprocess.run(cmd, check=True)
-        print(f"[OK] 视频下载完成: {output_path}")
+        print(f"[OK] Video downloaded: {output_path}")
         return True
     except subprocess.CalledProcessError as e:
-        print(f"[ERROR] 下载失败: {e}")
+        print(f"[ERROR] Download failed: {e}")
         return False
-    except FileNotFoundError:
-        print("[ERROR] yt-dlp 未安装，请运行: pip install yt-dlp")
+    except Exception as e:
+        print(f"[ERROR] Unexpected error: {e}")
         return False
 
 
 def extract_frames(video_path: str, output_dir: str = "./images", fps: float = 1.0) -> bool:
-    """从视频提取帧图片
+    """Extract frames from video
 
     Args:
-        video_path: 视频文件路径
-        output_dir: 输出目录
-        fps: 每秒提取帧数，默认1帧/秒
+        video_path: Video file path
+        output_dir: Output directory
+        fps: Frames per second, default 1
 
     Returns:
-        是否提取成功
+        Whether extraction succeeded
     """
-    print(f"[INFO] 正在提取帧图片 (fps={fps})")
+    print(f"[INFO] Extracting frames (fps={fps})")
+
+    ffmpeg_cmd = find_ffmpeg()
+    if not ffmpeg_cmd:
+        print("[ERROR] ffmpeg not found!")
+        print("        Windows: choco install ffmpeg / scoop install ffmpeg")
+        print("        macOS: brew install ffmpeg")
+        print("        Linux: sudo apt install ffmpeg")
+        return False
 
-    # 创建输出目录
+    # Create output directory
     os.makedirs(output_dir, exist_ok=True)
 
     output_pattern = os.path.join(output_dir, "frame_%04d.jpg")
 
     cmd = [
-        "ffmpeg",
+        ffmpeg_cmd,
         "-i", video_path,
         "-vf", f"fps={fps}",
         "-q:v", "2",
+        "-y",  # Overwrite existing files
         output_pattern
     ]
 
     try:
+        print(f"[INFO] Running ffmpeg...")
         result = subprocess.run(cmd, check=True, capture_output=True)
 
-        # 统计生成的图片数量
+        # Count generated images
         frame_count = len([f for f in os.listdir(output_dir) if f.startswith("frame_") and f.endswith(".jpg")])
-        print(f"[OK] 帧提取完成: {frame_count} 张图片保存到 {output_dir}/")
+        print(f"[OK] Frames extracted: {frame_count} images saved to {output_dir}/")
         return True
     except subprocess.CalledProcessError as e:
-        print(f"[ERROR] 帧提取失败: {e.stderr.decode() if e.stderr else e}")
+        stderr = e.stderr.decode('utf-8', errors='replace') if e.stderr else str(e)
+        print(f"[ERROR] Frame extraction failed: {stderr}")
         return False
-    except FileNotFoundError:
-        print("[ERROR] ffmpeg 未安装，请安装 ffmpeg")
+    except Exception as e:
+        print(f"[ERROR] Unexpected error: {e}")
         return False
 
 
 def main():
     parser = argparse.ArgumentParser(
-        description="下载B站视频并拆解成帧图片",
+        description="Download Bilibili video and extract frames",
         formatter_class=argparse.RawDescriptionHelpFormatter,
         epilog="""
-示例:
+Examples:
   python prepare.py "https://www.bilibili.com/video/BV1xx411c7mD"
   python prepare.py "https://www.bilibili.com/video/BV1xx411c7mD" --fps 0.5
   python prepare.py "https://www.bilibili.com/video/BV1xx411c7mD" -o ./output
         """
     )
 
-    parser.add_argument("url", help="B站视频URL")
-    parser.add_argument("-o", "--output", default=".", help="输出目录，默认当前目录")
-    parser.add_argument("--fps", type=float, default=1.0, help="每秒提取帧数，默认1")
-    parser.add_argument("--video-only", action="store_true", help="只下载视频，不提取帧")
-    parser.add_argument("--frames-only", action="store_true", help="只提取帧（需要已有video.mp4）")
+    parser.add_argument("url", help="Bilibili video URL")
+    parser.add_argument("-o", "--output", default=".", help="Output directory (default: current)")
+    parser.add_argument("--fps", type=float, default=1.0, help="Frames per second (default: 1)")
+    parser.add_argument("--video-only", action="store_true", help="Only download video, skip frame extraction")
+    parser.add_argument("--frames-only", action="store_true", help="Only extract frames (requires existing video.mp4)")
 
     args = parser.parse_args()
 
-    # 设置路径
+    # Set paths
     output_dir = args.output
     video_path = os.path.join(output_dir, "video.mp4")
     images_dir = os.path.join(output_dir, "images")
 
-    # 创建输出目录
+    # Create output directory
     os.makedirs(output_dir, exist_ok=True)
 
-    # 下载视频
+    print("=" * 50)
+    print("Bilibili Video Analyzer - Prepare Script")
+    print("=" * 50)
+    print(f"URL: {args.url}")
+    print(f"Output: {output_dir}")
+    print(f"FPS: {args.fps}")
+    print("=" * 50)
+
+    # Download video
     if not args.frames_only:
         if not download_video(args.url, video_path):
             sys.exit(1)
 
-    # 提取帧
+    # Extract frames
     if not args.video_only:
         if not os.path.exists(video_path):
-            print(f"[ERROR] 视频文件不存在: {video_path}")
+            print(f"[ERROR] Video file not found: {video_path}")
             sys.exit(1)
 
         if not extract_frames(video_path, images_dir, args.fps):
             sys.exit(1)
 
-    print("\n[OK] 完成!")
-    print(f"    视频: {video_path}")
-    print(f"    图片: {images_dir}/")
+    print("")
+    print("=" * 50)
+    print("[OK] Done!")
+    print(f"  Video: {video_path}")
+    print(f"  Images: {images_dir}/")
+    print("=" * 50)
 
 
 if __name__ == "__main__":