Update app.py

app.py

@@ -1,19 +1,31 @@
 import gradio as gr
 from transformers import pipeline
 
-# نموذج عربي خفيف وسريع
-chatbot = pipeline("text-generation", model="asafaya/bert-base-arabic")
+# نموذج عربي مولد نصوص خفيف
+chatbot = pipeline(
+    "text-generation",
+    model="aubmindlab/aragpt2-small",
+    tokenizer="aubmindlab/aragpt2-small"
+)
 
 def chat(message, history=[]):
-    response = chatbot(message, max_length=50, do_sample=True, temperature=0.7)[0]["generated_text"]
-    history.append((message, response))
+    prompt = f"المحادثة:\nالمستخدم: {message}\nالمساعد:"
+    response = chatbot(
+        prompt,
+        max_length=80,
+        num_return_sequences=1,
+        do_sample=True,
+        temperature=0.7
+    )[0]["generated_text"]
+    reply = response.split("المساعد:")[-1].strip()
+    history.append((message, reply))
     return history, history
 
 with gr.Blocks() as demo:
     gr.ChatInterface(
         fn=chat,
         title="شات عربي مجاني",
-        description="بوت دردشة عربي بسيط وسريع",
+        description="بوت دردشة عربي مبسط يعمل عبر Hugging Face",
         theme="soft"
     )