Update app.py

app.py

@@ -3,23 +3,34 @@ from dotenv import load_dotenv
 from huggingface_hub import InferenceClient
 import gradio as gr
 
-# تحميل المتغيرات من .env (هذا السطر ممكن ما يكون ضروري في Spaces، لكن خليه احتياطًا)
+# تحميل المتغيرات من .env
 load_dotenv()
 
 # استدعاء مفتاح API من متغير البيئة
 api_key = os.getenv("FIREWORKS_API_KEY")
 
-# تهيئة عميل InferenceClient مع اسم المزود ومفتاح API
+# تهيئة عميل InferenceClient
 client = InferenceClient(
     provider="fireworks-ai",
-    api_key=api_key  # استخدام مفتاح API من متغير البيئة
+    api_key=api_key
 )
 
+# رسالة النظام لتوجيه سلوك النموذج
+system_message = {
+    "role": "system",
+    "content": "أنت نموذج لغوي متخصص في مجال الأشعة الطبية وتفسير الصور الشعاعية. يجب أن تكون إجاباتك دقيقة ومبنية على المعرفة الطبية المتعلقة بالأشعة. تجنب الإجابة على أي أسئلة لا تتعلق بهذا المجال."
+}
+
 def chat_fn(message, history):
     try:
+        # بناء قائمة الرسائل: رسالة النظام + رسالة المستخدم الحالية
+        messages = [
+            system_message,
+            {"role": "user", "content": message}
+        ]
         completion = client.chat.completions.create(
             model="deepseek-ai/DeepSeek-R1",
-            messages=[{"role": "user", "content": message}],
+            messages=messages,
             max_tokens=512,
         )
         return completion.choices[0].message.content
@@ -28,8 +39,8 @@ def chat_fn(message, history):
 
 chat_interface = gr.ChatInterface(
     fn=chat_fn,
-    title="RAY AI Chat with DeepSeek (Fireworks AI)",
-    description="نموذج DeepSeek R1 عبر مزود Fireworks AI.",
+    title="RAY AI Chat with DeepSeek (Fireworks AI) - متخصص في الأشعة",
+    description="نموذج DeepSeek R1 عبر مزود Fireworks AI - متخصص في مجال الأشعة الطبية وتفسير الصور الشعاعية.",
 )
 
 chat_interface.launch()