|
"""API endpoint code.""" |
|
import logging |
|
|
|
import uvicorn |
|
from fastapi import FastAPI, status |
|
from endpoints import autotrain as autotrain_router |
|
import logging |
|
import sys |
|
|
|
app = FastAPI() |
|
app.include_router(autotrain_router.router, prefix="/autotrain") |
|
|
|
logger = logging.getLogger(__name__) |
|
logger.setLevel(logging.INFO) |
|
stream_handler = logging.StreamHandler(sys.stdout) |
|
log_formatter = logging.Formatter("%(asctime)s [%(processName)s: %(process)d] [%(threadName)s: %(thread)d] [%(levelname)s] %(name)s: %(message)s") |
|
stream_handler.setFormatter(log_formatter) |
|
logger.addHandler(stream_handler) |
|
|
|
logger.info('API is starting up') |
|
|
|
@app.get("/", status_code=status.HTTP_200_OK) |
|
async def read_root(): |
|
message = f"Hello world! From FastAPI running on Uvicorn with Gunicorn" |
|
logging.info(message) |
|
return {"message": message} |
|
|
|
|
|
if __name__ == '__main__': |
|
uvicorn.run(app, host='0.0.0.0', port=7860) |
|
|