Update app.py

app.py

@@ -34,17 +34,25 @@ long_context_examples = [
      "The cafe is experiencing a slow, quiet morning"]
 ]
 
-def get_label_color(label):
-    """Return color based on NLI label."""
-    colors = {
-        'entailment': '#90EE90',  # Light green
-        'neutral': '#FFE5B4',     # Peach
-        'contradiction': '#FFB6C1' # Light pink
+def get_label_color(label, confidence=1.0):
+    """Return color based on NLI label with confidence-based saturation."""
+    base_colors = {
+        'entailment': 'rgb(144, 238, 144)',    # Light green
+        'neutral': 'rgb(255, 229, 180)',       # Peach
+        'contradiction': 'rgb(255, 182, 193)'   # Light pink
     }
-    return colors.get(label, '#FFFFFF')
+    
+    # Convert RGB color to RGBA with confidence-based alpha
+    if label in base_colors:
+        rgb = base_colors[label].replace('rgb(', '').replace(')', '').split(',')
+        r, g, b = map(int, rgb)
+        # Adjust the color based on confidence
+        alpha = 0.3 + (0.7 * confidence)  # Range from 0.3 to 1.0
+        return f"rgba({r},{g},{b},{alpha})"
+    return '#FFFFFF'
 
-def create_analysis_html(sentence_results, global_label):
-    """Create HTML table for sentence analysis with color coding."""
+def create_analysis_html(sentence_results, global_label, global_confidence):
+    """Create HTML table for sentence analysis with color coding and confidence."""
     html = """
     <style>
         .analysis-table {
@@ -67,13 +75,18 @@ def create_analysis_html(sentence_results, global_label):
             border-radius: 5px;
             font-weight: bold;
         }
+        .confidence {
+            font-size: 0.9em;
+            color: #666;
+        }
     </style>
     """
     
-    # Add global prediction box
+    # Add global prediction box with confidence
     html += f"""
-    <div class="global-prediction" style="background-color: {get_label_color(global_label)}">
-        Global Prediction: {global_label}
+    <div class="global-prediction" style="background-color: {get_label_color(global_label, global_confidence)}">
+        Global Prediction: {global_label} 
+        <span class="confidence">(Confidence: {global_confidence:.2%})</span>
     </div>
     """
     
@@ -83,15 +96,17 @@ def create_analysis_html(sentence_results, global_label):
         <tr>
             <th>Sentence</th>
             <th>Prediction</th>
+            <th>Confidence</th>
         </tr>
     """
     
     # Add rows for each sentence
     for result in sentence_results:
         html += f"""
-        <tr style="background-color: {get_label_color(result['prediction'])}">
+        <tr style="background-color: {get_label_color(result['prediction'], result['confidence'])}">
             <td>{result['sentence']}</td>
             <td>{result['prediction']}</td>
+            <td>{result['confidence']:.2%}</td>
         </tr>
         """