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"]
'''