Update static/script.js

static/script.js

@@ -52,6 +52,90 @@ async function runNER() {
     }
 }
 
+async function runEAE() {
+    const text = document.getElementById("text").value;
+    const output = document.getElementById("output");
+
+    if (!text.trim()) {
+        output.textContent = "Please enter Arabic text.";
+        return;
+    }
+
+    output.textContent = "Processing event arguments...";
+
+    try {
+        const response = await fetch("/predict_eae", {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json"
+            },
+            body: JSON.stringify({ text: text })
+        });
+
+        const data = await response.json();
+        output.textContent = JSON.stringify(data, null, 2);
+    } catch (error) {
+        output.textContent = "Error extracting event arguments:\n" + error;
+    }
+}
+
+async function runAllRelations() {
+    const text = document.getElementById("text").value;
+    const output = document.getElementById("output");
+
+    if (!text.trim()) {
+        output.textContent = "Please enter Arabic text.";
+        return;
+    }
+
+    output.textContent = "Extracting relations and event arguments...";
+
+    try {
+        const [reResponse, eaeResponse] = await Promise.all([
+            fetch("/predict_re", {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json"
+                },
+                body: JSON.stringify({ text: text })
+            }),
+
+            fetch("/predict_eae", {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json"
+                },
+                body: JSON.stringify({ text: text })
+            })
+        ]);
+
+        const reData = await reResponse.json();
+        const eaeData = await eaeResponse.json();
+
+        const relationResults = reData.resp || [];
+        const eventResults = eaeData.resp || [];
+
+        const combinedResults = {
+            resp: [
+                ...relationResults.map(item => ({
+                    ...item,
+                    Task: "Relation Extraction"
+                })),
+                ...eventResults.map(item => ({
+                    ...item,
+                    Task: "Event Argument Extraction"
+                }))
+            ],
+            statusText: "OK",
+            statusCode: 0
+        };
+
+        output.textContent = JSON.stringify(combinedResults, null, 2);
+
+    } catch (error) {
+        output.textContent = "Error extracting all relations:\n" + error;
+    }
+}
 
 // ======================
 // Relation Extraction