from fastapi import FastAPI
from pydantic import BaseModel
from src.infer import predict

app = FastAPI(title="Expon Emotions Service")

class InputText(BaseModel):
    transcript: str

@app.post("/analysis/emotions")
def analyze_emotions(item: InputText):
    label, prob, dist = predict(item.transcript)
    return {
        "dominant_emotion": label,
        "probability": prob,
        "distribution": dist
    }