{% extends "base.html" %}

{% block title %}Run #{{ run.id }} · Sparsifire{% endblock %}

{% block extra_js %}
<script>
    document.addEventListener('DOMContentLoaded', function () {
        const runStatus = "{{ run.status }}";

        // Only poll if running or queued
        if (runStatus === 'running' || runStatus === 'queued') {
            const pollInterval = setInterval(function () {
                fetch(window.location.href)
                    .then(response => response.text())
                    .then(html => {
                        const parser = new DOMParser();
                        const doc = parser.parseFromString(html, 'text/html');

                        // Update Status Panel
                        const newStatusPanel = doc.getElementById('status-panel');
                        const currentStatusPanel = document.getElementById('status-panel');
                        if (newStatusPanel && currentStatusPanel) {
                            currentStatusPanel.innerHTML = newStatusPanel.innerHTML;
                        }

                        // Update Log Container
                        const newLogContainer = doc.getElementById('log-container');
                        const currentLogContainer = document.getElementById('log-container');
                        if (newLogContainer && currentLogContainer) {
                            currentLogContainer.innerHTML = newLogContainer.innerHTML;
                        }

                        // Check if status changed to completed or failed to stop polling
                        // We can check the text content of the status badge or a data attribute
                        // For simplicity, let's look for the "Completed" or "Failed" text in the new status panel
                        const statusText = newStatusPanel.textContent;
                        if (statusText.includes('Completed') || statusText.includes('Failed')) {
                            clearInterval(pollInterval);
                            // Optional: Reload one last time to ensure everything is synced or just let the DOM update stand
                            window.location.reload();
                        }
                    })
                    .catch(err => console.error('Polling error:', err));
            }, 2000); // Poll every 2 seconds
        }
    });
</script>
{% endblock %}

{% block content %}
<div class="d-flex justify-content-between align-items-end mb-4">
    <div>
        <h6 class="text-uppercase text-muted mb-1 small ls-1">
            <a href="{% url 'sae:run_list' %}" class="text-decoration-none text-muted hover-white">Experiments</a>
            <span class="mx-1">/</span> Detail
        </h6>
        <h1 class="h3 mb-0 d-flex align-items-center gap-2">
            Run #{{ run.id }}
            <span class="badge bg-dark border border-secondary border-opacity-25 text-muted fw-normal fs-6 ms-2">
                <i class="bi bi-database me-1"></i> {{ run.dataset.name }}
            </span>
        </h1>
    </div>

    <div>
        {% if run.status == 'queued' or run.status == 'failed' %}
        <a href="{% url 'sae:start_run' pk=run.pk %}" class="btn-accent shadow-sm text-decoration-none">
            <i class="bi bi-play-fill me-1"></i> Start Training
        </a>
        {% elif run.status == 'completed' %}
        <a href="{% url 'explorer:download_sae_weights' run.id %}"
            class="btn btn-ghost text-muted border-secondary border-opacity-25 me-2">
            <i class="bi bi-download me-2"></i> Download .pt
        </a>
        <button class="btn-ghost text-success border-success border-opacity-25" disabled>
            <i class="bi bi-check-circle-fill me-2"></i> Completed
        </button>
        {% else %}
        <button class="btn-ghost text-warning border-warning border-opacity-25" disabled>
            <span class="spinner-border spinner-border-sm me-2" role="status" aria-hidden="true"></span>
            Training in progress...
        </button>
        {% endif %}
    </div>
</div>

