implemented detection

wm_detector/core/utils.py

@@ -46,6 +46,7 @@ def get_token_details(
             pvalue = float(pvalue)
             
         display_info.append({
+            'is_scored': token_detail['is_scored'],
             'token': token_detail['token_text'],
             'color': color_from_score(score),
             'score': score,

wm_detector/static/styles.css

@@ -59,13 +59,27 @@ h1 {
     bottom: 125%;
     left: 50%;
     transform: translateX(-50%);
-    padding: 8px;
-    background-color: rgba(0, 0, 0, 0.8);
+    padding: 6px 10px;
+    background-color: rgba(0, 0, 0, 0.9);
     color: white;
     border-radius: 4px;
     font-size: 12px;
+    line-height: 1.4;
     white-space: nowrap;
     pointer-events: none;
+    font-family: monospace;
+    box-shadow: 0 2px 4px rgba(0,0,0,0.2);
+}
+
+.token-tooltip::after {
+    content: "";
+    position: absolute;
+    top: 100%;
+    left: 50%;
+    margin-left: -5px;
+    border-width: 5px;
+    border-style: solid;
+    border-color: rgba(0, 0, 0, 0.9) transparent transparent transparent;
 }
 
 .token:hover .token-tooltip {

wm_detector/templates/index.html

@@ -117,15 +117,20 @@
             const data = await response.json();
             
             // Update token display
-            tokenDisplay.innerHTML = data.tokens.map((token, i) => 
-                `<span class="token" style="background-color: ${data.colors[i]}">
+            tokenDisplay.innerHTML = data.tokens.map((token, i) => {
+                const score = data.scores[i];
+                const pvalue = data.pvalues[i];
+                const scoreDisplay = (score !== null && !isNaN(score)) ? score.toFixed(3) : 'N/A';
+                const pvalueDisplay = (pvalue !== null && !isNaN(pvalue)) ? pvalue.toFixed(3) : 'N/A';
+                
+                return `<span class="token" style="background-color: ${data.colors[i]}">
                     ${token}
                     <div class="token-tooltip">
-                        Score: ${data.scores[i] !== null ? data.scores[i].toFixed(3) : 'N/A'}<br>
-                        P-value: ${data.pvalues[i] !== null ? data.pvalues[i].toFixed(3) : 'N/A'}
+                        Score: ${scoreDisplay}<br>
+                        P-value: ${pvalueDisplay}
                     </div>
-                </span>`
-            ).join('');
+                </span>`;
+            }).join('');
             
             // Update counts and stats
             tokenCount.textContent = data.token_count;