{% extends "base.html" %}
{% block title %}Trainline — psyc{% endblock %}
{% block content %}
<section class="panel">
  <div class="panel-head">
    <h1>Trainline — Datasets</h1>
    <span class="count">{{ datasets|length }} dataset{{ '' if datasets|length == 1 else 's' }}</span>
  </div>
  {% if not datasets %}
    <p class="empty">No datasets yet. Run <code>psyc train-build-all</code>.</p>
  {% else %}
  <table class="cases">
    <thead>
      <tr><th>Dataset</th><th>Examples</th><th>Size</th><th>Built</th></tr>
    </thead>
    <tbody>
    {% for d in datasets %}
      <tr>
        <td><code>{{ d.name }}</code></td>
        <td>{{ d.examples }}</td>
        <td class="muted">{{ d.size_bytes }} B</td>
        <td class="muted">{{ d.modified[:16] }}</td>
      </tr>
    {% endfor %}
    </tbody>
  </table>
  {% endif %}
</section>

<section class="panel" style="margin-top: 20px;">
  <div class="panel-head">
    <h1>Trainline — Adapters</h1>
    <span class="count">{{ adapters|length }} adapter{{ '' if adapters|length == 1 else 's' }}</span>
  </div>
  {% if not adapters %}
    <p class="empty">No adapters yet. Build <code>Dockerfile.train</code> and run a QLoRA fine-tune.</p>
  {% else %}
  <div class="grid">
    {% for a in adapters %}
    <div class="card wide">
      <h2>{{ a.name }}{% if a.has_adapter %} <span class="outcome-badge outcome-actioned">trained</span>{% else %} <span class="outcome-badge outcome-rejected">incomplete</span>{% endif %}</h2>
      <dl>
        <dt>Base model</dt><dd><code>{{ a.base_model }}</code></dd>
        <dt>Examples</dt><dd>{{ a.examples }}</dd>
        <dt>Epochs</dt><dd>{{ a.epochs }}</dd>
        <dt>LoRA r</dt><dd>{{ a.lora_r }}</dd>
        <dt>Learning rate</dt><dd>{{ a.lr }}</dd>
        <dt>Final train loss</dt><dd>{% if a.train_loss is not none %}{{ '%.4f'|format(a.train_loss) }}{% else %}<span class="muted">— (trained before loss capture)</span>{% endif %}</dd>
        <dt>Datasets</dt><dd>{% for ds in a.datasets %}<code>{{ ds }}</code> {% endfor %}{% if not a.datasets %}—{% endif %}</dd>
      </dl>
      {% if a.loss_history %}
      <h3>Loss by step</h3>
      <table class="cases">
        <thead><tr><th>Step</th><th>Epoch</th><th>Loss</th></tr></thead>
        <tbody>
        {% for h in a.loss_history %}
          <tr><td>{{ h.step }}</td><td class="muted">{{ h.epoch }}</td><td>{{ '%.4f'|format(h.loss) }}</td></tr>
        {% endfor %}
        </tbody>
      </table>
      {% endif %}
    </div>
    {% endfor %}
  </div>
  {% endif %}
</section>
{% endblock %}