"""
API路由 - 纯净版RAG系统
保留RAG核心接口，为bbav20311接入预留接口
"""

from fastapi import APIRouter, UploadFile, File, HTTPException
from pathlib import Path
import os

from api.rag_routes import router as rag_router
from api.case_routes import router as case_router

router = APIRouter()

# ============================================
# RAG核心接口（已实现）
# ============================================

# RAG对话接口
router.include_router(rag_router, prefix="/rag", tags=["RAG"])

# 真实复现案例问答接口（确定性QA，不调用LLM）
router.include_router(case_router, prefix="/case", tags=["Case QA"])

# ============================================
# 预留接口（为bbav20311接入准备）
# ============================================

# 文件上传接口（预留）
@router.post("/upload")
async def upload_file(file: UploadFile = File(...)):
    """文件上传接口 - 预留"""
    raise HTTPException(status_code=501, detail="待bbav20311接入后实现")

# 批量上传接口（预留）
@router.post("/upload/batch")
async def upload_batch(files: list[UploadFile] = File(...)):
    """批量上传接口 - 预留"""
    raise HTTPException(status_code=501, detail="待bbav20311接入后实现")

# 工作空间接口（预留）
@router.get("/workspace/{job_id}/files")
async def get_workspace_files(job_id: str):
    """工作空间文件接口 - 预留"""
    raise HTTPException(status_code=501, detail="待bbav20311接入后实现")

# 基因家族分析接口（预留）
@router.post("/analysis/gene-family")
async def analyze_gene_family(job_id: str, species: str, gene_family: str):
    """基因家族分析接口 - 预留"""
    raise HTTPException(status_code=501, detail="待bbav20311接入后实现")

# 工具执行接口（预留）
@router.post("/tools/{tool_id}/execute")
async def execute_tool(tool_id: str, params: dict):
    """工具执行接口 - 预留"""
    raise HTTPException(status_code=501, detail="待bbav20311接入后实现")

# ============================================
# 状态查询接口
# ============================================

@router.get("/status")
async def get_status():
    """获取RAG系统状态"""
    try:
        from src.rag.retriever import retriever
        indexed_count = retriever.count_chunks() if retriever else 0
    except Exception as e:
        # 如果retriever未初始化，使用RAG路由的状态
        indexed_count = 0
    
    return {
        "status": "ready",
        "indexed_documents": indexed_count,
        "rag_enabled": True,
        "analysis_enabled": False,  # 待接入bbav20311后开启
    }