{% extends "base/base.html" %}
{% comment %}
    SPDX-License-Identifier: AGPL-3.0-or-later
    Copyright (C) 2024-2025 Andrew Rechnitzer
    Copyright (C) 2024-2026 Colin B. Macdonald
    Copyright (C) 2024-2025 Deep Shah
    Copyright (C) 2026 Aidan Murphy
{% endcomment %}
{% load static %}
{% block title %}
    ID Predictions
{% endblock title %}
{% block page_heading %}
    ID Predictions
{% endblock page_heading %}
{% block main_content %}
    <div>
        {% if id_reader_task_status.status == "Complete" %}
            <p class="my-4">
                <a class="btn btn-primary" href="{% url 'get_id_box_parent' %}">Re-run ID auto-reader tool</a>
                {% for predictor in list_of_ML_predictors %}
                    <button type="button"
                            class="btn btn-warning"
                            hx-delete="{% url 'id_prediction_delete' predictor %}"
                            hx-confirm="Are you sure?">Delete {{ predictor }} predictions</button>
                {% endfor %}
            </p>
            <span class="alert alert-success my-4">{{ id_reader_task_status.message }}</span>
        {% elif id_reader_task_status.status == "Error" %}
            <p class="my-4">
                <a class="btn btn-primary" href="{% url 'get_id_box_parent' %}">Re-run ID auto-reader tool</a>
                {% for predictor in list_of_ML_predictors %}
                    <button type="button"
                            class="btn btn-warning"
                            hx-delete="{% url 'id_prediction_delete' predictor %}"
                            hx-confirm="Are you sure?">Delete {{ predictor }} predictions</button>
                {% endfor %}
            </p>
            <span class="alert alert-warning my-4">{{ id_reader_task_status.message }}</span>
        {% elif id_reader_task_status.status == "To Do" %}
            <p class="my-4">
                <a class="btn btn-primary" href="{% url 'get_id_box_parent' %}">Run ID auto-reader tool</a>
                {% for predictor in list_of_ML_predictors %}
                    <button type="button"
                            class="btn btn-warning"
                            hx-delete="{% url 'id_prediction_delete' predictor %}"
                            hx-confirm="Are you sure?">Delete {{ predictor }} predictions</button>
                {% endfor %}
            </p>
            <span class="alert alert-info my-4">Reader not yet run.</span>
        {% else %}
            <p class="my-4">
                <button disabled
                        type="button"
                        class="btn btn-primary"
                        href="{% url 'get_id_box_parent' %}">Run ID auto-reader tool</button>
                {% for predictor in list_of_ML_predictors %}
                    <button disabled type="button" class="btn btn-warning">Delete {{ predictor }} predictions</button>
                {% endfor %}
                <a class="btn btn-primary"
                   href="{% url 'id_prediction_home' %}"><i class="bi bi-arrow-clockwise"></i></a>
            </p>
            <span class="alert alert-info my-4">{{ id_reader_task_status.message }}</span>
        {% endif %}
    </div>
    <div class="row gx-0 my-4">
        <div class="col-8">
            <div class="card">
                <div class="card-body">
                    <h5 class="card-title">Auto-read paper IDs</h5>
                    <p>Note that this table only shows papers which are prenamed or have predicted IDs</p>
                    <div style="max-height: 75vh; overflow-y: auto;">
                        <table class="table table-striped sortable">
                            <tr>
                                <th>Paper</th>
                                <th>Identified</th>
                                <th>Prenamed ID</th>
                                <th>Predicted ID (LAP)</th>
                                <th>Certainty</th>
                                <th>Predicted ID (Greedy)</th>
                                <th>Certainty</th>
                                <th>Best Guess (Raw)</th>
                                <th>Certainty</th>
                                <th>Other Predictions</th>
                            </tr>
                            {% for pn, dat in predictions.items %}
                                <!-- Note: table-xxx styling "most likely won’t see" dark mode until
                                     Bootstrap v6, https://getbootstrap.com/docs/5.3/content/tables/ -->
                                <tr {% if 'identified' in dat %}class="table-success" {% elif 'prename' in dat %}class="table-secondary" {% endif %}>
                                    <td>
                                        {{ pn }}
                                        {% if dat.image_pk %}
                                            <button type="button"
                                                    class="btn btn-outline-success"
                                                    hx-get="{% url 'ID_img_wrap' dat.image_pk %}"
                                                    hx-target="#imageHere"
                                                    hx-replace="innerHTML">view</button>
                                        {% endif %}
                                    </td>
                                    <!-- TODO: Do more in the service/view, not the template -->
                                    {% if "identified" in dat %}
                                        <td>{{ dat.identified }}</td>
                                    {% else %}
                                        <td>{{ dat.status }}</td>
                                    {% endif %}
                                    {% if "prename" in dat %}
                                        <td>{{ dat.prename.0 }}</td>
                                    {% else %}
                                        <td></td>
                                    {% endif %}
                                    {% if "MLLAP" in dat %}
                                        <td>{{ dat.MLLAP.0 }}</td>
                                        <td>{{ dat.MLLAP.1 }}</td>
                                    {% else %}
                                        <td></td>
                                        <td></td>
                                    {% endif %}
                                    {% if "MLGreedy" in dat %}
                                        <td>{{ dat.MLGreedy.0 }}</td>
                                        <td>{{ dat.MLGreedy.1 }}</td>
                                    {% else %}
                                        <td></td>
                                        <td></td>
                                    {% endif %}
                                    {% if "MLBestGuess" in dat %}
                                        <td>{{ dat.MLBestGuess.0 }}</td>
                                        <td>{{ dat.MLBestGuess.1 }}</td>
                                    {% else %}
                                        <td></td>
                                        <td></td>
                                    {% endif %}
                                    <td>
                                        {% for name, prediction_data in dat.items %}
                                            {% if name == "identified" %}
                                            {% elif name == "prename" %}
                                            {% elif name == "MLLAP" %}
                                            {% elif name == "MLGreedy" %}
                                            {% elif name == "MLBestGuess" %}
                                            {% elif name == "image_pk" %}
                                            {% else %}
                                                {{ name }}: {{ prediction_data.0 }} ({{ prediction_data.1 }})
                                                <br />
                                            {% endif %}
                                        {% endfor %}
                                    </td>
                                </tr>
                            {% endfor %}
                        </table>
                    </div>
                </div>
            </div>
        </div>
        <div class="col-4">
            <div id="imageHere"></div>
        </div>
    </div>
{% endblock main_content %}