Spaces:
Sleeping
Sleeping
import logging | |
import os | |
import uvicorn | |
from contextlib import asynccontextmanager | |
from fastapi import FastAPI | |
from fastapi.middleware.cors import CORSMiddleware | |
from fastapi.responses import RedirectResponse | |
from app.api.routers.chat import chat_router | |
from app.api.routers.user import user_router | |
from app.api.routers.transaction import transaction_router | |
from app.api.routers.file_upload import file_upload_router | |
from app.api.routers.income_statement import income_statement_router | |
from app.settings import init_settings | |
from fastapi.staticfiles import StaticFiles | |
from alembic.config import Config | |
from alembic import command | |
from app.engine.postgresdb import postgresdb | |
from config.index import config as env | |
logger = logging.getLogger("uvicorn") | |
init_settings() | |
# init_observability() | |
def init_app(init_db: bool = True) -> FastAPI: | |
lifespan = None | |
if init_db: | |
postgresdb.init(env.SQLALCHEMY_DATABASE_URL, {"echo": True, "future": True}) | |
async def lifespan(app: FastAPI): | |
yield | |
if postgresdb.get_engine() is not None: | |
await postgresdb.close() | |
app = FastAPI(lifespan=lifespan) | |
if env.ENVIRONMENT == "dev": | |
logger.warning("Running in development mode - allowing CORS for all origins") | |
app.add_middleware( | |
CORSMiddleware, | |
allow_origins=["*"], | |
allow_credentials=True, | |
allow_methods=["*"], | |
allow_headers=["*"], | |
) | |
if os.path.exists("data"): | |
app.mount("/api/data", StaticFiles(directory="data"), name="static") | |
app.include_router(chat_router, prefix="/api/chat") | |
app.include_router(user_router) | |
app.include_router(transaction_router) | |
app.include_router(file_upload_router) | |
app.include_router(income_statement_router) | |
return app | |
app = init_app() | |
# Redirect to documentation page when accessing base URL | |
async def redirect_to_docs(): | |
return RedirectResponse(url="/docs") | |
if __name__ == "__main__": | |
uvicorn.run(app="main:app", host=env.APP_HOST, port=env.APP_PORT, reload=(env.ENVIRONMENT == "dev")) | |