Spaces:
Sleeping
Sleeping
File size: 1,846 Bytes
0c8f7e3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import asyncio
from contextlib import asynccontextmanager
from fastapi import FastAPI, APIRouter
import sys
from loguru import logger
from api_server.auth_middleware import auth_middleware
from api_server.v1_utils_router import v1_utils_router
from api_server.v1_media_router import v1_media_api_router
from video.config import device
logger.remove()
logger.add(
sys.stdout,
colorize=True,
format="<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level> | <blue>{extra}</blue>",
level="DEBUG",
)
logger.info("This server was created by the 'AI Agents A-Z' YouTube channel")
logger.info("https://www.youtube.com/@aiagentsaz")
logger.info("Using device: {}", device)
@asynccontextmanager
async def lifespan(app: FastAPI):
logger.info("Starting up the server...")
yield
logger.info("Shutting down the server...")
app = FastAPI(lifespan=lifespan)
# add middleware to app, besides the /health endpoint
app.middleware("http")(auth_middleware)
@app.api_route("/", methods=["GET", "HEAD"])
def root():
return {
"message": "Welcome to the AI Agents A-Z No-Code Server",
"version": "0.3.5",
"documentation": "/docs",
"created_by": "https://www.youtube.com/@aiagentsaz"
}
@app.api_route("/health", methods=["GET", "HEAD"])
def healthcheck():
return {"status": "ok"}
api_router = APIRouter()
v1_api_router = APIRouter()
# todo auto-delete files after 30 minutes (env var)
v1_api_router.include_router(v1_media_api_router, prefix="/media", tags=["media"])
v1_api_router.include_router(v1_utils_router, prefix="/utils", tags=["utils"])
api_router.include_router(v1_api_router, prefix="/v1", tags=["v1"])
app.include_router(api_router, prefix="/api", tags=["api"])
|