import gradio as gr
from setfit import AbsaModel

# Load the ABSA model
model = AbsaModel.from_pretrained(
    "pahri/setfit-indo-restomix-aspect",
    "pahri/setfit-indo-restomix-polarity",
    spacy_model="id_core_news_trf",
)

def analyze_text(text):
    """
    Analyzes the input text using the ABSA model and returns aspects and sentiment.
    Args:
        text: The text to be analyzed.
    Returns:
        A formatted string containing aspects and sentiment.
    """
    pred = model.predict(text)
    return pred

title = "Analisa Review Restoran Anda"
examples = [["Penyajiannya super aesthetic & instagramable.. Tempatnya juga keren, bikin betah & pengen foto-foto terus. Pelayanan ramah, parkir lumayan memadai. Recommended"]]
description = "Masukkan review restoran Anda untuk menganalisa aspek dan sentimen."

interface = gr.Interface(
    fn=analyze_text,
    inputs=gr.Textbox(lines=2, placeholder="Masukkan review restoran Anda di sini..."),
    outputs=gr.Text(),
    description=description,
    title=title,
    examples=examples
)

interface.launch()