import uvicorn
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from gradio import mount_gradio_app

from src.api.routes import router as api_router
from src.ui.components import create_ui

app = FastAPI(title="Code LLM Service")

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_methods=["*"],
    allow_headers=["*"],
)

# Подключаем API эндпоинты OpenAI-типа
app.include_router(api_router, prefix="/v1")

# Подключаем Gradio интерфейс
ui_app = create_ui()
app = mount_gradio_app(app, ui_app, path="/")

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=7860)