static/script.js

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.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: text, mode: "1" })
        });

        const data = await handleResponse(response);

        if (data.error) {
            document.getElementById("output").textContent =
                "NER Backend Error:\n" + data.error;
            return;
        }

        if (!data.resp || data.resp.length === 0) {
            document.getElementById("output").textContent =
                "No NER results.";
            return;
        }

        // pretty print
        document.getElementById("output").textContent =
            JSON.stringify(data.resp, null, 2);

    } catch (err) {
        document.getElementById("output").textContent =
            "NER Error: " + err.message;
    }
}

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
// ======================
async function runRE() {
    try {
        const text = document.getElementById("text").value.trim();

        if (!text) {
            document.getElementById("output").textContent = "Please enter text.";
            return;
        }

        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;
        }

        if (!data.resp || data.resp.length === 0) {
            document.getElementById("output").textContent =
                "No relations found.";
            return;
        }

        // =========================
        // CLEAN JSON OUTPUT FORMAT
        // =========================
        const formattedList = data.resp.map(r => ({
            Subject: {
                Type: r.Subject.Type,
                Label: r.Subject.Label
            },
            Relation: r.Relation,
            Object: {
                Type: r.Object.Type,
                Label: r.Object.Label
            },
            Confidence: r.Confidence
        }));

        document.getElementById("output").textContent =
            JSON.stringify(formattedList, null, 2);

    } catch (err) {
        document.getElementById("output").textContent =
            "RE Error: " + err.message;
    }
}