import os import sys from contextlib import asynccontextmanager from pathlib import Path # ─── 插件目录(必须在 import app.* 之前执行)──────────────────────────── # 桌面端(PyInstaller 冻结)读不到系统 site-packages。把用户插件目录加进 # sys.path,让用户可以自装可选依赖(如 mlx_whisper): # python3.11 -m pip install --target "<插件目录>" mlx_whisper # PyInstaller 的 FrozenImporter 优先于 sys.path,内置包不会被插件目录覆盖, # 插件目录只补「包里没有」的模块。安装后重启应用生效。 if getattr(sys, "frozen", False): _plugin_dir = os.path.join( os.getenv("APPDATA") or str(Path.home()), "VideoMemo", "python-packages" ) os.makedirs(_plugin_dir, exist_ok=True) if _plugin_dir not in sys.path: sys.path.insert(0, _plugin_dir) import uvicorn from fastapi import FastAPI from starlette.middleware.cors import CORSMiddleware from starlette.middleware.gzip import GZipMiddleware from starlette.staticfiles import StaticFiles from dotenv import load_dotenv from app.db.init_db import init_db from app.db.provider_dao import seed_default_providers from app.exceptions.exception_handlers import register_exception_handlers # from app.db.model_dao import init_model_table # from app.db.provider_dao import init_provider_table from app.utils.logger import get_logger from app.utils.path_helper import get_runtime_dir from app import create_app from app.services.transcriber_config_manager import TranscriberConfigManager from app.services.scheduler import get_scheduler from events import register_handler from ffmpeg_helper import ensure_ffmpeg_or_raise logger = get_logger(__name__) load_dotenv() # 读取 .env 中的路径 static_path = os.getenv('STATIC', '/static') # 静态资源根目录用 get_runtime_dir:开发/Docker 维持 "./static",PyInstaller 打包后切到 # exe 同级目录。挂载目录必须和 note.py / video_helper.py 的写入目录同源——Tauri sidecar 的 # cwd 不是项目目录,之前用相对 "static" 会让两者错位,导致桌面端正文截图和侧边栏封面全 404。 static_dir = get_runtime_dir("static") uploads_dir = get_runtime_dir("uploads") @asynccontextmanager async def lifespan(app: FastAPI): # 启动序列拆成 5 步、每步独立日志 + 异常时打明确的 [startup N/5 FAILED] 标记。 # 目的:用户 docker logs 一眼能看出后端死在哪一步,避免「容器一直重启但看不出原因」。 try: logger.info("[startup 1/5] register_handler() — 注册事件处理器") register_handler() logger.info("[startup 2/5] init_db() — 初始化 SQLite 数据库") init_db() logger.info("[startup 3/5] TranscriberConfigManager — 读取转写器配置") # 转写器不再在启动时强制初始化,而是在首次生成笔记时按需创建。 # 如果配置了不可用的类型(如 mlx-whisper 未安装),会在使用时报错而非静默回退。 _cfg = TranscriberConfigManager().get_config() logger.info( f" 当前转写器: type={_cfg['transcriber_type']}, " f"model_size={_cfg['whisper_model_size']}" ) logger.info("[startup 4/5] seed_default_providers() — 初始化默认 LLM 供应商") seed_default_providers() logger.info("[startup 5/5] 启动完成,等待请求") get_scheduler().start() except Exception: logger.exception("[startup FAILED] 后端启动期异常,详见堆栈;容器会退出并由 restart 策略决定是否重试") raise yield get_scheduler().stop() app = create_app(lifespan=lifespan) # 允许的源:本地 web 端 + Tauri 桌面端 + 浏览器扩展(chrome/edge/firefox) # 用 regex 是因为 chrome-extension:// 的 id 在每次开发版加载时不固定 # Tauri 2 不同平台 webview origin 不一样,必须全列: # - macOS: tauri://localhost (自定义协议) # - Windows: https://tauri.localhost (Edge WebView2) # - Linux: http://tauri.localhost (WebKitGTK) # 漏掉哪个都会导致桌面端 fetch 返回 200 但 browser 因为 CORS 拒绝读响应, # 表现为前端「连不上后端」但后端日志一片 200 OK。 CORS_ORIGIN_REGEX = ( r"^chrome-extension://[a-z]+$" r"|^moz-extension://.+$" r"|^http://(localhost|127\.0\.0\.1)(:\d+)?$" r"|^tauri://localhost$" r"|^https?://tauri\.localhost$" # Cloudflare Pages:.pages.dev 及其预览子域 ..pages.dev r"|^https://([a-z0-9-]+\.)*pages\.dev$" ) app.add_middleware( CORSMiddleware, allow_origin_regex=CORS_ORIGIN_REGEX, allow_credentials=True, allow_methods=["*"], allow_headers=["*"], ) app.add_middleware(GZipMiddleware, minimum_size=1000) register_exception_handlers(app) app.mount(static_path, StaticFiles(directory=static_dir), name="static") app.mount("/uploads", StaticFiles(directory=uploads_dir), name="uploads") if __name__ == "__main__": port = int(os.getenv("BACKEND_PORT", 8483)) host = os.getenv("BACKEND_HOST", "0.0.0.0") logger.info(f"Starting server on {host}:{port}") uvicorn.run(app, host=host, port=port, reload=False)