from contextlib import asynccontextmanager
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware


@asynccontextmanager
async def lifespan(app):
    from TextGen.router import startup
    await startup()
    yield


app = FastAPI(title="Deploying FastAPI Apps on Huggingface", lifespan=lifespan)
app.add_middleware(
    CORSMiddleware,
    allow_origins=["https://jofthomas-everchanging-quest.static.hf.space"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

from TextGen import router