from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from app.api.routes import api_router
from app.core.config import settings
from app.utils.logger import logger
from app.services import milvus_service

# 创建应用
app = FastAPI(
    title="车辆法规智能检索系统",
    description="基于RAG技术的法规检索与合规分析后端API",
    version="1.0.0",
)

# CORS配置
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 注册路由
app.include_router(api_router, prefix="/api")


@app.on_event("startup")
async def startup_event():
    """启动时初始化"""
    logger.info("Starting application...")

    # 初始化Milvus集合（仅在服务可用时）
    try:
        if milvus_service is not None:
            milvus_service.create_regulations_collection()
            logger.info("Milvus collection initialized")
        else:
            logger.warning("Milvus service not available, using mock data")
    except Exception as e:
        logger.warning(f"Milvus initialization failed: {e}")


@app.on_event("shutdown")
async def shutdown_event():
    """关闭时清理"""
    logger.info("Shutting down application...")
    try:
        if milvus_service is not None:
            milvus_service.disconnect()
    except Exception as e:
        logger.warning(f"Shutdown cleanup error: {e}")


@app.get("/")
async def root():
    """根路径"""
    return {
        "message": "车辆法规智能检索系统 API",
        "version": "1.0.0",
        "docs": "/docs",
    }


@app.get("/health")
async def health():
    """健康检查"""
    return {"status": "healthy"}