from fastapi import FastAPI
from pydantic import BaseModel
from transformers import pipeline
import uvicorn

app = FastAPI()

# Carga el modelo de Hugging Face para el análisis de emociones
emotion_model = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base")

class TextInput(BaseModel):
    text: str

@app.post("/predict/")
def predict_emotion(input_data: TextInput):
    result = emotion_model(input_data.text)
    return {"emotion": result[0]['label'], "score": result[0]['score']}

if __name__ == "__main__":
    # Ejecuta el servidor Uvicorn en el puerto 8000
    uvicorn.run(app, host="0.0.0.0", port=8000)