import gradio as gr
from transformers import pipeline

pipeline = pipeline(task="text-classification", model="Preetham04/sentiment-analysis")

def predict(input_text):
    predictions = pipeline(input_text)
    print(predictions)
    result = [] 
    for p in predictions:
        review = "POSITIVE" if p["label"] == "LABEL_1" else "NEGATIVE"
        result.append({
            "SENTIMENT": review,
            "SCORE": p["score"]
        })
    return result

description = """
The Bot was trained to answer questions based on Clothing. Ask anything!
"""

gradio_app = gr.Interface(
    predict,
    inputs="textbox",
    outputs="text",
    title="Sentiment- good or bad?",
    description=description,
    examples=[["Loved it!!"], ["Bad quality"]],
)

if __name__ == "__main__":
    gradio_app.launch(share=True)