from fastapi import FastAPI, Request
from rag_core.business import answer_query, rescan_index
from ui import app_ui
import gradio as gr
import logging

app = FastAPI()

@app.post("/ask")
async def ask_api(req: Request):
    data = await req.json()
    query = data.get("query", "")
    return answer_query(query)

@app.post("/rescan")
async def rescan_api():
    return rescan_index()

# Mount Gradio UI vào FastAPI tại root
app = gr.mount_gradio_app(app, app_ui, path="")

logging.info("✅ Gradio UI đã mount vào root /")