import gradio as gr
from transformers import pipeline

# تحميل نموذج دردشة قوي مفتوح المصدر
chatbot = pipeline("text-generation", model="tiiuae/falcon-7b-instruct", device=-1)

# دالة الرد على المستخدم
def chat_with_ai(user_input):
    prompt = f"User: {user_input}\nAI:"
    response = chatbot(prompt, max_new_tokens=200, do_sample=True, temperature=0.7)[0]["generated_text"]
    
    # تنظيف الرد ليظهر بشكل مرتب
    if "AI:" in response:
        response = response.split("AI:")[-1].strip()
    return response

# واجهة المستخدم
interface = gr.Interface(
    fn=chat_with_ai,
    inputs=gr.Textbox(lines=3, placeholder="اكتب سؤالك هنا بأي لغة...", label="🧠 سؤالك"),
    outputs=gr.Textbox(label="🤖 رد الذكاء الاصطناعي"),
    title="💬 مساعد الذكاء الاصطناعي - حسين محمد",
    description="مشروع ذكاء اصطناعي تجريبي يعمل باللغة العربية والإنجليزية باستخدام نموذج مفتوح المصدر. مطور بواسطة حسين محمد.",
    theme="soft",
    examples=[
        "ما هي عاصمة العراق؟",
        "اشرح لي مفهوم الذكاء الاصطناعي",
        "What is machine learning?",
        "اكتب لي مقدمة لمقال عن الطاقة الشمسية"
    ]
)

interface.launch()