better confidence score

wasm-demo.js

@@ -493,32 +493,32 @@ $('btnDecrypt').onclick = () => {
         console.log('[Main] Decryption successful');
         console.log(`[Main] Result - flag: ${flag}, raw_score: ${rawScore}, total_g: ${total_g}`);
         
-        // Convert to confidence value between 0 and 1
-        const confidence = Math.max(0, Math.min(1, rawScore));
+        // Convert raw score to watermark probability
+        // rawScore: 0.5 = not watermarked, 1.0 = watermarked
+        // Transform to: 0% = not watermarked, 100% = watermarked
+        const watermarkProbability = Math.max(0, Math.min(1, 2 * (rawScore - 0.5)));
+        const probabilityPercent = watermarkProbability * 100;
         
-        // Determine result based on confidence ranges
-        let resultText, confidenceText, resultClass;
+        // Determine result based on watermark probability ranges
+        let resultText, probabilityText, resultClass;
         
-        if (confidence < 0.5) {
-            // Problematic range - should not be addressed according to user
-            resultText = '⚠️ Inconclusive';
-            confidenceText = `Confidence: ${(confidence * 100).toFixed(1)}% (insufficient token(s))`;
+        if (probabilityPercent < 30) {
+            resultText = '👤 Likely Human-Generated';
+            probabilityText = `Probability of being watermarked: ${probabilityPercent.toFixed(1)}%`;
+            resultClass = 'inconclusive';
+        } else if (probabilityPercent >= 30 && probabilityPercent < 70) {
+            resultText = '⚠️ Uncertain';
+            probabilityText = `Probability of being watermarked: ${probabilityPercent.toFixed(1)}%`;
             resultClass = 'inconclusive';
-        } else if (confidence >= 0.5 && confidence < 0.6) {
-            // Very likely to be watermarked (close to 0.5)
-            resultText = '✅ AI-Generated (Watermarked)';
-            confidenceText = `Confidence: ${(confidence * 100).toFixed(1)}%`;
-            resultClass = 'watermarked';
         } else {
-            // Between 0.6 and 1 - highly likely to be AI generated
-            resultText = '🤖 AI-Generated';
-            confidenceText = `Confidence: ${(confidence * 100).toFixed(1)}%`;
-            resultClass = 'ai-generated';
+            resultText = '✅ Likely AI-Generated (Watermarked)';
+            probabilityText = `Probability of being watermarked: ${probabilityPercent.toFixed(1)}%`;
+            resultClass = 'watermarked';
         }
         
         $('decResult').innerHTML = `
             <div class="watermark-flag ${resultClass}">${resultText}</div>
-            <div class="watermark-score">${confidenceText}</div>
+            <div class="watermark-score">${probabilityText}</div>
         `;
     } catch (e) {
         console.error('[Main] Decryption error:', e);