import os
import gradio as gr
import requests
import spaces 

# Récupérer le token depuis les secrets
api_token = os.environ.get("HF_API_TOKEN")

API_URL = "https://api-inference.huggingface.co/models/distilbert-base-uncased-finetuned-sst-2-english"
headers = {"Authorization": f"Bearer {api_token}"}
@spaces.GPU

# Texte à analyser
# Envoi de la requête

def query(payload):
    response = requests.post(API_URL, headers=headers, json=payload)
    return response.json()
def analyze_sentiment(input_text):
    output = query({"inputs": input_text})
    label = output[0]['label']
    if label == 'POSITIVE':
        return 'Positif'
    else:
        return 'Négatif'
        
interface = gr.Interface(
    fn=analyze_sentiment, 
    inputs="text", 
    outputs="text",
    title="Analyse de Sentiment",
    description="Entrez un texte pour savoir s'il est positif ou négatif."
)

# Lancer l'interface
interface.launch()

iface.launch()