color coding

app.py

@@ -213,20 +213,20 @@ def create_demo() -> gr.Blocks:
         empirical_context = rec.get("empirical_context", "—")
         explanation = rec.get("explanation", "—")
         is_mixed = rec.get("is_mixed_type", False)
+        llm_is_dataset = rec.get("llm_is_dataset_contextual")
 
-        # Highlight term in empirical context
-        if rec.get("ner_text") and rec.get("text"):
+        # Apply conditional highlighting based on LLM assessment
+        if rec.get("ner_text") and rec.get("text") and llm_is_dataset is not None:
             try:
-                # Get the term from the full text
                 start, end = rec["ner_text"][0][0], rec["ner_text"][0][1]
                 term = rec["text"][start:end]
-                
-                # Highlight it in the empirical context if present
-                # We use HTML styling for better visibility
+                if llm_is_dataset:
+                    highlight_style = 'background-color: #90ee90; color: black; padding: 2px 4px; border-radius: 4px; font-weight: bold; border: 1px solid #5cb85c;'
+                else:
+                    highlight_style = 'background-color: #ff7f7f; color: black; padding: 2px 4px; border-radius: 4px; font-weight: bold; border: 1px solid #d9534f;'
                 if term and term in empirical_context:
-                    highlight_style = 'background-color: #ffd700; color: black; padding: 2px 4px; border-radius: 4px; font-weight: bold; border: 1px solid #e6c200;'
                     empirical_context = empirical_context.replace(term, f'<span style="{highlight_style}">{term}</span>')
-            except:
+            except Exception:
                 pass
 
         # Build HTML
@@ -247,7 +247,6 @@ def create_demo() -> gr.Blocks:
         """
         
         # Add LLM contextual analysis section if available
-        llm_is_dataset = rec.get("llm_is_dataset_contextual")
         llm_reasons = rec.get("llm_contextual_reason", [])
         llm_thinking = rec.get("llm_thinking_contextual", "")