Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import os
 
 
 model_path = "." 
@@ -21,16 +22,17 @@ def apply_clinical_rules(text, predicted_label, confidence):
     text = text.lower()
     
 
-    ssd_keywords = ['حرف', 'صوت', 'ينطق', 'بدل', 'لثغة', 'الراء', 'السين', 'كلامه مو واضح', 'مخارج']
-    
+    voice_keywords = ['مبحوح', 'بحّة', 'بحة', 'خشن', 'صوته رايح', 'مكتوم', 'انقطاع النفس']
+    if any(k in text for k in voice_keywords):
+        return 'Other/Medical (أخرى/طبية)', 0.95
 
-    fluency_keywords = ['تأتأة', 'يعلق', 'يكرر', 'تمتمة', 'طويل', 'شد']
-    
 
+    ssd_keywords = ['حرف', 'ينطق', 'بدل', 'لثغة', 'الراء', 'السين', 'كلامه مو واضح', 'مخارج', 'ابدال', 'حذف', 'تشويه']
     if any(k in text for k in ssd_keywords) and predicted_label != 'Speech Sound Disorder (مخارج الحروف)':
-        return 'Speech Sound Disorder (مخارج الحروف)', 0.85 # نرفع الثقة يدوياً
+        return 'Speech Sound Disorder (مخارج الحروف)', 0.85
         
 
+    fluency_keywords = ['تأتأة', 'يعلق', 'يكرر', 'تمتمة', 'طويل', 'شد في الكلام']
     if any(k in text for k in fluency_keywords) and predicted_label != 'Fluency Disorder (تأتأة)':
         return 'Fluency Disorder (تأتأة)', 0.90
         
@@ -49,40 +51,42 @@ def predict_speech_issue(text):
 
     ai_label = id2label[pred_id]
     
-  
+
     final_label, final_confidence = apply_clinical_rules(text, ai_label, confidence)
     
-    # التوصيات
+
     recommendations = {
         'Speech Sound Disorder (مخارج الحروف)': "ينصح بتقييم مخارج الحروف وفحص أعضاء النطق (Oral Motor Exam).",
         'Language Disorder (تأخر لغة)': "ينصح بعمل اختبار لغة شامل لتقييم اللغة الاستقبالية والتعبيرية.",
         'Social Communication/ASD (تواصل واجتماعي)': "ينصح بمراجعة طبيب نمو وسلوك وعمل تقييم تواصل اجتماعي.",
         'Fluency Disorder (تأتأة)': "ينصح بمراقبة بيئة الطفل وتقليل الضغوط التواصلية.",
         'Motor Speech Disorder (اضطراب حركي)': "يحتاج تقييم دقيق لاحتمالية وجود Apraxia أو Dysarthria.",
-        'Other/Medical (أخرى/طبية)': "يفضل مراجعة طبيب أنف وأذن وحنجرة (ENT) أو مختص سمعيات."
+        'Other/Medical (أخرى/طبية)': " هذه الحالة قد تكون طبية (صوت/سمع). ينصح فوراً بمراجعة طبيب أنف وأذن وحنجرة (ENT) لفحص الحبال الصوتية أو السمع."
     }
     
     advice = recommendations.get(final_label, "يرجى استشارة مختص.")
     
-
     correction_note = ""
     if final_label != ai_label:
-        correction_note = "<br><span style='font-size:0.8em; color:gray;'>(تم تعديل النتيجة بناءً على كلمات مفتاحية سريرية)</span>"
+        correction_note = "<br><span style='font-size:0.8em; color:gray;'>(تم تعديل النتيجة بناءً على الفحص السريري للنص)</span>"
 
     return f"""
-    <div style="direction: rtl; text-align: right; padding: 15px; background-color: #f9f9f9; border-radius: 10px; border: 1px solid #ddd;">
-        <h2 style="color: #2c3e50;">التصنيف: {final_label}</h2>
-        <p><b>درجة الثقة:</b> {final_confidence:.1%} {correction_note}</p>
-        <hr>
-        <p style="color: #e67e22; font-weight: bold;"> التوصية: {advice}</p>
+    <div style="direction: rtl; text-align: right; padding: 20px; background-color: #fff; border-radius: 12px; box-shadow: 0 4px 6px rgba(0,0,0,0.1);">
+        <h2 style="color: #2e86de; margin-bottom: 10px;">التصنيف المقترح: {final_label}</h2>
+        <p style="color: #555;"><b>درجة المطابقة:</b> {final_confidence:.1%} {correction_note}</p>
+        <div style="background-color: #ffeaa7; padding: 15px; border-radius: 8px; margin-top: 15px; border-right: 5px solid #fdcb6e;">
+            <p style="color: #d35400; font-weight: bold; margin: 0;"> التوصية المختصة:</p>
+            <p style="margin-top: 5px;">{advice}</p>
+        </div>
     </div>
     """
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown(" أداة الفرز الذكية لاضطرابات النطق - المراعية للهجات السعودية")
-    input_txt = gr.Textbox(label="وصف الحالة")
-    btn = gr.Button("تحليل", variant="primary")
-    output = gr.HTML(label="النتيجة")
+    gr.Markdown("<h1 style='text-align: center'>أداة الفرز الذكية لاضطرابات النطق - المراعية للهجات السعودية</h1>")
+    input_txt = gr.Textbox(label="وصف ا��حالة (بالعامية)", placeholder="مثال: ولدي صوته متغير ومبحوح.. أو بنتي تبدل الراء غين")
+    btn = gr.Button("تحليل الحالة", variant="primary")
+    output = gr.HTML(label="التقرير")
     btn.click(fn=predict_speech_issue, inputs=input_txt, outputs=output)
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()