backend/core/deployment_generator.py

class DeploymentGenerator:
    def generate_fastapi_app(self, model_path):
        template = f'''

from fastapi import FastAPI

import joblib

import pandas as pd



app = FastAPI()

model = joblib.load("{model_path}")



@app.post("/predict")

async def predict(data: dict):

    df = pd.DataFrame([data])

    preds = model.predict(df)

    return {{"prediction": preds.tolist()}}

'''
        return template

    def generate_dockerfile(self):
        return '''

FROM python:3.9

WORKDIR /app

COPY . /app

RUN pip install fastapi uvicorn pandas scikit-learn joblib

CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]

'''