import uvicorn
from dotenv import load_dotenv
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

from api.base_router import base_router
import os
os.environ['TRANSFORMERS_HOME'] = './.cache'
if __name__ == '__main__':
    load_dotenv()
    uvicorn.run(
        'app:app',
        reload=True,
    )


app = FastAPI(
    docs_url='/docs',
)

app.include_router(base_router)

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