45 lines
1.2 KiB
Python
Executable File
45 lines
1.2 KiB
Python
Executable File
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)
|