init

backend/app/main.py

@@ -0,0 +1,64 @@
+import os
+import sys
+import logging
+from contextlib import asynccontextmanager
+from fastapi import FastAPI
+from fastapi.staticfiles import StaticFiles
+from fastapi.middleware.cors import CORSMiddleware
+
+from app.config import settings, BASE_DIR
+from app.database import init_db
+from app.api import files, extraction, stories, matching, export, settings as settings_api
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+
+def _get_frontend_dir() -> str:
+    """获取前端静态文件目录（支持 PyInstaller 打包）"""
+    if getattr(sys, 'frozen', False):
+        # 打包后：exe 同级目录下的 frontend/dist
+        return os.path.join(BASE_DIR, "frontend", "dist")
+    else:
+        # 开发模式：项目根目录下的 frontend/dist
+        return os.path.join(os.path.dirname(BASE_DIR), "frontend", "dist")
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """应用生命周期：启动时初始化数据库"""
+    logger.info("初始化数据库...")
+    await init_db()
+    logger.info("数据库初始化完成")
+    yield
+    logger.info("应用关闭")
+
+
+app = FastAPI(
+    title="Story Extractor",
+    description="直播文字稿故事提取工具",
+    version="1.0.0",
+    lifespan=lifespan,
+)
+
+# CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# API 路由
+app.include_router(files.router, prefix="/api/v1")
+app.include_router(extraction.router, prefix="/api/v1")
+app.include_router(stories.router, prefix="/api/v1")
+app.include_router(matching.router, prefix="/api/v1")
+app.include_router(export.router, prefix="/api/v1")
+app.include_router(settings_api.router, prefix="/api/v1")
+
+# 托管前端静态文件（放在最后，作为 fallback）
+frontend_dist = _get_frontend_dir()
+if os.path.exists(frontend_dist):
+    app.mount("/", StaticFiles(directory=frontend_dist, html=True), name="static")