Create-knowledge-find-decode/creation_knowledge/api.py

"""FastAPI 静态服务：托管单页前端 + 媒体/抽帧落盘目录。

启动：
    CK_ENV_FILE=/path/.env uvicorn creation_knowledge.api:app --host 0.0.0.0 --port 8123
前端单页在 /，媒体在 /data，视频帧在 /frames。
"""
from __future__ import annotations

import os
from functools import lru_cache
from pathlib import Path

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles

from creation_knowledge.config import Settings

WEB_DIR = Path(__file__).resolve().parent.parent / "web"


def _env_file() -> str:
    return os.getenv("CK_ENV_FILE", ".env")


@lru_cache(maxsize=1)
def _settings() -> Settings:
    return Settings.from_env(_env_file())


app = FastAPI(title="创作知识")
app.add_middleware(
    CORSMiddleware, allow_origins=["*"], allow_methods=["*"], allow_headers=["*"]
)

# 视频帧静态服务（card.url = /frames/<post_id>/<file>）
_frames_dir = Path(_settings().frames_dir)
_frames_dir.mkdir(parents=True, exist_ok=True)
app.mount("/frames", StaticFiles(directory=str(_frames_dir)), name="frames")

# 媒体静态服务（card.url = /data/<run_id>/<platform>/<post_id>/<file>）
_data_dir = Path(_settings().data_dir or "data")
_data_dir.mkdir(parents=True, exist_ok=True)
app.mount("/data", StaticFiles(directory=str(_data_dir)), name="data")

# 单页前端挂在最后（catch-all），不影响 /frames、/data
if WEB_DIR.exists():
    app.mount("/", StaticFiles(directory=str(WEB_DIR), html=True), name="web")