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

_ROOT = Path(__file__).resolve().parent.parent
WEB_DIR = _ROOT / "web"
SKILL_EX_DIR = _ROOT / "创作知识提取-skill" / "extraction"  # ②③ 提示词真源
READ_PROMPT_DIR = _ROOT / "prompts"  # ① 读懂提示词真源


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")

# 提示词真源（前端「看提示词」直接 fetch，不再写死副本，避免漂移）
if SKILL_EX_DIR.exists():
    app.mount("/prompt-skill", StaticFiles(directory=str(SKILL_EX_DIR)), name="prompt-skill")
if READ_PROMPT_DIR.exists():
    app.mount("/prompt-read", StaticFiles(directory=str(READ_PROMPT_DIR)), name="prompt-read")

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