"""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)