import gradio as gr
from rag_pipeline import load_rag_chain

rag_chain = load_rag_chain()

def ask_question(query):
    result = rag_chain.invoke(query)
    return result['result']

iface = gr.Interface(
    fn=ask_question,
    inputs=gr.Textbox(lines=3, placeholder="اكتب سؤالك هنا...", label="سؤالك", rtl=True),
    outputs=gr.Textbox(label="الإجابة", rtl=True),
    title="🧠 روبوت دردشة عربي باستخدام ملفات PDF",
    description="اكتب سؤالك بالعربية (يدعم اللهجة المصرية)، وسنبحث عن الإجابة داخل الملفات التي قمت بتحميلها."
)

if __name__ == "__main__":
    iface.launch()