"""Internal Maintainer/Research Probe profile catalog.

These are operator diagnostics, not fork-facing product configuration.
"""
from . import maintainer_probe as maintainer_probe_backend
from . import research_probe as research_probe_backend

MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_CONV = maintainer_probe_backend.FULL_SFT_30S_STAGE2_CONV_PROFILE
MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_DIFFUSION = maintainer_probe_backend.FULL_SFT_30S_STAGE2_DIFFUSION_PROFILE
MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_CONV_FRAMEWISE = maintainer_probe_backend.FULL_SFT_30S_STAGE2_CONV_FRAMEWISE_PROFILE
MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_LATENT = maintainer_probe_backend.FULL_SFT_30S_STAGE2_LATENT_PROFILE
MAINTAINER_PROBE_FULL_SFT_30S_UPSAMPLE = maintainer_probe_backend.FULL_SFT_30S_UPSAMPLE_PROFILE
MAINTAINER_PROBE_FULL_SFT_30S_FLF2V_STAGE1 = maintainer_probe_backend.FULL_SFT_30S_FLF2V_STAGE1_PROFILE
MAINTAINER_PROBE_FULL_SFT_30S_STAGE1 = maintainer_probe_backend.FULL_SFT_30S_STAGE1_PROFILE

# P35 long-FLF2V structural gate: exercise the exact 721f Start+End condition-pipeline
# Stage 1 with two steps only. Long Stage 2/decode paths are already closed by P28/P33.
MAINTAINER_PROBE_CURRENT_GATES = (MAINTAINER_PROBE_FULL_SFT_30S_FLF2V_STAGE1,)
MAINTAINER_PROBE_CHOICES = [
    ("Current P35 gate · Long FLF2V · 30s Stage 1 structural scout", MAINTAINER_PROBE_FULL_SFT_30S_FLF2V_STAGE1),
    ("Closed P34 witness · Probe Progress smoke · 30s latent upsample only", MAINTAINER_PROBE_FULL_SFT_30S_UPSAMPLE),
    ("Closed P33 witness · Full/SFT 30s · Stage 2 release → Diffusion Decoder", MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_DIFFUSION),
    ("Closed P28 witness · Full/SFT 30s · Stage 2 release → Conv VAE", MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_CONV),
    ("Fallback · Full/SFT 30s · Stage 2 release → framewise Conv VAE", MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_CONV_FRAMEWISE),
    ("Diagnostic · Full/SFT 30s · Stage 2 latent-only", MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_LATENT),
    ("Closed witness rerun · Full/SFT 30s Stage 1 · 5-step", MAINTAINER_PROBE_FULL_SFT_30S_STAGE1),
]


