import gradio as gr
from transformers import pipeline
from bertopic import BERTopic

topic_model = BERTopic.load("TarekAi/ArBERTopic")

topic, prob = topic_model.transform("عدم إنجاز الواجبات المدرسية")

examples=["عدم الالتزام بالتشريع المنظم للحياة المدرسية من طرف الاستاذ", "عدم إنجاز الواجبات المدرسية"]

io = gr.Interface(fn=text_classification, 
                         inputs= gr.Textbox(lines=2, label="Text", placeholder="Enter your idea here..."), 
                         outputs=gr.Textbox(lines=2, label="Text Classification Result"),
                         title="Text Classification",
                         description="Enter a text and see the text classification result!",
                         examples=examples)

io.launch()