import gradio as gr
from transformers import pipeline
from langdetect import detect

model_name = "aubmindlab/aragpt2-base"
chatbot = pipeline("text-generation", model=model_name)

def reply(prompt):
    if prompt.strip() == "":
        return "يرجى إدخال نص."
    try:
        lang = detect(prompt)
    except:
        lang = "ar"
    response = chatbot(prompt, max_length=100, do_sample=True)[0]["generated_text"]
    return response.strip()

interface = gr.Interface(
    fn=reply,
    inputs=gr.Textbox(label="📝 أدخل نصًا بأي لغة"),
    outputs=gr.Textbox(label="💬 رد الوكيل"),
    title="🌍 TunisAI-Agent — ذكاء صناعي متعدد اللغات",
    description="وكيل ذكاء صناعي مجاني يعمل بأي لغة 🧠🌐"
)

interface.launch(share=True)