Update static/script.js

static/script.js

@@ -1,17 +1,40 @@
 async function runNER() {
     const text = document.getElementById("text").value;
-    const mode = document.getElementById("mode").value;
 
     const response = await fetch("/predict", {
         method: "POST",
-        headers: {
-            "Content-Type": "application/json"
-        },
-        body: JSON.stringify({ text, mode })
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ text, mode: "1" })
     });
 
     const data = await response.json();
 
     document.getElementById("output").textContent =
         JSON.stringify(data, null, 2);
+}
+
+
+async function runRE() {
+    const text = document.getElementById("text").value;
+
+    const response = await fetch("/predict_re", {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ text })
+    });
+
+    const data = await response.json();
+
+    // nicer formatting for relations
+    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`;
+        });
+    }
+
+    document.getElementById("output").textContent = formatted;
 }