{% extends "base.html" %}
{% from "_page_header.html" import page_header %}
{% from "_components.html" import badge, field %}
{% block content %}
{{ page_header("Query Lab", "Run a query through the production retrieval pipeline and see why each chunk was found and why it ranks where it does: authorization, dense and sparse candidates, RRF fusion, context assembly, answerability and per-stage timings. Scores are ranking signals, not probabilities.") }}

<div class="grid grid-cols-1 xl:grid-cols-3 gap-md items-start">
  <section aria-labelledby="query-form-heading" class="glass-panel rounded-lg p-lg xl:col-span-1 xl:sticky xl:top-[76px]">
    <h2 id="query-form-heading" class="text-headline-md font-headline-md text-on-surface mb-md">Query</h2>
    {% if form_error %}
    <p role="alert" class="text-body-md font-body-md text-error bg-error-container/20 border-l-4 border-error rounded-DEFAULT p-md mb-md">{{ form_error }}</p>
    {% endif %}
    {% if not catalog.available %}
    <p role="status" class="text-body-sm font-body-sm text-on-surface-variant bg-surface-variant/40 border border-[#1E293B] rounded-DEFAULT p-sm mb-md">Selector lists could not be loaded from the control plane; use the Advanced fields below to target spaces, sources or documents by ID.</p>
    {% endif %}
    <form method="post" action="/admin/query-lab/run" class="flex flex-col gap-md">
      {% call field("query", "Query text") %}
        <textarea id="query" name="query" required maxlength="8000" rows="4" class="input-dark w-full rounded-DEFAULT py-sm px-md text-body-md font-body-md">{{ trace.query if trace else "" }}</textarea>
      {% endcall %}
      <div class="grid grid-cols-2 gap-md">
        {% call field("profile", "Retrieval profile") %}
          <select id="profile" name="profile" class="input-dark w-full rounded-DEFAULT py-sm px-md text-body-md font-body-md">
            {% for value, label in profiles %}<option value="{{ value }}" {{ "selected" if (trace.profile if trace else "hybrid-v1") == value else "" }}>{{ value }}</option>{% endfor %}
          </select>
        {% endcall %}
        {% call field("max_results", "Max results") %}
          <input id="max_results" name="max_results" type="number" min="1" max="50" value="10" class="input-dark w-full rounded-DEFAULT py-sm px-md text-body-md font-body-md">
        {% endcall %}
      </div>

      {% call field("space_ids", "Knowledge spaces", "Only spaces you are granted are ever searched; leave empty for all granted spaces.") %}
        <select id="space_ids" name="space_ids" multiple size="{{ [catalog.spaces | length, 5] | min if catalog.spaces else 2 }}" class="input-dark w-full rounded-DEFAULT py-sm px-md text-body-md font-body-md">
          {% for space_id, name in catalog.spaces %}<option value="{{ space_id }}" {{ "selected" if (space_id | string) in selected.get("space_ids", []) else "" }}>{{ name }}</option>{% else %}<option disabled>No spaces available</option>{% endfor %}
        </select>
      {% endcall %}

      {% call field("source_ids", "Sources", "Restrict to documents of these sources.") %}
        <select id="source_ids" name="source_ids" multiple size="{{ [catalog.sources | length, 5] | min if catalog.sources else 2 }}" class="input-dark w-full rounded-DEFAULT py-sm px-md text-body-md font-body-md">
          {% for source_id, name, space_name in catalog.sources %}<option value="{{ source_id }}" {{ "selected" if (source_id | string) in selected.get("source_ids", []) else "" }}>{{ name }}{% if space_name %} — {{ space_name }}{% endif %}</option>{% else %}<option disabled>No sources registered</option>{% endfor %}
        </select>
      {% endcall %}

      {% call field("document_ids", "Documents", ("Most recent " ~ (catalog.documents | length) ~ " documents shown; use Advanced for others.") if catalog.truncated_documents else "Restrict to specific documents.") %}
        <select id="document_ids" name="document_ids" multiple size="{{ [catalog.documents | length, 6] | min if catalog.documents else 2 }}" class="input-dark w-full rounded-DEFAULT py-sm px-md text-body-md font-body-md">
          {% for document_id, title, space_name in catalog.documents %}<option value="{{ document_id }}" {{ "selected" if (document_id | string) in selected.get("document_ids", []) else "" }}>{{ title }}{% if space_name %} — {{ space_name }}{% endif %}</option>{% else %}<option disabled>No documents yet</option>{% endfor %}
        </select>
      {% endcall %}

      <div class="grid grid-cols-1 md:grid-cols-3 gap-md">
        {% call field("languages", "Languages") %}
          <select id="languages" name="languages" multiple size="4" class="input-dark w-full rounded-DEFAULT py-sm px-md text-body-md font-body-md">
            {% for language in catalog.languages %}<option value="{{ language }}" {{ "selected" if language in selected.get("languages", []) else "" }}>{{ language }}</option>{% endfor %}
          </select>
        {% endcall %}
        {% call field("content_types", "Content types") %}
          <select id="content_types" name="content_types" multiple size="4" class="input-dark w-full rounded-DEFAULT py-sm px-md text-body-sm font-body-sm">
            {% for content_type in catalog.content_types %}<option value="{{ content_type }}" {{ "selected" if content_type in selected.get("content_types", []) else "" }}>{{ content_type }}</option>{% endfor %}
          </select>
        {% endcall %}
        {% call field("trust_levels", "Trust levels") %}
          <select id="trust_levels" name="trust_levels" multiple size="4" class="input-dark w-full rounded-DEFAULT py-sm px-md text-body-md font-body-md">
            {% for level in catalog.trust_levels %}<option value="{{ level }}" {{ "selected" if level in selected.get("trust_levels", []) else "" }}>{{ level }}</option>{% endfor %}
          </select>
        {% endcall %}
      </div>

      <details class="border border-[#1E293B] rounded-DEFAULT">
        <summary class="cursor-pointer px-md py-sm text-body-md font-headline-md text-on-surface">Advanced: raw identifiers</summary>
        <div class="p-md flex flex-col gap-md border-t border-[#1E293B]">
          <p class="text-body-sm font-body-sm text-on-surface-variant">Comma-separated UUIDs, merged with the selections above. Every filter is applied server-side by the same filter compiler <span class="text-code-sm font-code-sm">/v1/search</span> uses.</p>
          {% call field("requested_space_ids", "Knowledge-space IDs") %}
            <input id="requested_space_ids" name="requested_space_ids" value="{{ (selected.get('requested_space_ids') or [''])[0] }}" placeholder="uuid, uuid" autocomplete="off" class="input-dark w-full rounded-DEFAULT py-sm px-md text-code-sm font-code-sm">
          {% endcall %}
          {% call field("raw_source_ids", "Source IDs") %}
            <input id="raw_source_ids" name="raw_source_ids" value="{{ (selected.get('raw_source_ids') or [''])[0] }}" autocomplete="off" class="input-dark w-full rounded-DEFAULT py-sm px-md text-code-sm font-code-sm">
          {% endcall %}
          {% call field("raw_document_ids", "Document IDs") %}
            <input id="raw_document_ids" name="raw_document_ids" value="{{ (selected.get('raw_document_ids') or [''])[0] }}" autocomplete="off" class="input-dark w-full rounded-DEFAULT py-sm px-md text-code-sm font-code-sm">
          {% endcall %}
        </div>
      </details>

      <button type="submit" class="btn-primary rounded-DEFAULT py-sm px-md text-body-sm font-headline-md flex items-center justify-center gap-xs">
        <span class="material-symbols-outlined text-[18px]" aria-hidden="true">play_arrow</span>
        Run query
      </button>
    </form>
  </section>

  <section id="lab-results" aria-labelledby="lab-results-heading" class="glass-panel rounded-lg p-lg xl:col-span-2">
    <h2 id="lab-results-heading" class="text-headline-md font-headline-md text-on-surface mb-md">Retrieval inspector</h2>
    {% if trace is none %}
    <p role="status" class="text-body-md font-body-md text-on-surface-variant">Run a query above to inspect the retrieval pipeline.</p>
    {% else %}

    {#- Summary -#}
    <div role="group" aria-label="Run summary" class="grid grid-cols-2 md:grid-cols-4 xl:grid-cols-6 gap-md bg-background border border-[#1E293B] rounded-DEFAULT p-md mb-md">
      <div>
        <p class="text-label-caps font-label-caps text-on-surface-variant uppercase tracking-wider mb-1">Answerability</p>
        {{ badge(trace.answerability.status | replace("_", " "), "ok" if trace.answerability.status == "answerable" else ("warn" if trace.answerability.status in ("partially_answerable", "conflicting_evidence") else "neutral")) }}
      </div>
      <div>
        <p class="text-label-caps font-label-caps text-on-surface-variant uppercase tracking-wider mb-1">Evidence count</p>
        <span class="text-body-md font-headline-md text-on-surface">{{ trace.answerability.evidence_count }}</span>
      </div>
      <div>
        <p class="text-label-caps font-label-caps text-on-surface-variant uppercase tracking-wider mb-1">Candidates</p>
        <span class="text-body-md font-headline-md text-on-surface">{{ trace.dense_candidates | length }} dense · {{ trace.sparse_candidates | length }} sparse</span>
      </div>
      <div>
        <p class="text-label-caps font-label-caps text-on-surface-variant uppercase tracking-wider mb-1">Fused</p>
        <span class="text-body-md font-headline-md text-on-surface">{{ trace.fusion.candidates | length }}{% if trace.fusion.suppressed_count %} <span class="text-on-surface-variant font-body-md">({{ trace.fusion.suppressed_count }} suppressed)</span>{% endif %}</span>
      </div>
      <div>
        <p class="text-label-caps font-label-caps text-on-surface-variant uppercase tracking-wider mb-1">Degraded</p>
        {{ badge("Yes", "error") if trace.degraded else badge("No", "ok") }}
      </div>
      <div>
        <p class="text-label-caps font-label-caps text-on-surface-variant uppercase tracking-wider mb-1">Latency</p>
        <span class="text-code-md font-code-md text-on-surface">{{ "%.1f" | format(trace.timings_ms.values() | sum) }} ms</span>
        <p class="text-body-sm font-body-sm text-on-surface-variant">profile {{ trace.profile }}</p>
      </div>
    </div>

    {% if trace.degraded %}
    <p role="alert" class="text-body-md font-body-md text-error bg-error-container/20 border-l-4 border-error rounded-DEFAULT p-md mb-md">
      <strong>Degraded run.</strong> Codes: {{ trace.degradation_codes | join(", ") if trace.degradation_codes else "none" }}
    </p>
    {% endif %}
    {% if not trace.fusion.candidates and not trace.degraded %}
    <p role="status" class="text-body-md font-body-md text-on-surface-variant bg-surface-variant/40 border border-[#1E293B] rounded-DEFAULT p-md mb-md">Zero results: no published chunk in the effective spaces matched the query and filters.</p>
    {% endif %}

    {#- Authorization & filters -#}
    <details open class="border border-[#1E293B] rounded-DEFAULT mb-sm">
      <summary class="cursor-pointer px-md py-sm text-body-md font-headline-md text-on-surface">Effective authorization &amp; filters</summary>
      <div class="p-md border-t border-[#1E293B] text-body-sm font-body-sm text-on-surface-variant space-y-1">
        <p><strong class="text-on-surface">Requested spaces:</strong> <span class="text-code-sm font-code-sm">{{ trace.effective_filters.requested_space_ids | join(", ") or "all granted" }}</span></p>
        <p><strong class="text-on-surface">Effective spaces (server-computed):</strong> <span class="text-code-sm font-code-sm">{{ trace.effective_filters.effective_space_ids | join(", ") or "none" }}</span></p>
        <p><strong class="text-on-surface">Denied / not granted:</strong> <span class="text-code-sm font-code-sm">{{ trace.effective_filters.denied_space_ids | join(", ") or "none" }}</span></p>
        <p><strong class="text-on-surface">Trust-required spaces:</strong> <span class="text-code-sm font-code-sm">{{ trace.effective_filters.trust_required_space_ids | join(", ") or "none" }}</span></p>
        <p><strong class="text-on-surface">Content types applied:</strong> {{ trace.effective_filters.applied_content_types | join(", ") or "none" }}</p>
        <p><strong class="text-on-surface">Languages applied:</strong> {{ trace.effective_filters.applied_languages | join(", ") or "none" }}</p>
        <p>Source, document and trust-level filters are applied server-side in the same Qdrant filter as the space scope.</p>
      </div>
    </details>

    {#- Candidates -#}
    <details class="border border-[#1E293B] rounded-DEFAULT mb-sm">
      <summary class="cursor-pointer px-md py-sm text-body-md font-headline-md text-on-surface">Candidates — dense ({{ trace.dense_candidates | length }})</summary>
      <div class="border-t border-[#1E293B]">
        {% set candidate_list = trace.dense_candidates %}{% set fused = false %}
        {% include "query_lab/_candidate_table.html" %}
      </div>
    </details>
    <details class="border border-[#1E293B] rounded-DEFAULT mb-sm">
      <summary class="cursor-pointer px-md py-sm text-body-md font-headline-md text-on-surface">Candidates — sparse ({{ trace.sparse_candidates | length }})</summary>
      <div class="border-t border-[#1E293B]">
        {% set candidate_list = trace.sparse_candidates %}{% set fused = false %}
        {% include "query_lab/_candidate_table.html" %}
      </div>
    </details>

    {#- Fusion -#}
    <details open class="border border-[#1E293B] rounded-DEFAULT mb-sm">
      <summary class="cursor-pointer px-md py-sm text-body-md font-headline-md text-on-surface">Fusion — RRF ranking ({{ trace.fusion.candidates | length }}{% if trace.fusion.suppressed_count %}, {{ trace.fusion.suppressed_count }} suppressed as duplicate or over the per-document cap{% endif %})</summary>
      <div class="border-t border-[#1E293B]">
        <p class="px-md py-sm text-body-sm font-body-sm text-on-surface-variant">RRF score = Σ 1 / (k + rank) over the backends that returned the chunk. A chunk found by both backends outranks one found by only one at the same rank. This fused order is the real ranking: no reranker is active in this deployment.</p>
        {% set candidate_list = trace.fusion.candidates %}{% set fused = true %}
        {% include "query_lab/_candidate_table.html" %}
      </div>
    </details>

    {% if rerank_active %}
    <details class="border border-[#1E293B] rounded-DEFAULT mb-sm">
      <summary class="cursor-pointer px-md py-sm text-body-md font-headline-md text-on-surface">Reranking ({{ trace.rerank.state }})</summary>
      <div class="border-t border-[#1E293B]">
        {% if trace.rerank.degradation_code %}<p class="text-code-sm font-code-sm text-primary px-md pt-sm">Degradation code: {{ trace.rerank.degradation_code }}</p>{% endif %}
        {% set candidate_list = trace.rerank.candidates %}{% set fused = false %}
        {% include "query_lab/_candidate_table.html" %}
      </div>
    </details>
    {% endif %}

    {#- Context -#}
    <details class="border border-[#1E293B] rounded-DEFAULT mb-sm">
      <summary class="cursor-pointer px-md py-sm text-body-md font-headline-md text-on-surface">Context — assembly &amp; parent expansion</summary>
      <div class="p-md border-t border-[#1E293B] text-body-sm font-body-sm text-on-surface-variant space-y-1">
        <p>Token budget: <span class="text-code-sm font-code-sm">{{ trace.context.token_count }} / {{ trace.context.max_context_tokens }}</span> · Blocks: {{ trace.context.block_count }} · Citations in context: {{ trace.context.citation_ids | length }}</p>
        <p>Parent expansions applied: {{ trace.parent_expansion.expanded_count }} · Skipped: {{ trace.parent_expansion.skipped_count }} · Expansion tokens: {{ trace.parent_expansion.total_expansion_tokens }}</p>
      </div>
      <div class="overflow-x-auto border-t border-[#1E293B]">
        <table class="w-full text-left border-collapse">
          <caption class="sr-only">Citations included in the final context</caption>
          <thead class="bg-[#0F172A]">
            <tr>
              <th scope="col" class="py-sm px-md text-label-caps font-label-caps text-on-surface-variant uppercase tracking-wider">Provenance</th>
              <th scope="col" class="py-sm px-md text-label-caps font-label-caps text-on-surface-variant uppercase tracking-wider">Excerpt</th>
            </tr>
          </thead>
          <tbody class="divide-y divide-[#1E293B] text-body-md font-body-md text-on-surface">
            {% for citation in trace.citations %}
            <tr class="table-row-hover transition-colors align-top">
              <th scope="row" class="py-sm px-md font-body-md">
                <p class="text-on-surface">{{ citation.display or citation.document_title or ("Document " ~ (citation.document_id | string)[:8] ~ "…") }}</p>
                <p class="text-body-sm font-body-sm text-on-surface-variant">{% if citation.source_name %}{{ citation.source_name }} · {% endif %}<span class="text-code-sm font-code-sm" title="{{ citation.source_uri }}">{{ citation.source_uri }}</span></p>
                <p class="text-code-sm font-code-sm text-on-surface-variant" title="document {{ citation.document_id }} · chunk {{ citation.chunk_id }}">doc {{ (citation.document_id | string)[:8] }}… · chunk {{ (citation.chunk_id | string)[:8] }}…</p>
              </th>
              <td class="py-sm px-md text-body-sm font-body-sm">{{ citation.excerpt_preview }}{% if citation.excerpt_truncated %} &hellip; (truncated){% endif %}</td>
            </tr>
            {% else %}
            <tr><td colspan="2" class="py-sm px-md text-on-surface-variant">No citations.</td></tr>
            {% endfor %}
          </tbody>
        </table>
      </div>
    </details>

    {#- Answer -#}
    <details class="border border-[#1E293B] rounded-DEFAULT mb-sm">
      <summary class="cursor-pointer px-md py-sm text-body-md font-headline-md text-on-surface">Answer &amp; answerability</summary>
      <div class="p-md border-t border-[#1E293B] text-body-sm font-body-sm text-on-surface-variant space-y-1">
        <p>{{ badge(trace.answerability.status | replace("_", " "), "ok" if trace.answerability.status == "answerable" else "neutral") }} <span>(a category, not a confidence percentage)</span></p>
        <p><strong class="text-on-surface">Reason codes:</strong> {{ trace.answerability.reasons | join(", ") }}</p>
        <p>Evidence count: {{ trace.answerability.evidence_count }} &middot; Document diversity: {{ trace.answerability.document_diversity }} &middot; Source independence: {{ trace.answerability.source_independence }} &middot; Facet coverage: {{ "%.2f" | format(trace.answerability.facet_coverage_ratio) }} &middot; Thresholds version: {{ trace.answerability.thresholds_version }}</p>
        <p>Generation is not run by Query Lab: it inspects retrieval and answerability; <span class="text-code-sm font-code-sm">/v1/answers</span> produces grounded answers from the same context.</p>
      </div>
    </details>

    {#- Trace -#}
    <details class="border border-[#1E293B] rounded-DEFAULT mb-sm">
      <summary class="cursor-pointer px-md py-sm text-body-md font-headline-md text-on-surface">Trace — timings &amp; revisions</summary>
      <div class="p-md border-t border-[#1E293B] text-body-sm font-body-sm text-on-surface-variant">
        <h3 class="text-body-md font-headline-md text-on-surface mb-1">Phase timings</h3>
        <table class="text-code-sm font-code-sm mb-sm">
          <caption class="sr-only">Per-stage timings in milliseconds</caption>
          <tbody>
          {% for name, value in trace.timings_ms.items() %}
          <tr><th scope="row" class="text-left pr-md font-normal text-on-surface-variant">{{ name }}</th><td class="text-right tabular-nums text-on-surface">{{ "%.1f" | format(value) }} ms</td></tr>
          {% endfor %}
          </tbody>
        </table>
        <h3 class="text-body-md font-headline-md text-on-surface mb-1">Revisions</h3>
        <ul class="mb-sm space-y-0.5 text-code-sm font-code-sm">
          <li>Embedding profile: {{ trace.revisions.embedding_profile or "unavailable" }}</li>
          <li>Sparse profile: {{ trace.revisions.sparse_profile }}</li>
          <li>Collection alias: {{ trace.revisions.collection_alias or "unavailable" }}</li>
          {% if rerank_active %}<li>Reranker profile: {{ trace.revisions.reranker_profile or "unavailable" }}</li>{% endif %}
          <li>Answerability thresholds: {{ trace.revisions.answerability_thresholds_version }}</li>
        </ul>
        <p>Retrieval run ID: <span class="text-code-sm font-code-sm">{{ trace.retrieval_run_id }}</span></p>
      </div>
    </details>

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