Initial commit: lesson-highlights generator

src/core/constants.py

@@ -0,0 +1,130 @@
+# -*- coding: utf-8 -*-
+"""
+环境配置和常量定义
+
+所有硬编码路径和配置集中管理，确保可维护性
+"""
+
+import os
+import sys
+import shutil
+
+# ============== 项目根目录 ==============
+def _get_project_root():
+    """获取项目根目录"""
+    # Assume this file is in src/core/, so project root is 2 levels up
+    return os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+PROJECT_ROOT = _get_project_root()
+
+# ============== FFmpeg路径（动态检测）==============
+def _find_ffmpeg():
+    """动态查找FFmpeg路径"""
+    # 1. Check local ffmpeg folder (for portable distribution)
+    #    Direct: ffmpeg/bin/ffmpeg.exe
+    local_ffmpeg = os.path.join(PROJECT_ROOT, "ffmpeg", "bin")
+    if os.path.exists(os.path.join(local_ffmpeg, "ffmpeg.exe")):
+        return local_ffmpeg
+
+    #    Nested: ffmpeg/ffmpeg-8.1-full_build/bin/ffmpeg.exe
+    ffmpeg_base = os.path.join(PROJECT_ROOT, "ffmpeg")
+    if os.path.isdir(ffmpeg_base):
+        for subdir in os.listdir(ffmpeg_base):
+            nested_path = os.path.join(ffmpeg_base, subdir, "bin")
+            if os.path.exists(os.path.join(nested_path, "ffmpeg.exe")):
+                return nested_path
+
+    # 2. Check system PATH
+    system_ffmpeg = shutil.which("ffmpeg")
+    if system_ffmpeg:
+        return os.path.dirname(system_ffmpeg)
+
+    # 3. Return empty - will fail later with clear error
+    return ""
+
+FFMPEG_BIN = _find_ffmpeg()
+FFMPEG_CMD = os.path.join(FFMPEG_BIN, "ffmpeg.exe") if FFMPEG_BIN else "ffmpeg"
+FFPROBE_CMD = os.path.join(FFMPEG_BIN, "ffprobe.exe") if FFMPEG_BIN else "ffprobe"
+
+# ============== Whisper模型（动态检测）==============
+def _find_whisper_model():
+    """查找Whisper模型路径"""
+    # 1. Check local whisper_models folder
+    local_models = os.path.join(PROJECT_ROOT, "whisper_models")
+    if os.path.exists(local_models):
+        # Find any model folder
+        for name in os.listdir(local_models):
+            model_path = os.path.join(local_models, name)
+            if os.path.isdir(model_path):
+                return model_path.replace('\\', '/')
+
+    # 2. Check environment variable
+    env_path = os.environ.get("WHISPER_MODEL_PATH")
+    if env_path and os.path.exists(env_path):
+        return env_path.replace('\\', '/')
+
+    # 3. Default path (for this user's environment)
+    default_path = r"D:/AI/LM-Models/faster-whisper/large-v3"
+    if os.path.exists(default_path):
+        return default_path.replace('\\', '/')
+
+    return ""
+
+WHISPER_MODEL_PATH = _find_whisper_model()
+
+# ============== 输出目录默认值 ==============
+DEFAULT_OUTPUT_DIR = os.path.join(PROJECT_ROOT, "output")
+
+# ============== 视频参数默认值 ==============
+DEFAULT_VIDEO_PARAMS = {
+    "fade_duration": 1,
+    "title_duration": 3,
+    "title_fontsize": 90,
+    "title_color": "FFFF00",
+    "subtitle_fontsize": 24,
+    "subtitle_color": "FFFFFF",
+    "use_fast_whisper": True,
+    "whisper_model": "large",
+}
+
+# ============== API配置 ==============
+DEFAULT_API_HOST = "https://ark.cn-beijing.volces.com/api/coding/v3"
+API_KEY_ENVS = ["VOLCENGINE_API_KEY", "MINIMAX_API_KEY"]
+
+# ============== LLM配置 ==============
+LLM_MODEL = "doubao-seed-2.0-lite"
+LLM_TIMEOUT = 60
+LLM_MAX_RETRIES = 3
+LLM_TITLE_TIMEOUT = 60
+LLM_VALIDATE_TIMEOUT = 30
+
+# ============== 字幕样式 ==============
+SUBTITLE_STYLE = "FontName=微软雅黑,FontSize=24,PrimaryColour=&H00FFFFFF"
+
+# ============== Windows控制台编码 ==============
+def setup_windows_encoding():
+    """Windows控制台UTF-8编码设置"""
+    if sys.platform == 'win32':
+        import io
+        sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8', errors='replace')
+        sys.stderr = io.TextIOWrapper(sys.stderr.buffer, encoding='utf-8', errors='replace')
+
+# ============== FFmpeg PATH设置 ==============
+def setup_ffmpeg_path():
+    """将FFmpeg添加到PATH（仅当使用本地FFmpeg时）"""
+    if FFMPEG_BIN and FFMPEG_BIN not in os.environ.get('PATH', ''):
+        os.environ['PATH'] = FFMPEG_BIN + os.pathsep + os.environ.get('PATH', '')
+
+# ============== 环境初始化 ==============
+def init_environment():
+    """初始化环境设置"""
+    setup_windows_encoding()
+    setup_ffmpeg_path()
+
+def get_api_key():
+    """获取API密钥"""
+    for env_name in API_KEY_ENVS:
+        key = os.environ.get(env_name)
+        if key:
+            return key
+    return None