import spaces
import transformers
from transformers import pipeline

# Chargement du modèle de classification de sentiment

classifier = pipeline("sentiment-analysis")

# Fonction pour prédire le sentiment d'un post

def predict_sentiment(post):
    result = classifier(post)
    sentiment = result[0]['label']
    return sentiment


# Fonction de génération de texte basée sur le sentiment

def generate_response(prompt):
    sentiment = predict_sentiment(prompt)
    if sentiment == 'POSITIVE':
        return "Je suis content de discuter avec vous!"
    elif sentiment == 'NEGATIVE':
        return "Je suis désolé que vous ressentiez cela. Comment puis-je vous aider?"
    else:
        return "Merci pour vos commentaires."