import gradio as gr
from transformers import pipeline
import spaces 

# Charger le pipeline de classification des sentiments
sentiment_analyzer = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")

@spaces.GPU 

def analyze_sentiment(text):
    # Analyser le sentiment du texte
    result = sentiment_analyzer(text)
    # Retourner le résultat sous forme de texte
    sentiment = result[0]['label']
    score = result[0]['score']
    return f"Sentiment: {sentiment}, Score: {score:.2f}"

# Créer l'interface Gradio
interface = gr.Interface(
    fn=analyze_sentiment,
    inputs="text",
    outputs="text",
    title="Analyse des Sentiments",
    description="Entrez un texte pour analyser son sentiment (positif, négatif ou neutre)."
)

# Lancer l'interface dans un Space Hugging Face
if __name__ == "__main__":
    interface.launch()