From 43206c261e28d8dee1f87f670638f73bf45a9f2c Mon Sep 17 00:00:00 2001 From: root Date: Tue, 17 Feb 2026 17:53:07 +0800 Subject: [PATCH] =?UTF-8?q?feat(startup):=20=E5=90=AF=E5=8A=A8=E6=97=B6?= =?UTF-8?q?=E9=A2=84=E4=B8=8B=E8=BD=BDffmpeg=E8=B5=84=E6=BA=90=E5=88=B0?= =?UTF-8?q?=E6=9C=AC=E5=9C=B0=E9=9D=99=E6=80=81=E7=9B=AE=E5=BD=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/static/public/pages/video.html | 6 +- main.py | 115 ++++++++++++++++++++++++++++- 2 files changed, 115 insertions(+), 6 deletions(-) diff --git a/app/static/public/pages/video.html b/app/static/public/pages/video.html index a97d575..712746b 100644 --- a/app/static/public/pages/video.html +++ b/app/static/public/pages/video.html @@ -256,9 +256,9 @@ - - - + + + diff --git a/main.py b/main.py index c30905b..adf1e64 100644 --- a/main.py +++ b/main.py @@ -5,12 +5,16 @@ FastAPI 应用初始化和路由注册 """ from contextlib import asynccontextmanager +import argparse import os import platform import sys from pathlib import Path +from typing import Dict, List +import asyncio from dotenv import load_dotenv +import httpx BASE_DIR = Path(__file__).resolve().parent APP_DIR = BASE_DIR / "app" @@ -34,6 +38,7 @@ from app.core.exceptions import register_exception_handlers # noqa: E402 from app.core.response_middleware import ResponseLoggerMiddleware # noqa: E402 from app.api.v1.chat import router as chat_router # noqa: E402 from app.api.v1.image import router as image_router # noqa: E402 +from app.api.v1.nsfw import router as nsfw_router # noqa: E402 from app.api.v1.files import router as files_router # noqa: E402 from app.api.v1.models import router as models_router # noqa: E402 from app.services.token import get_scheduler # noqa: E402 @@ -48,6 +53,70 @@ setup_logging( ) +FFMPEG_VENDOR_DIR = APP_DIR / "static" / "vendor" / "ffmpeg" +FFMPEG_VENDOR_ASSETS: Dict[str, List[str]] = { + "ffmpeg.js": [ + "https://cdn.jsdelivr.net/npm/@ffmpeg/ffmpeg@0.12.10/dist/umd/ffmpeg.js", + "https://unpkg.com/@ffmpeg/ffmpeg@0.12.10/dist/umd/ffmpeg.js", + ], + "ffmpeg-util.js": [ + "https://cdn.jsdelivr.net/npm/@ffmpeg/util@0.12.1/dist/umd/index.js", + "https://unpkg.com/@ffmpeg/util@0.12.1/dist/umd/index.js", + ], + "ffmpeg-core.js": [ + "https://cdn.jsdelivr.net/npm/@ffmpeg/core@0.12.6/dist/umd/ffmpeg-core.js", + "https://unpkg.com/@ffmpeg/core@0.12.6/dist/umd/ffmpeg-core.js", + ], + "ffmpeg-core.wasm": [ + "https://cdn.jsdelivr.net/npm/@ffmpeg/core@0.12.6/dist/umd/ffmpeg-core.wasm", + "https://unpkg.com/@ffmpeg/core@0.12.6/dist/umd/ffmpeg-core.wasm", + ], +} + + +async def _download_first_success( + client: httpx.AsyncClient, filename: str, urls: List[str], target_path: Path +) -> bool: + for url in urls: + try: + resp = await client.get(url, follow_redirects=True) + if resp.status_code == 200 and resp.content: + target_path.write_bytes(resp.content) + logger.info( + f"FFmpeg vendor downloaded: {filename} <- {url} ({len(resp.content)} bytes)" + ) + return True + logger.warning( + f"FFmpeg vendor candidate failed: {filename} <- {url} status={resp.status_code}" + ) + except Exception as e: + logger.warning(f"FFmpeg vendor download error: {filename} <- {url}, error={e}") + return False + + +async def ensure_ffmpeg_vendor_assets() -> None: + """启动时预热 ffmpeg 前端依赖到本地静态目录，避免浏览器跨域/CORS 问题。""" + FFMPEG_VENDOR_DIR.mkdir(parents=True, exist_ok=True) + timeout = httpx.Timeout(connect=8.0, read=30.0, write=30.0, pool=8.0) + async with httpx.AsyncClient(timeout=timeout) as client: + tasks = [] + pending_names: list[str] = [] + for filename, urls in FFMPEG_VENDOR_ASSETS.items(): + target = FFMPEG_VENDOR_DIR / filename + if target.exists() and target.stat().st_size > 0: + logger.info(f"FFmpeg vendor exists: {filename}") + continue + pending_names.append(filename) + tasks.append(_download_first_success(client, filename, urls, target)) + + if not tasks: + return + results = await asyncio.gather(*tasks, return_exceptions=False) + failed = [name for name, ok in zip(pending_names, results) if not ok] + if failed: + logger.warning(f"FFmpeg vendor partial ready, failed={failed}") + + @asynccontextmanager async def lifespan(app: FastAPI): """应用生命周期管理""" @@ -64,6 +133,7 @@ async def lifespan(app: FastAPI): logger.info("Starting Grok2API...") logger.info(f"Platform: {platform.system()} {platform.release()}") logger.info(f"Python: {sys.version.split()[0]}") + await ensure_ffmpeg_vendor_assets() # 4. 启动 Token 刷新调度器 refresh_enabled = get_config("token.auto_refresh", True) @@ -119,6 +189,9 @@ def create_app() -> FastAPI: app.include_router( image_router, prefix="/v1", dependencies=[Depends(verify_api_key)] ) + app.include_router( + nsfw_router, prefix="/v1", dependencies=[Depends(verify_api_key)] + ) app.include_router( models_router, prefix="/v1", dependencies=[Depends(verify_api_key)] ) @@ -143,9 +216,36 @@ app = create_app() if __name__ == "__main__": import uvicorn - host = os.getenv("SERVER_HOST", "0.0.0.0") - port = int(os.getenv("SERVER_PORT", "8000")) - workers = int(os.getenv("SERVER_WORKERS", "1")) + parser = argparse.ArgumentParser(description="启动 Grok2API 服务") + parser.add_argument( + "--host", + default=os.getenv("SERVER_HOST", "0.0.0.0"), + help="监听地址，默认读取 SERVER_HOST 或 0.0.0.0", + ) + parser.add_argument( + "--port", + type=int, + default=int(os.getenv("SERVER_PORT", "8000")), + help="监听端口，默认读取 SERVER_PORT 或 8000", + ) + parser.add_argument( + "--workers", + type=int, + default=int(os.getenv("SERVER_WORKERS", "1")), + help="Worker 数量，默认读取 SERVER_WORKERS 或 1", + ) + parser.add_argument( + "--reload", + action="store_true", + help="开启热重载（开发模式）", + ) + + args = parser.parse_args() + + host = args.host + port = args.port + workers = args.workers + reload_enabled = args.reload # 平台检查 is_windows = platform.system() == "Windows" @@ -158,10 +258,19 @@ if __name__ == "__main__": ) workers = 1 + # uvicorn 不支持 reload 与多 worker 同时启用 + if reload_enabled and workers > 1: + logger.warning( + f"Reload mode detected. Multiple workers ({workers}) is not supported. " + "Using single worker instead." + ) + workers = 1 + uvicorn.run( "main:app", host=host, port=port, workers=workers, + reload=reload_enabled, log_level=os.getenv("LOG_LEVEL", "INFO").lower(), )