<div class="row g-4">

    <div class="col-lg-4">
        <div class="panel mb-4" id="status-panel">
            <h6 class="text-uppercase text-muted small mb-3 ls-1">Status</h6>
            <div class="d-flex align-items-center">
                {% if run.status == 'completed' %}
                <div class="rounded-circle bg-success bg-opacity-10 p-3 text-success me-3">
                    <i class="bi bi-check-lg fs-4"></i>
                </div>
                <div>
                    <div class="fs-5 fw-bold text-white">Completed</div>
                    <div class="small text-muted">Loss: <span
                            class="text-white font-monospace">{{run.final_loss|floatformat:5 }}</span></div>
                </div>
                {% elif run.status == 'running' %}
                <div class="rounded-circle bg-warning bg-opacity-10 p-3 text-warning me-3">
                    <i class="bi bi-hourglass-split fs-4"></i>
                </div>
                <div>
                    <div class="fs-5 fw-bold text-white">Training</div>
                    <div class="small text-muted">Check log for progress</div>
                </div>
                {% elif run.status == 'failed' %}
                <div class="rounded-circle bg-danger bg-opacity-10 p-3 text-danger me-3">
                    <i class="bi bi-x-lg fs-4"></i>
                </div>
                <div>
                    <div class="fs-5 fw-bold text-white">Failed</div>
                    <div class="small text-muted">Error during execution</div>
                </div>
                {% else %}
                <div class="rounded-circle bg-secondary bg-opacity-10 p-3 text-muted me-3">
                    <i class="bi bi-pause-fill fs-4"></i>
                </div>
                <div>
                    <div class="fs-5 fw-bold text-white">Queued</div>
                    <div class="small text-muted">Ready to start</div>
                </div>
                {% endif %}
            </div>

            {% if run.error_message %}
            <div
                class="alert alert-danger d-flex align-items-center mt-3 mb-0 border-0 bg-danger bg-opacity-10 text-danger">
                <i class="bi bi-exclamation-triangle-fill me-2"></i>
                <div class="small">{{ run.error_message|truncatechars:100 }}</div>
            </div>
            {% endif %}
        </div>

        <div class="panel">
            <h6 class="text-uppercase text-muted small mb-3 ls-1">Hyperparameters</h6>

            <div class="d-flex justify-content-between mb-2 pb-2 border-bottom border-secondary border-opacity-10">
                <span class="text-muted">Input Dimension</span>
                <span class="text-white font-monospace">{{ run.input_dim }}</span>
            </div>

            <div class="d-flex justify-content-between mb-2 pb-2 border-bottom border-secondary border-opacity-10">
                <span class="text-muted">Expansion Factor</span>
                <span class="text-white font-monospace">{{ run.expansion_factor }}x</span>
            </div>

            <div class="d-flex justify-content-between mb-2 pb-2 border-bottom border-secondary border-opacity-10">
                <span class="text-muted">Latent Dimension</span>
                <span class="text-accent font-monospace">{% widthratio run.input_dim 1 run.expansion_factor %}</span>
            </div>

            <div class="d-flex justify-content-between mb-2 pb-2 border-bottom border-secondary border-opacity-10">
                <span class="text-muted">K-Sparsity</span>
                <span class="text-white font-monospace">{{ run.k_sparsity }}</span>
            </div>

            <div class="d-flex justify-content-between mb-2 pb-2 border-bottom border-secondary border-opacity-10">
                <span class="text-muted">Alpha Aux</span>
                <span class="text-white font-monospace">{{ run.alpha_aux }}</span>
            </div>

            <div class="d-flex justify-content-between mb-2 pb-2 border-bottom border-secondary border-opacity-10">
                <span class="text-muted">Learning Rate</span>
                <span class="text-white font-monospace">{{ run.learning_rate }}</span>
            </div>

            <div class="d-flex justify-content-between">
                <span class="text-muted">Epochs</span>
                <span class="text-white font-monospace">{{ run.epochs }}</span>
            </div>
        </div>
    </div>

    <div class="col-lg-8">

        {% if run.sparsity_heatmap %}
        <div class="panel p-0 overflow-hidden">
            <div
                class="p-4 border-bottom border-secondary border-opacity-10 d-flex justify-content-between align-items-center">
                <h6 class="text-uppercase text-muted small mb-0 ls-1">Latent Activation Analysis</h6>
                <div class="d-flex gap-2">
                    <a href="{{ run.sparsity_heatmap.url }}" download class="btn btn-sm btn-ghost text-muted"
                        title="Download PNG">
                        <i class="bi bi-download"></i>
                    </a>
                    <button type="button" class="btn btn-sm btn-ghost text-muted" data-bs-toggle="modal"
                        data-bs-target="#heatmapModal" title="Expand">
                        <i class="bi bi-arrows-fullscreen"></i>
                    </button>
                </div>
            </div>

            <div class="p-4 text-center rounded-bottom position-relative" style="background-color: black;">
                <a href="#" data-bs-toggle="modal" data-bs-target="#heatmapModal" class="d-block">
                    <img src="{{ run.sparsity_heatmap.url }}" alt="Sparsity Matrix" class="img-fluid shadow-sm"
                        style="border: 1px solid #333;">
                </a>
                <p class="text-muted small mt-2 mb-0">
                    Vertical: Documents (Sampled) &middot; Horizontal: Top Active Features
                </p>
            </div>
        </div>

        <div class="modal fade" id="heatmapModal" tabindex="-1" aria-labelledby="heatmapModalLabel" aria-hidden="true">
            <div class="modal-dialog modal-xl modal-dialog-centered">
                <div class="modal-content bg-dark border-secondary">
                    <div class="modal-header border-secondary">
                        <h5 class="modal-title text-white" id="heatmapModalLabel">Sparsity Heatmap Analysis</h5>
                        <button type="button" class="btn-close btn-close-white" data-bs-dismiss="modal"
                            aria-label="Close"></button>
                    </div>

                    <div class="modal-body p-0 text-center" style="background-color: black;">
                        <img src="{{ run.sparsity_heatmap.url }}" class="img-fluid" alt="Full Sparsity Heatmap">
                    </div>

                    <div class="modal-footer border-secondary py-2">
                        <small class="text-muted me-auto">Run #{{ run.id }} &middot; k={{ run.k_sparsity }}</small>
                        <button type="button" class="btn btn-sm btn-secondary" data-bs-dismiss="modal">Close</button>
                    </div>
                </div>
            </div>
        </div>
        {% endif %}

        <div class="panel h-100 p-0 d-flex flex-column mb-4" style="max-height: 500px;" id="log-container">

            <div
                class="p-4 border-bottom border-secondary border-opacity-10 d-flex justify-content-between align-items-center sticky-top bg-panel">

                <h6 class="text-uppercase text-muted small mb-0 ls-1">Training Log</h6>
                {% if run.training_log %}
                <span class="badge bg-dark border border-secondary border-opacity-25 text-muted">
                    {{ run.training_log|length }} steps
                </span>
                {% endif %}
            </div>

            <div class="table-responsive flex-grow-1">
                <table class="table table-custom mb-0 align-middle w-100">
                    <thead class="bg-dark bg-opacity-50 sticky-top" style="top: 0;">
                        <tr>
                            <th class="ps-4" style="width: 20%;">Epoch</th>
                            <th style="width: 40%;">Loss Value</th>
                            <th class="text-end pe-4">Timestamp</th>
                        </tr>
                    </thead>
                    <tbody>
                        {% for entry in run.training_log %}
                        <tr>
                            <td class="ps-4">
                                <span class="text-white fw-medium">#{{ entry.epoch }}</span>
                            </td>
                            <td>
                                <div class="d-flex align-items-center">
                                    <span class="font-monospace text-accent me-3">{{ entry.loss|floatformat:5 }}</span>
                                    <div class="progress flex-grow-1"
                                        style="height: 4px; max-width: 100px; background: #333;">
                                        <div class="progress-bar bg-accent" role="progressbar"
                                            style="width: {% widthratio entry.loss 1 20 %}%"></div>
                                    </div>
                                </div>
                            </td>
                            <td class="text-end pe-4 text-muted font-monospace small">
                                {{ entry.timestamp|date:"H:i:s" }}
                            </td>
                        </tr>
                        {% empty %}
                        <tr>
                            <td colspan="3" class="text-center py-5">
                                <div class="mb-3 text-muted opacity-25">
                                    <i class="bi bi-terminal" style="font-size: 2rem;"></i>
                                </div>
                                <p class="text-muted small mb-0">
                                    {% if run.status == 'queued' %}
                                    Waiting to start training...
                                    {% else %}
                                    No logs available yet.
                                    {% endif %}
                                </p>
                            </td>
                        </tr>
                        {% endfor %}
                    </tbody>
                </table>
            </div>
        </div>



    </div>
</div>
{% endblock %}