import gradio as gr
from transformers import pipeline
import spaces 

@spaces.GPU 

# 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."

# Interface de chatbot utilisant Gradio
gr.Interface(fn=generate_response, inputs="text", outputs="text").launch()