# Importar la librería necesaria de Hugging Face
from transformers import pipeline

# Cargar el modelo preentrenado de Hugging Face para detección de emociones
emotion_detector = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base")

# Función que toma el texto y devuelve la emoción detectada
def detect_emotion(text):
    return emotion_detector(text)

# Si el script se ejecuta directamente, pedimos al usuario un texto y mostramos el resultado
if __name__ == "__main__":
    text = input("Introduce el texto para detectar la emoción: ")
    result = detect_emotion(text)
    
    # Mostrar el resultado
    print(f"Emoción detectada: {result[0]['label']} con una probabilidad de {result[0]['score']:.2f}")