Update app.py

app.py

@@ -3,6 +3,15 @@ import torch
 import gradio as gr
 import spaces
 from threading import Thread
+from huggingface_hub import login
+import os
+
+# ======================================================
+# تسجيل الدخول في حال كان الموديل خاص (اختياري)
+# ======================================================
+if os.getenv("HF_TOKEN"):
+    login(token=os.getenv("HF_TOKEN"))
+    print("🔐 Logged in to Hugging Face Hub")
 
 # ======================================================
 # إعدادات الموديل
@@ -51,7 +60,6 @@ print("✅ Model ready!\n")
 # دالة المحادثة
 # ======================================================
 def chat(message, history):
-    # تحويل تاريخ المحادثة إلى صيغة messages
     messages = []
     for msg in history:
         if msg["role"] == "user":
@@ -59,17 +67,14 @@ def chat(message, history):
         elif msg["role"] == "assistant":
             messages.append({"role": "assistant", "content": msg["content"]})
 
-    # نضيف السؤال الحالي مع system prompt
     messages.append({"role": "user", "content": f"{SYSTEM_PROMPT}\n\nالسؤال: {message}"})
 
-    # تجهيز الإدخال عبر chat template
     input_ids = tokenizer.apply_chat_template(
         messages,
         return_tensors="pt",
         add_generation_prompt=True
     ).to(model.device)
 
-    # Streamer للبث الحي للنص الناتج
     streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
 
     generation_kwargs = dict(
@@ -82,7 +87,6 @@ def chat(message, history):
         repetition_penalty=1.15,
     )
 
-    # تشغيل التوليد في Thread منفصل للبث المباشر
     thread = Thread(target=model.generate, kwargs=generation_kwargs)
     thread.start()
 
@@ -100,18 +104,18 @@ def chat(message, history):
 demo = gr.ChatInterface(
     fn=chat,
     type="messages",
-    title="🇮🇶 Gemma 3 4B – Iraqi Arabic Chat",
+    title="🇮🇶 Gemma 3 Iraqi Chat",
     description="""
-**نموذج Gemma 3 4B Instruct (Unsloth Optimized)**  
-مدرّب على العربية والفصحى ومهيّأ للحوار باللهجة العراقية 🇮🇶
+**نموذج Gemma 3 4B – باللهجة العراقية 🇮🇶**  
+مدرّب ومهيّأ للحوار بالعربية والفصحى، يفهم السياق ويتكلم بوضوح.
 
-🧠 أمثلة جاهزة:
-- "شنو معنى إنكار معقول؟"
+🧠 أمثلة:
+- "شلونك اليوم؟"
 - "اشرحلي شنو يعني تصعيد إداري"
 - "وضحلي الفرق بين الدليل والعبرة"
 """,
     examples=[
-        ["شنو معنى إنكار معقول؟"],
+        ["شلونك اليوم؟"],
         ["اشرحلي شنو يعني تصعيد إداري"],
         ["وضحلي الفرق بين الدليل والعبرة"],
     ],