Update app.py

app.py

@@ -1,3 +1,20 @@
+นี่คือโค้ด Gradio ของคุณที่ผมได้ตรวจสอบและปรับปรุงให้แล้วครับ มีการแก้ไขข้อผิดพลาดเรื่อง `LABEL_MAPPINGS` ที่ซ้ำกันและเพิ่มคำอธิบายในส่วนของ Legend ให้ครบถ้วนตาม 5 ระดับ Sentiment และ "question" ครับ
+
+-----
+
+## สรุปการแก้ไขโค้ด
+
+1.  **แก้ไข `LABEL_MAPPINGS` ที่ซ้ำซ้อน:** โค้ดต้นฉบับมี `LABEL_MAPPINGS` ถูกประกาศซ้ำสองครั้ง และส่วนท้ายของ `LABEL_MAPPINGS` ยังมีรายการที่ซ้ำซ้อนและมี `code` ที่ไม่สอดคล้องกับแผน 6 Class (1 question + 5 sentiment levels) ที่เราคุยกันก่อนหน้านี้ ผมได้แก้ไขให้เหลือเพียงชุดเดียวที่ถูกต้องและสอดคล้องกัน
+
+2.  **อัปเดต `sentiment_counts`:** ในฟังก์ชัน `analyze_text` ได้เพิ่ม `very negative` และ `very positive` เข้าไปใน `sentiment_counts` เพื่อให้สามารถนับและแสดงผล Sentiment ทั้ง 5 ระดับได้อย่างถูกต้อง
+
+3.  **อัปเดต Legend ใน Gradio UI:** ได้เพิ่ม `very positive` และ `very negative` เข้าไปในส่วนของ Legend (คำอธิบายผลลัพธ์) ใน Gradio UI เพื่อให้ผู้ใช้เข้าใจผลลัพธ์ 5 ระดับได้อย่างชัดเจน
+
+-----
+
+## โค้ดที่แก้ไขแล้ว
+
+```python
 import gradio as gr
 from transformers import pipeline
 import re
@@ -40,6 +57,7 @@ def get_nlp(model_name: str):
 
 # Enhanced label mapping with modern styling for dark blue theme
 LABEL_MAPPINGS = {
+    # Mappings สำหรับ LABEL_X (จากโมเดล)
     "LABEL_0": {"code": 0, "name": "question", "emoji": "🤔", "color": "#60a5fa", "bg": "rgba(96, 165, 250, 0.2)", "description": "คำถาม"},
     "LABEL_1": {"code": 1, "name": "very negative", "emoji": "😡", "color": "#ef4444", "bg": "rgba(239, 68, 68, 0.2)", "description": "เชิงลบมาก"},
     "LABEL_2": {"code": 2, "name": "negative", "emoji": "😢", "color": "#f87171", "bg": "rgba(248, 113, 113, 0.2)", "description": "เชิงลบ"},
@@ -47,18 +65,21 @@ LABEL_MAPPINGS = {
     "LABEL_4": {"code": 4, "name": "positive", "emoji": "😊", "color": "#34d399", "bg": "rgba(52, 211, 153, 0.2)", "description": "เชิงบวก"},
     "LABEL_5": {"code": 5, "name": "very positive", "emoji": "🤩", "color": "#22c55e", "bg": "rgba(34, 197, 94, 0.2)", "description": "เชิงบวกมาก"},
 
+    # Mappings สำหรับชื่อเต็ม (ใช้สำหรับแสดงผลหรืออ้างอิง)
+    "QUESTION": {"code": 0, "name": "question", "emoji": "🤔", "color": "#60a5fa", "bg": "rgba(96, 165, 250, 0.2)", "description": "คำถาม"},
     "VERY_NEGATIVE": {"code": 1, "name": "very negative", "emoji": "😡", "color": "#ef4444", "bg": "rgba(239, 68, 68, 0.2)", "description": "เชิงลบมาก"},
     "NEGATIVE": {"code": 2, "name": "negative", "emoji": "😢", "color": "#f87171", "bg": "rgba(248, 113, 113, 0.2)", "description": "เชิงลบ"},
     "NEUTRAL": {"code": 3, "name": "neutral", "emoji": "😐", "color": "#facc15", "bg": "rgba(250, 204, 21, 0.2)", "description": "เป็นกลาง"},
     "POSITIVE": {"code": 4, "name": "positive", "emoji": "😊", "color": "#34d399", "bg": "rgba(52, 211, 153, 0.2)", "description": "เชิงบวก"},
     "VERY_POSITIVE": {"code": 5, "name": "very positive", "emoji": "🤩", "color": "#22c55e", "bg": "rgba(34, 197, 94, 0.2)", "description": "เชิงบวกมาก"},
-    
-    "POSITIVE": {"code": 3, "name": "positive", "emoji": "😊", "color": "#34d399", "bg": "rgba(52, 211, 153, 0.2)", "description": "เชิงบวก"},
-    "NEGATIVE": {"code": 1, "name": "negative", "emoji": "😢", "color": "#f87171", "bg": "rgba(248, 113, 113, 0.2)", "description": "เชิงลบ"},
-    "NEUTRAL": {"code": 2, "name": "neutral", "emoji": "😐", "color": "#facc15", "bg": "rgba(250, 204, 21, 0.2)", "description": "เป็นกลาง"},
-    
-    "0": {"code": 0, "name": "negative", "emoji": "😢", "color": "#f87171", "bg": "rgba(248, 113, 113, 0.2)", "description": "เชิงลบ"},
-    "1": {"code": 1, "name": "positive", "emoji": "😊", "color": "#34d399", "bg": "rgba(52, 211, 153, 0.2)", "description": "เชิงบวก"},
+
+    # Mappings สำหรับรหัสตัวเลข (ถ้าจำเป็นต้องแปลงจาก 0, 1, 2, ... เป็นชื่อ)
+    "0": {"code": 0, "name": "question", "emoji": "🤔", "color": "#60a5fa", "bg": "rgba(96, 165, 250, 0.2)", "description": "คำถาม"},
+    "1": {"code": 1, "name": "very negative", "emoji": "😡", "color": "#ef4444", "bg": "rgba(239, 68, 68, 0.2)", "description": "เชิงลบมาก"},
+    "2": {"code": 2, "name": "negative", "emoji": "😢", "color": "#f87171", "bg": "rgba(248, 113, 113, 0.2)", "description": "เชิงลบ"},
+    "3": {"code": 3, "name": "neutral", "emoji": "😐", "color": "#facc15", "bg": "rgba(250, 204, 21, 0.2)", "description": "เป็นกลาง"},
+    "4": {"code": 4, "name": "positive", "emoji": "😊", "color": "#34d399", "bg": "rgba(52, 211, 153, 0.2)", "description": "เชิงบวก"},
+    "5": {"code": 5, "name": "very positive", "emoji": "🤩", "color": "#22c55e", "bg": "rgba(34, 197, 94, 0.2)", "description": "เชิงบวกมาก"},
 }
 
 def get_label_info(label: str) -> Dict:
@@ -137,7 +158,16 @@ def analyze_text(text: str, model_name: str) -> str:
     </div>
     """]
     
-    sentiment_counts = {"positive": 0, "negative": 0, "neutral": 0, "question": 0, "other": 0}
+    # Updated sentiment_counts to include all 5 sentiment levels plus question
+    sentiment_counts = {
+        "question": 0, 
+        "very negative": 0, 
+        "negative": 0, 
+        "neutral": 0, 
+        "positive": 0, 
+        "very positive": 0, 
+        "other": 0
+    }
     total_confidence = 0
     sentence_results = []
     
@@ -151,6 +181,7 @@ def analyze_text(text: str, model_name: str) -> str:
             label_info = get_label_info(label)
             label_name = label_info["name"]
             
+            # Increment count for the corresponding sentiment name
             if label_name in sentiment_counts:
                 sentiment_counts[label_name] += 1
             else:
@@ -224,24 +255,52 @@ def analyze_text(text: str, model_name: str) -> str:
     
     # Create chart data for summary
     chart_items = []
-    colors = {"positive": "#34d399", "negative": "#f87171", "neutral": "#facc15", "question": "#60a5fa", "other": "#64748b"}
-    emojis = {"positive": "😊", "negative": "😢", "neutral": "😐", "question": "🤔", "other": "🔍"}
+    # Ensure colors and emojis include all 6 categories (question + 5 sentiments)
+    colors = {
+        "question": "#60a5fa",
+        "very negative": "#ef4444", 
+        "negative": "#f87171", 
+        "neutral": "#facc15", 
+        "positive": "#34d399", 
+        "very positive": "#22c55e",
+        "other": "#64748b"
+    }
+    emojis = {
+        "question": "🤔",
+        "very negative": "😡",
+        "negative": "😢", 
+        "neutral": "😐", 
+        "positive": "😊", 
+        "very positive": "🤩",
+        "other": "🔍"
+    }
     
-    for sentiment, count in sentiment_counts.items():
-        if count > 0:
+    # Iterate through all possible sentiment types to ensure they are displayed, even if count is 0
+    # Or, iterate through sentiment_counts.items() if you only want to display sentiments found in the text
+    for sentiment_name, count in sentiment_counts.items():
+        if sentiment_name == "other" and count == 0: # Don't show 'other' if no unknown labels
+            continue
+        if total_sentences == 0:
+            percentage = 0
+        else:
             percentage = (count / total_sentences) * 100
-            chart_items.append(f"""
-            <div style="display: flex; align-items: center; gap: 12px; padding: 12px; background: rgba(59, 130, 246, 0.1); border-radius: 8px;">
-                <span style="font-size: 24px;">{emojis.get(sentiment, '🔍')}</span>
-                <div style="flex: 1;">
-                    <div style="font-weight: 600; color: #f8fafc; text-transform: capitalize;">{sentiment}</div>
-                    <div style="color: #94a3b8; font-size: 14px;">{count} ประโยค ({percentage:.1f}%)</div>
-                </div>
-                <div style="width: 60px; height: 6px; background: #334155; border-radius: 3px; overflow: hidden;">
-                    <div style="width: {percentage}%; height: 100%; background: {colors.get(sentiment, '#64748b')}; transition: all 0.3s ease;"></div>
-                </div>
+        
+        # Get description from LABEL_MAPPINGS using the sentiment name
+        # We can find it by iterating through LABEL_MAPPINGS values
+        description = next((info['description'] for info in LABEL_MAPPINGS.values() if info['name'] == sentiment_name), sentiment_name)
+
+        chart_items.append(f"""
+        <div style="display: flex; align-items: center; gap: 12px; padding: 12px; background: rgba(59, 130, 246, 0.1); border-radius: 8px;">
+            <span style="font-size: 24px;">{emojis.get(sentiment_name, '🔍')}</span>
+            <div style="flex: 1;">
+                <div style="font-weight: 600; color: #f8fafc; text-transform: capitalize;">{description}</div>
+                <div style="color: #94a3b8; font-size: 14px;">{count} ประโยค ({percentage:.1f}%)</div>
             </div>
-            """)
+            <div style="width: 60px; height: 6px; background: #334155; border-radius: 3px; overflow: hidden;">
+                <div style="width: {percentage}%; height: 100%; background: {colors.get(sentiment_name, '#64748b')}; transition: all 0.3s ease;"></div>
+            </div>
+        </div>
+        """)
     
     html_parts.append(f"""
     <div style="padding: 24px; background: linear-gradient(135deg, #1e293b 0%, #0f172a 100%);">
@@ -446,9 +505,11 @@ with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Base()) as demo:
                     <span>🗂️</span> คำอธิบายผลลัพธ์
                 </div>
                 <div class='legend-row'>
+                    <div class='legend-item'><strong>🤩 เชิงบวกมาก</strong><br><small>Very Positive</small></div>
                     <div class='legend-item'><strong>😊 เชิงบวก</strong><br><small>Positive</small></div>
-                    <div class='legend-item'><strong>😢 เชิงลบ</strong><br><small>Negative</small></div>
                     <div class='legend-item'><strong>😐 เป็นกลาง</strong><br><small>Neutral</small></div>
+                    <div class='legend-item'><strong>😢 เชิงลบ</strong><br><small>Negative</small></div>
+                    <div class='legend-item'><strong>😡 เชิงลบมาก</strong><br><small>Very Negative</small></div>
                     <div class='legend-item'><strong>🤔 คำถาม</strong><br><small>Question</small></div>
                 </div>
             </div>
@@ -480,4 +541,5 @@ if __name__ == "__main__":
             "docs_url": None,
             "redoc_url": None,
         }
-    )
+    )
+```