Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,46 +6,39 @@ import torch
 # =======================
 # RTL & UI Styling
 # =======================
-
 st.markdown(
     """
     <style>
-    /* General page direction */
     html, body, [class*="css"] {
         direction: rtl !important;
         text-align: right !important;
         font-family: 'Arial', sans-serif !important;
     }
 
-    /* Title and headers */
     .stApp header, h1, h2, h3, h4, h5, h6 {
         direction: rtl !important;
         text-align: right !important;
         font-family: 'Arial', sans-serif !important;
     }
 
-    /* Markdown text */
     .stMarkdown, .stMarkdown p {
         direction: rtl !important;
         text-align: right !important;
         font-family: 'Arial', sans-serif !important;
     }
 
-    /* Input fields and buttons */
     .stTextArea, .stTextInput, .stButton {
         direction: rtl !important;
         text-align: right !important;
         font-family: 'Arial', sans-serif !important;
     }
 
-    /* Warning, Error, and Success messages */
-    [data-testid="stAlert"] {
+    [data-testid="stAlert"], [data-testid="stSpinnerText"] {
         direction: rtl !important;
         text-align: right !important;
         font-family: 'Arial', sans-serif !important;
     }
 
-    /* Sidebar  */
     section[data-testid="stSidebar"] {
         direction: rtl !important;
         text-align: right !important;
@@ -71,7 +64,7 @@ def classify_question(text):
     return LABELS[pred]
 
 # =======================
-# LLM double-check
+# LLM first-aid check
 # =======================
 def llm_first_aid_check(query, api_key):
     url = "https://openrouter.ai/api/v1/chat/completions"
@@ -100,7 +93,7 @@ def llm_first_aid_check(query, api_key):
         return False
 
 # =======================
-# Mistral response with explicit few-shot
+# Mistral Response 
 # =======================
 def get_mistral_response(prompt, api_key):
     url = "https://openrouter.ai/api/v1/chat/completions"
@@ -109,34 +102,31 @@ def get_mistral_response(prompt, api_key):
         "Content-Type": "application/json"
     }
 
-    # Explicit few-shot prompt text
-    few_shot_text = """
-أنت مساعد طبي محترف متخصص في الإسعافات الأولية. أجب دائمًا بنصائح إسعاف أولية واضحة ودقيقة وآمنة، باللغة العربية فقط، واتبِع أسلوب جميع الأمثلة التالية:
+    few_shot_examples = """
+أنت مساعد طبي محترف متخصص في الإسعافات الأولية. أجب دائمًا بنصائح إسعاف أولية واضحة ودقيقة وآمنة، باللغة العربية فقط.
 
 سؤال 1: أصابتني ضربة على عظمة حاجبي الأيسر وحدوث انتفاخ بسيط. وضعت كمادات باردة. ماذا ينبغي أن أفعل أيضًا؟
 جواب 1: استمر بوضع كمادات باردة لفترات قصيرة (10–15 دقيقة كل ساعة) لتخفيف الورم. ارفع رأسك عند الاستلقاء لتقليل التورم، وتجنّب الضغط على المنطقة. راقب أي تغيّر في الرؤية، فقدان للوعي، أو تقيؤ — إذا ظهرت أي من هذه الأعراض فراجع الطوارئ فورًا.
 
-سؤال 2: حدث نزيف داخل المخ ثم تلا ذلك ضعف في الساق اليسرى وخرف وهلوسة. هل هذا زهايمر أم شيء آخر؟
-جواب 2: هذه أعراض خطيرة تتطلب تقييمًا عصبيًا عاجلًا. حالات النزيف والجلطة بالمخ قد تسبب ضعفًا موضعي واضطرابًا إدراكيًا؛ لذلك يلزم تصوير مقطعي أو رنين مغناطيسي وتقييم من أخصائي أعصاب لتحديد السبب وخطة العلاج. تواصل مع مقدم الرعاية الطارئة أو قسم الأعصاب فورًا.
-
-سؤال 3: كيف أتعامل مع شخص يتعرض لنوبة صرع؟
-جواب 3: أبعد حوله الأشياء الحادة وحافظ على الهدوء. ضع شيئًا ناعمًا تحت رأسه، لا تُقيد حركاته، ولا تضع شيئًا في فمه. بعد انتهاء النوبة، ضع الشخص على جنبه للحفاظ على مجرى الهواء وراقبه حتى يستعيد وعيه. إذا استمرت النوبة أكثر من 5 دقائق أو تكررت، اطلب الإسعاف فورًا.
+سؤال 2: كيف أتعامل مع شخص يتعرض لنوبة صرع؟
+جواب 2: أبعد حوله الأشياء الحادة وحافظ على الهدوء. ضع شيئًا ناعمًا تحت رأسه، لا تُقيد حركاته، ولا تضع شيئًا في فمه. بعد انتهاء النوبة، ضع الشخص على جنبه للحفاظ على مجرى الهواء وراقبه حتى يستعيد وعيه. إذا استمرت النوبة أكثر من 5 دقائق أو تكررت، اطلب الإسعاف فورًا.
+"""
 
-سؤال 4: السلام عليكم، كيف أعرف ما إذا كان جرح الرأس يحتاج خياطة، علماً أنه لا يوجد نزيف الآن؟
-جواب 4: إن كانت الحافة واسعة أو مفتوحة بعمق أو كان الحجم أكبر من 1 سم أو كان الجرح في وجهٍ يؤثر على الشكل، فقد يحتاج للخياطة ويُفضل عرض المصاب على الطبيب. نظف الجرح بمحلول فيزيولوجي، غطّه بشاش معقم، وتجنب إعادة الخياطة بعد مرور أكثر من 12 ساعة دون استشارة طبية. إذا ظهرت علامات التهاب أو عدم التئام، راجع الطبيب.
+    user_instruction = f"""
+سؤال: {prompt}
 
-سؤال 5: {user_query}
-جواب 5:
-""".format(user_query=prompt)
+اكتب الجواب بشكل مختصر ومباشر دون تكرار أو إعادة صياغة.
+الجواب:
+"""
 
     data = {
         "model": "mistralai/mistral-7b-instruct",
         "messages": [
-            {"role": "system", "content": "أنت مساعد طبي محترف متخصص في الإسعافات الأولية."},
-            {"role": "user", "content": few_shot_text}
+            {"role": "system", "content": few_shot_examples},
+            {"role": "user", "content": user_instruction}
         ],
-        "temperature": 0.2,  # more consistent answers
-        "max_tokens": 768
+        "temperature": 0.3,
+        "max_tokens": 512
     }
 
     response = requests.post(url, headers=headers, json=data)
@@ -159,7 +149,7 @@ if st.button("اسأل"):
     elif not user_input.strip():
         st.warning("من فضلك اكتب سؤالك")
     else:
-        with st.spinner("جار تحليل سؤالك..."):
+        with st.spinner("جارٍ تحليل سؤالك..."):
             category = classify_question(user_input)
 
         if category == "LABEL_1":