Update static/script.js

static/script.js

@@ -1,20 +1,50 @@
+async function handleResponse(response) {
+    const raw = await response.text();
+
+    // Try to parse JSON safely
+    try {
+        return JSON.parse(raw);
+    } catch (e) {
+        throw new Error("Invalid JSON response:\n" + raw);
+    }
+}
+
+
+// ======================
+// NER
+// ======================
 async function runNER() {
     try {
-        const text = document.getElementById("text").value;
+        const text = document.getElementById("text").value.trim();
+
+        if (!text) {
+            document.getElementById("output").textContent = "Please enter text.";
+            return;
+        }
 
         const response = await fetch("/predict", {
             method: "POST",
             headers: { "Content-Type": "application/json" },
-            body: JSON.stringify({ text, mode: "1" })
+            body: JSON.stringify({ text: text, mode: "1" })
         });
 
-        const raw = await response.text();  // 👈 IMPORTANT
-        console.log("RAW RESPONSE:", raw);
+        const data = await handleResponse(response);
+
+        if (data.error) {
+            document.getElementById("output").textContent =
+                "NER Backend Error:\n" + data.error;
+            return;
+        }
 
-        const data = JSON.parse(raw);
+        if (!data.resp || data.resp.length === 0) {
+            document.getElementById("output").textContent =
+                "No NER results.";
+            return;
+        }
 
+        // pretty print
         document.getElementById("output").textContent =
-            JSON.stringify(data, null, 2);
+            JSON.stringify(data.resp, null, 2);
 
     } catch (err) {
         document.getElementById("output").textContent =
@@ -23,27 +53,54 @@ async function runNER() {
 }
 
 
+// ======================
+// Relation Extraction
+// ======================
 async function runRE() {
-    const text = document.getElementById("text").value;
+    try {
+        const text = document.getElementById("text").value.trim();
 
-    const response = await fetch("/predict_re", {
-        method: "POST",
-        headers: { "Content-Type": "application/json" },
-        body: JSON.stringify({ text })
-    });
+        if (!text) {
+            document.getElementById("output").textContent = "Please enter text.";
+            return;
+        }
 
-    const data = await response.json();
+        const response = await fetch("/predict_re", {
+            method: "POST",
+            headers: { "Content-Type": "application/json" },
+            body: JSON.stringify({ text: text })
+        });
+
+        const data = await handleResponse(response);
+
+        if (data.error) {
+            document.getElementById("output").textContent =
+                "RE Backend Error:\n" + data.error;
+            return;
+        }
 
-    // nicer formatting for relations
-    let formatted = "";
+        if (!data.resp || data.resp.length === 0) {
+            document.getElementById("output").textContent =
+                "No relations found.";
+            return;
+        }
+
+        // formatted output
+        let formatted = "";
 
-    if (data.resp.length === 0) {
-        formatted = "No relations found.";
-    } else {
         data.resp.forEach(r => {
-            formatted += `${r[0]} → ${r[1]} → ${r[2]} (conf: ${r[3].toFixed(2)})\n`;
+            const subj = r[0];
+            const rel = r[1];
+            const obj = r[2];
+            const conf = r[3];
+
+            formatted += `${subj} → ${rel} → ${obj} (conf: ${conf.toFixed(2)})\n`;
         });
-    }
 
-    document.getElementById("output").textContent = formatted;
+        document.getElementById("output").textContent = formatted;
+
+    } catch (err) {
+        document.getElementById("output").textContent =
+            "RE Error: " + err.message;
+    }
 }