src/verseflow/api/server.py

"""
FastAPI REST 网关与 Prometheus 监控接口
Author: XiaoZhe (Commercial Contact: janejulius119@gmail.com / WeChat: julius119)
"""

from fastapi import FastAPI, BackgroundTasks
from pydantic import BaseModel
import uvicorn
import time

app = FastAPI(title="VerseFlow Studio v5.0 API", version="5.0.0")
METRICS = {"total_requests": 0, "successful_tasks": 0}

class GenerationRequest(BaseModel):
    prompt: str
    scene_type: str = "product_demo"
    quality_tier: str = "standard"

@app.post("/api/v5/video/generate")
async def generate_video(req: GenerationRequest, background_tasks: BackgroundTasks):
    METRICS["total_requests"] += 1
    task_id = f"vf_task_{int(time.time() * 1000)}"
    return {"code": 200, "task_id": task_id, "author": "XiaoZhe (janejulius119@gmail.com / WeChat: julius119)"}

@app.get("/metrics")
async def metrics():
    return METRICS

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8000)