def profile_description(profile_id) -> str:
    selected = str(profile_id or "")
    if selected == MAINTAINER_PROBE_FULL_SFT_30S_FLF2V_STAGE1:
        return (
            "**Current P35 gate · Long FLF2V 30s structural scout**  \n"
            "Full/SFT · Blue bird Start+End · 512×512 · 721f · seed 42 · no user LoRA. "
            "Runs only **two Stage-1 denoising steps** through `LTX2ConditionPipeline` with Start index `0` and End index `-1`; "
            "no latent upsample, Stage 2, decoder, MP4, or visual-quality claim. This isolates the only long-FLF2V-specific stage before a later product-quality run. "
            f"Reservation: **{maintainer_probe_backend.FULL_SFT_30S_FLF2V_STAGE1_ZEROGPU_DURATION_SECONDS}s**."
        )
    if selected == MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_DIFFUSION:
        return f"**Closed P33 witness · 721f Stage 2 + Diffusion Decoder**  \nFull/SFT · I2V · Cats · 512×512 · 721f · seed 42 · no user LoRA. P33 LIVE-pass済みの再現用profileです。Two Stage-1 steps create the exact 721f latent shape; real Stage 2 + Diffusion Decoder runs, with no 30-step Stage 1 or MP4 encode. Reservation: **{maintainer_probe_backend.FULL_SFT_30S_STAGE2_DIFFUSION_ZEROGPU_DURATION_SECONDS}s**."
    if selected == MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_CONV:
        return (
            "**Closed P28 witness rerun · 721f Stage 2 + Conv VAE measurement**  \n"
            "Fixed contract: **Full/SFT · I2V · Cats · 512×512 · 721f · seed 42 · no user LoRA**. "
            "Two Stage-1 steps create the exact temporal latent shape, then the real ×2 latent upsampler and internal distilled-LoRA Stage 2 run. Stage 2 returns latents, the internal adapter is released, earlier latent references are dropped, and only then the normal Conv VAE/audio decode runs. P28R4 already LIVE-passed this profile; it is retained for reproducibility. "
            f"Reservation: **{maintainer_probe_backend.FULL_SFT_30S_STAGE2_CONV_ZEROGPU_DURATION_SECONDS}s**. It does not repeat the expensive 30-step Stage 1 and does not MP4-encode."
        )
    if selected == MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_CONV_FRAMEWISE:
        return (
            "**Fallback diagnostic · post-Stage2 release + framewise Conv VAE**  \n"
            "Same 721f two-step preparation and real Stage 2, but the Conv VAE temporarily enables pinned LTX2 temporal framewise decoding after Stage-2 adapter release. This is not the current product gate; use it only if the product-equivalent release-before-decode gate still OOMs. "
            f"Reservation: **{maintainer_probe_backend.FULL_SFT_30S_STAGE2_CONV_FRAMEWISE_ZEROGPU_DURATION_SECONDS}s**."
        )
    if selected == MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_LATENT:
        return (
            "**Optional diagnostic · 721f Stage 2 latent-only**  \n"
            "Same two-step Stage-1 + ×2 upsample preparation, then the real internal distilled-LoRA Stage 2 returns latents instead of decoding. "
            f"Reservation: **{maintainer_probe_backend.FULL_SFT_30S_STAGE2_LATENT_ZEROGPU_DURATION_SECONDS}s**. Use this only if the Conv-VAE gate needs Stage-2/decode isolation."
        )
    if selected == MAINTAINER_PROBE_FULL_SFT_30S_UPSAMPLE:
        return (
            "**Closed P34 witness · Probe Progress visibility smoke**  \n"
            "Reuses the closed 721f two-step latent-upsample witness only. Stage 1 now emits explicit step callbacks and the event pins Gradio Progress to this status surface. "
            "Confirm that progress visibly advances while the Probe runs; no Stage 2, decoder, MP4, or visual-quality judgment is required. "
            f"Reservation: **{maintainer_probe_backend.FULL_SFT_30S_UPSAMPLE_ZEROGPU_DURATION_SECONDS}s**."
        )
    if selected == MAINTAINER_PROBE_FULL_SFT_30S_STAGE1:
        return (
            "**Closed witness rerun · Stage-1 timing scout**  \n"
            "Full/SFT · I2V · Cats · 512×512 · 721f · 5 Stage-1 steps · seed 42 · no user LoRA · latent output only. "
            f"Reservation: **{maintainer_probe_backend.FULL_SFT_30S_STAGE1_ZEROGPU_DURATION_SECONDS}s**. This profile is retained for reproducibility; it is not the current gate."
        )
    return "Select a Probe profile. No Probe runs until **Run Probe** is pressed."

RESEARCH_PROBE_CHOICES = [
    ("Research · 30s Stage 1 · 8-step timing curve", research_probe_backend.FULL_SFT_30S_STAGE1_8STEP_PROFILE),
    ("Research · 30s Stage 1 · 12-step timing curve", research_probe_backend.FULL_SFT_30S_STAGE1_12STEP_PROFILE),
    ("Research · 25s Stage 1 · 5-step timing curve", research_probe_backend.FULL_SFT_25S_STAGE1_5STEP_PROFILE),
]