from fastapi import FastAPI
from pydantic import BaseModel
import joblib

app = FastAPI()

# Load your model at startup
model = joblib.load("app/emotion_classifier_pipe_lr.pkl")

class Request(BaseModel):
    text: str

@app.post("/predict")
async def predict_emotion(request: Request):
    prediction = model.predict([request.text])[0]
    return {"emotion": prediction}