v4.2: Update extension/background.js

extension/background.js

@@ -4,7 +4,8 @@
  * FIXED: Error handling and retry logic
  */
 
-const API_BASE = "https://gaurv007-clauseguard-api.hf.space";
+// FIX v4.2: Corrected API_BASE URL to match the actual Gradio Space
+const API_BASE = "https://gaurv007-clauseguard.hf.space";
 const FREE_SCANS_PER_MONTH = 10;
 const API_TIMEOUT_MS = 45000;
 
@@ -181,13 +182,19 @@ function localAnalyze(text) {
   });
 
   const flagged = results.filter(r => r.categories.length > 0);
-  const sev = { HIGH: 0, MEDIUM: 0, LOW: 0 };
-  flagged.forEach(r => r.categories.forEach(c => sev[c.severity]++));
-  const risk = Math.min(100, Math.round((sev.HIGH*20 + sev.MEDIUM*10 + sev.LOW*5) / Math.max(1, clauses.length) * 100));
+  const sev = { CRITICAL: 0, HIGH: 0, MEDIUM: 0, LOW: 0 };
+  flagged.forEach(r => r.categories.forEach(c => {
+    if (sev.hasOwnProperty(c.severity)) sev[c.severity]++;
+    else sev.MEDIUM++;  // default for unknown severity
+  }));
+  // FIX v4.2: Use the same diminishing-returns formula as the backend (app.py)
+  // instead of normalizing by clause count (which gave different scores)
+  const weighted = sev.CRITICAL*40 + sev.HIGH*20 + sev.MEDIUM*10 + sev.LOW*3;
+  const risk = Math.min(100, Math.round(100 * (1 - (1 / (1 + weighted / 30)))));
 
   return {
     risk_score: risk,
-    grade: risk >= 60 ? "F" : risk >= 40 ? "D" : risk >= 20 ? "C" : risk >= 10 ? "B" : "A",
+    grade: risk >= 70 ? "F" : risk >= 50 ? "D" : risk >= 30 ? "C" : risk >= 15 ? "B" : "A",
     total_clauses: clauses.length, flagged_count: flagged.length, results,
   };
 }