import logging.config
from pathlib import Path

import uvicorn
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from logging_env import LOGGER_CONFIG_DICT
from src.routers import deep_agent_chat
from src.routers import generate_3D
from src.routers import flux2_gen_img
from src.routers import seg_furniture
from src.routers import canvas_assistant

logging.config.dictConfig(LOGGER_CONFIG_DICT)

app_server = FastAPI(
    title="Gemini Furniture Designer API",
    description="基于 LangGraph + Gemini 2.0 Flash 的家具设计 Agent 接口",
    version="1.0.0"
)

# 配置跨域，方便前端调用
app_server.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_methods=["*"],
    allow_headers=["*"],
)

# 包含路由
app_server.include_router(deep_agent_chat.router)
app_server.include_router(generate_3D.router)
app_server.include_router(flux2_gen_img.router)
app_server.include_router(seg_furniture.router)
app_server.include_router(canvas_assistant.router)


@app_server.get("/")
async def root():
    return {"message": "Furniture Design Agent API is running."}


if __name__ == "__main__":
    uvicorn.run("main:app_server", host="0.0.0.0", port=7777, reload=False)