fix: web/app/api/compare/route.ts

web/app/api/compare/route.ts

@@ -13,7 +13,7 @@ export async function POST(req: NextRequest) {
     }
 
     const body = await req.json();
-    let { text_a, text_b } = body;
+    const { text_a, text_b } = body;
 
     if (!text_a || !text_b || text_a.trim().length < 50 || text_b.trim().length < 50) {
       return NextResponse.json(
@@ -21,13 +21,14 @@ export async function POST(req: NextRequest) {
         { status: 400 }
       );
     }
-    
-    // Sanitize basically
-    text_a = text_a.replace(/</g, "&lt;").replace(/>/g, "&gt;");
-    text_b = text_b.replace(/</g, "&lt;").replace(/>/g, "&gt;");
+
+    // FIX v4.3: REMOVED HTML-escaping that CORRUPTED contract text before analysis.
+    // The old code did text_a.replace(/</g, "&lt;") which permanently mutated
+    // the text (e.g., ">$10,000" → "&gt;$10,000"). Sanitization is the
+    // frontend's job — React auto-escapes in JSX. Never mutate analysis input.
 
     // Call Gradio Space API
-    const submitRes = await fetch(`${GRADIO_URL}/gradio_api/call/run_comparison`, {
+    const submitRes = await fetch(`${GRADIO_URL}/gradio_api/call/compare`, {
       method: "POST",
       headers: { "Content-Type": "application/json" },
       body: JSON.stringify({ data: [text_a, text_b] }),
@@ -40,24 +41,44 @@ export async function POST(req: NextRequest) {
     const { event_id } = await submitRes.json();
     if (!event_id) throw new Error("No event_id from Gradio");
 
-    // Poll for result
-    const resultRes = await fetch(
-      `${GRADIO_URL}/gradio_api/call/run_comparison/${event_id}`,
-      { headers: { Accept: "text/event-stream" } }
-    );
+    // Poll for result with retry
+    let resultText = "";
+    let attempts = 0;
+    const maxAttempts = 60;
+    let delay = 500;
+
+    while (attempts < maxAttempts) {
+      const resultRes = await fetch(
+        `${GRADIO_URL}/gradio_api/call/compare/${event_id}`,
+        { headers: { Accept: "text/event-stream" } }
+      );
+
+      resultText = await resultRes.text();
+
+      if (resultText.includes("event: complete")) break;
+      if (resultText.includes("event: error")) {
+        const errMatch = resultText.match(/data:\s*(.+)/);
+        throw new Error(errMatch ? errMatch[1] : "Comparison failed in backend");
+      }
 
-    if (!resultRes.ok) {
-      throw new Error(`Gradio result failed: ${resultRes.status}`);
+      await new Promise(r => setTimeout(r, delay));
+      delay = Math.min(delay * 1.2, 2000);
+      attempts++;
     }
 
-    const resultText = await resultRes.text();
-    const dataMatch = resultText.match(/event:\s*complete\s*\ndata:\s*(.+)/);
-    if (!dataMatch) throw new Error("No complete event from Gradio");
+    if (!resultText.includes("event: complete")) {
+      throw new Error("Comparison timed out. Please try again.");
+    }
+
+    const completeIdx = resultText.indexOf("event: complete");
+    const dataIdx = resultText.indexOf("data: ", completeIdx);
+    if (dataIdx === -1) throw new Error("No data in response");
+
+    const dataStr = resultText.substring(dataIdx + 6).trim();
+    const gradioData = JSON.parse(dataStr);
 
-    const gradioData = JSON.parse(dataMatch[1]);
     // gradioData[0] = comparison HTML
     // gradioData[1] = raw JSON comparison data
-
     const comparisonResult = gradioData[1];
     if (typeof comparisonResult === "object" && comparisonResult !== null) {
       return NextResponse.json(comparisonResult);