fix commit

Dockerfile

@@ -0,0 +1,15 @@
+# syntax=docker/dockerfile:1
+FROM python:3.11 AS base_image
+
+WORKDIR /app
+RUN mkdir -p -m 0600 ~/.ssh && ssh-keyscan bitbucket.org >> ~/.ssh/known_hosts
+RUN --mount=type=ssh pip install -e git+ssh://git@bitbucket.org/faktionml/autotrain-advanced.git#egg=autotrain-advanced
+
+FROM base_image
+
+EXPOSE 7860
+
+COPY . .
+RUN pip install -r autotrain/requirements.txt
+
+CMD ["python", "autotrain/main.py"]

autotrain/endpoints/autotrain.py

@@ -0,0 +1,18 @@
+from fastapi import APIRouter, Request
+import yaml
+import os
+
+
+router = APIRouter()
+
+@router.post('/create_project')
+async def create_project(request: Request):
+    parameters = await request.json()
+    with open('autotrain_config.yaml', 'w+') as ff:
+        yaml.dump(parameters, ff)
+
+    os.system("autotrain --config autotrain_config.yaml")
+    
+    return {"status": "sucess", "message": "Project created successfully."}
+
+    

autotrain/main.py

@@ -0,0 +1,30 @@
+"""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)

autotrain/requirements.txt

@@ -0,0 +1,3 @@
+fastapi==0.110.2
+pyyaml==6.0.1
+uvicorn==0.29.0