import gradio as gr

# Description of the dataset and the model
description = """
The Persian Legal Question Answering System (PLQAS) is a machine learning model that can answer legal questions in Persian. 

It was trained on a dataset of legal questions https://huggingface.co/datasets/Kamtera/Persian-conversational-dataset .

The model is  https://huggingface.co/Aminrhmni/PersianLegalQAsystem.

You can check the accuracy of the answers at https://www.dadrah.ir/


سیستم پاسخگویی به سوالات حقوقی فارسی (PLQAS) یک مدل یادگیری ماشینی است که می تواند به سوالات حقوقی به زبان فارسی پاسخ دهد.

بر روی مجموعه داده ای از سوالات حقوقی https://huggingface.co/datasets/Kamtera/Persian-conversational-dataset آموزش داده شد.

مدل https://huggingface.co/Aminrhmni/PersianLegalQAsystem است.

می توانید صحت پاسخ ها را در https://www.dadrah.ir/ بررسی کنید.
"""

# Example test input
example_input = [
    "آیا کارفرما میتواند کارگر دائمی را اخراج کند؟",
    "آیا بازنشستگی آموزگار زن با 25 سال سابقه و 43 سال سن امکان پذیر است؟",
    'شخصی با مزاحمت تلفنی و پیامکی به بنده و خانمم که خواستگار قدیمیش بوده جمله ای توهین آمیز و زشت برای بنده ارسال کردند با این موضوع که خانمت رو میبری برا فلانی و مردم فلان کش بدبخت. چند باری هم تماس داشتن به محل کار خانمم که اونجا تخریب کنه خانومم رو. بنده روی توهین و و چه مواردی میتونم شکایت کنم؟',
    ' با سلام بعد تقاطع ماشینی از پشت به ماشینم برخورد کرد و متواری شد 110 اومدصورت جلسه کرد گفت اگه دوربین امنیتی نباشه میشه پیگیری کرد؟',
    ' ایا جوی آب شش دانگی که از باغی رد میشود وبه زمینهای دیگه مرتبط هس،مالک میتواند با کشیدن فنس جوی رو داخل باغ خود تصرف کند وحق ورود رو ندهد ،ایا ماده ای هس ؟',
    """سلام و عرض ادب
بنده چند روز پیش برای فردی در تلگرام عکس کارت ملی و شناسنامه و یه عکس سه در چهار ارسال کردم و بعد فهمیدم کلاهبرداره. چت رو دو طرفه پاک کردم. ایا امکان سو استفاده وجود داره؟ چه مشکلی ممکنه پیش بیاد؟ چجوری میتونم پیشگیری کنم؟
ممنون از شما""",
]

# Create the Gradio interface
demo = gr.Interface(
    fn=gr.load("models/Aminrhmni/PersianLegalQAsystem"),
    inputs="text",
    outputs="text",
    title="Persian Legal Question Answering System",
    description=description,
    examples=example_input,
    # live=False
)


# Launch the Gradio interface
demo.launch(share=False)