stations.py

"""Deterministic frequency → station mapping.

Each station defines:
- system_prompt: compact persona + opening anchor (the fine-tune already knows
  the style; short prompts = fast first token on CPU)
- user_prompt: the "on air" cue in the station's native language
- voice: animalese voice parameters (pitch, jitter, waveform)
- theme: visual identity (display glow color, flicker rhythm)
"""

STATION_TOLERANCE = 0.3  # MHz — within ±0.3 counts as "on station"
LOCK_TOLERANCE = 0.05      # MHz — perfect tuning
NEAR_TOLERANCE = 0.45    # MHz — beyond this: static only

USER_PROMPT_EN = "Write tonight's transmission. On-air script only."

# No verbose "rules" in the system prompt: the fine-tune already knows the
# radio format (markers, length, the [FIN DE TRANSMISION] sign-off), and a 1B
# model recites instruction-shaped text on air. We only give it the persona +
# a language cue + the opening line it should start from (that opening is the
# strongest, leak-safe steering lever — if echoed, it is correct output).
def _prompt_es(persona: str, apertura: str) -> str:
    return f"{persona}\nEscribes el guion al aire en español. Empiezas con: {apertura}"


def _prompt_en(persona: str, opening: str) -> str:
    return f"{persona}\nYou write the on-air script in English. You begin with: {opening}"


# Language directives: the listener picks the broadcast language.
# Short and declarative on purpose — a verbose "translate... example: [JINGLE]
# ..." block is exactly the kind of instruction-shaped text a 1B model recites
# on air. The fine-tune now covers ES/EN/FR, so one clean line is enough; the
# swapped opening + the user prompt do the heavy steering.
LANG_DIRECTIVES = {
    "es": "Hablas y escribes solo en español, nunca en inglés.",
    "en": "You speak and write only in English, never in Spanish. Translate every line into English.",
    "fr": "Tu parles et écris uniquement en français, jamais en espagnol. Traduis chaque ligne en français.",
}

# On-air cue per language: this is the last thing the model reads, so it is
# the strongest lever to pull the generation into the requested language.
USER_PROMPTS = {
    "es": "Escribe la transmisión de esta noche. Solo el guion al aire.",
    "en": "Write tonight's transmission, entirely in English. On-air script only.",
    "fr": "Écris la transmission de ce soir, entièrement en français. Uniquement le script à l'antenne.",
}

SUPPORTED_LANGS = tuple(LANG_DIRECTIVES.keys())


def normalize_lang(lang: str) -> str:
    lang = str(lang or "es").lower()[:2]
    return lang if lang in LANG_DIRECTIVES else "es"


def localized_system_prompt(base_prompt: str, native_lang: str, lang: str) -> str:
    """When the listener asks for another language, wrap the persona with the
    translation directive (before and after: small models obey anchors)."""
    lang = normalize_lang(lang)
    if lang == native_lang:
        return base_prompt
    directive = LANG_DIRECTIVES[lang]
    return f"{directive} {base_prompt} {directive}"


def localized_user_prompt(station: dict | None, lang: str) -> str:
    """On-air cue in the requested language (falls back to the station's own)."""
    lang = normalize_lang(lang)
    native = (station or {}).get("lang", "es")
    if lang == native and station and station.get("user_prompt"):
        return station["user_prompt"]
    return USER_PROMPTS[lang]


# Same station frequency → same broadcast for every listener.
STATIONS: dict[float, dict] = {
    87.6: {
        "name": "Night Drive AM",
        "lang": "en",
        "system_prompt": _prompt_en(
            persona=(
                "You are the host of Night Drive AM, a late-night call-in show on an "
                "empty highway between universes. Every caller tonight is, somehow, "
                "the same person calling from different years of their life. Calm, "
                "warm, slightly melancholic."
            ),
            opening="[JINGLE] You're on Night Drive. Go ahead, caller.",
        ),
        "user_prompt": USER_PROMPT_EN,
        "voice": {"pitch_base": 110, "pitch_step": 4, "pitch_jitter": 12, "waveform": "sine"},
        "theme": {"glow": "#ffb45e", "flicker": 0.25},
    },
    88.9: {
        "name": "Radio Selenita — Tropical lunar",
        "lang": "es",
        "system_prompt": _prompt_es(
            persona=(
                "Eres el animador de Radio Selenita, la emisora tropical de un "
                "Medellín que queda en la Luna: saludos con sabrosura paisa, cumbias, "
                "cráteres, gravedad baja y vista a la Tierra."
            ),
            apertura=(
                "[JINGLE] ¡Quiubo pues, mi gente selenita! Desde el cráter de la 70 "
                "les saluda su locutor de confianza."
            ),
        ),
        "voice": {"pitch_base": 260, "pitch_step": 8, "pitch_jitter": 30, "waveform": "triangle"},
        "theme": {"glow": "#7cfc9a", "flicker": 0.5},
    },
    91.7: {
        "name": "WFifty — Locutor de los 50",
        "lang": "es",
        "system_prompt": _prompt_es(
            persona=(
                "Eres Don Aurelio, locutor estrella de una radio AM de los años 50 en "
                "un universo paralelo. Narras EN VIVO, con voz cálida y melodramática, "
                "la final municipal de ajedrez entre los gatos Bartolo y Misifú."
            ),
            apertura=(
                "[JINGLE] Buenas noches, queridos oyentes. Les habla Don Aurelio "
                "desde el salón municipal."
            ),
        ),
        "voice": {"pitch_base": 145, "pitch_step": 5, "pitch_jitter": 18, "waveform": "sawtooth"},
        "theme": {"glow": "#ffd27c", "flicker": 0.35},
    },
    94.1: {
        "name": "The Midnight Archive",
        "lang": "en",
        "system_prompt": _prompt_en(
            persona=(
                "You are the announcer of The Midnight Archive, a BBC-style service "
                "that reads catalogue entries for objects lost between universes "
                "(umbrellas that remember rain, keys to doors never built). Measured, "
                "elegant, faintly haunted."
            ),
            opening="[JINGLE] This is The Midnight Archive.",
        ),
        "user_prompt": USER_PROMPT_EN,
        "voice": {"pitch_base": 170, "pitch_step": 5, "pitch_jitter": 10, "waveform": "sine"},
        "theme": {"glow": "#9ad8ff", "flicker": 0.2},
    },
    97.3: {
        "name": "Júpiter FM — Boletín meteorológico",
        "lang": "es",
        "system_prompt": _prompt_es(
            persona=(
                "Eres la voz oficial del Servicio Meteorológico de Júpiter, año 2187: "
                "tormentas de amoniaco, vientos de 900 km/h, la Gran Mancha Roja de "
                "mal humor. Tono serio, casi poético."
            ),
            apertura="[JINGLE] Servicio Meteorológico de Júpiter, parte de las 19 horas.",
        ),
        "voice": {"pitch_base": 340, "pitch_step": 7, "pitch_jitter": 28, "waveform": "square"},
        "theme": {"glow": "#ff8a5e", "flicker": 0.4},
    },
    99.9: {
        "name": "La Hora Exacta",
        "lang": "es",
        "system_prompt": _prompt_es(
            persona=(
                "Eres la voz automática de La Hora Exacta, la estación que da la hora "
                "en un universo donde el tiempo se rompió. Anuncias horas imposibles "
                "con seriedad burocrática, como un reloj parlante averiado."
            ),
            apertura="[JINGLE] Al tercer tono serán las veinticinco horas del martes pasado. ...Piiip.",
        ),
        "voice": {"pitch_base": 420, "pitch_step": 3, "pitch_jitter": 6, "waveform": "square"},
        "theme": {"glow": "#c6ff5e", "flicker": 0.15},
    },
    101.3: {
        "name": "Deep Sea Lounge",
        "lang": "en",
        "system_prompt": _prompt_en(
            persona=(
                "You are the smooth-jazz DJ of Deep Sea Lounge, broadcasting from a "
                "glass studio at the bottom of an ocean on a gas planet. Velvet voice, "
                "unhurried, dedicating songs to creatures drifting past the window."
            ),
            opening="[JINGLE] Mmm, welcome back to the Deep Sea Lounge.",
        ),
        "user_prompt": USER_PROMPT_EN,
        "voice": {"pitch_base": 95, "pitch_step": 4, "pitch_jitter": 14, "waveform": "sine"},
        "theme": {"glow": "#6ec8ff", "flicker": 0.18},
    },
    103.5: {
        "name": "Nube 103 — Comercial interdimensional",
        "lang": "es",
        "system_prompt": _prompt_es(
            persona=(
                "Eres el locutor publicitario de Nube 103, en un universo donde las "
                "nubes se alquilan por hora. Vendes el plan 'Cumulonimbus Premium' "
                "(sombra garantizada, llovizna opcional, truenos de cortesía) con "
                "entusiasmo desbordado, estilo años 60."
            ),
            apertura="[JINGLE] ¡Amigos! ¿Cansados de cielos vacíos?",
        ),
        "voice": {"pitch_base": 230, "pitch_step": 6, "pitch_jitter": 22, "waveform": "square"},
        "theme": {"glow": "#ffe97c", "flicker": 0.45},
    },
    106.1: {
        "name": "Cocina Imposible",
        "lang": "es",
        "system_prompt": _prompt_es(
            persona=(
                "Eres la chef de Cocina Imposible: explicas paso a paso, con cariño "
                "de abuela, recetas de ingredientes que no existen (sopa de ecos, pan "
                "de sombra, mermelada de domingos)."
            ),
            apertura="[JINGLE] Bienvenidos, mis amores, a Cocina Imposible.",
        ),
        "voice": {"pitch_base": 290, "pitch_step": 6, "pitch_jitter": 24, "waveform": "triangle"},
        "theme": {"glow": "#ffb0d8", "flicker": 0.3},
    },
}

STATION_FREQUENCIES = sorted(STATIONS.keys())

# ------------------------------------------------------------------
# Special frequencies (app.py routes them before regular stations)
# ------------------------------------------------------------------
CURSED_FREQUENCY = 95.5  # the unsolvable station: it always arrives corrupted

CURSED_SYSTEM_PROMPT = (
    "Eres una voz que transmite desde un lugar que ya no existe. Hablas en español, "
    "en fragmentos inconexos y bellos: mitades de frases, nombres de personas que "
    "nadie recuerda, coordenadas de sitios que no aparecen en los mapas, despedidas "
    "sin destinatario. Nunca explicas nada. Nunca completas la idea. "
    "Entre 50 y 80 palabras. Cierra con [FIN DE TRANSMISION]."
)

CURSED_VOICE = {"pitch_base": 80, "pitch_step": 3, "pitch_jitter": 40, "waveform": "sine"}
CURSED_THEME = {"glow": "#b9a4ff", "flicker": 0.8}

ENCRYPTED_VOICE = {"pitch_base": 600, "pitch_step": 0, "pitch_jitter": 0, "waveform": "sine"}
ENCRYPTED_THEME = {"glow": "#58ff8f", "flicker": 0.6}


def station_seed(frequency: float) -> int:
    """Deterministic per-frequency seed: same frequency → same broadcast."""
    return int(round(frequency * 10)) * 7919 % (2**31 - 1)


def nearest_station_frequency(frequency: float) -> float | None:
    """Canonical frequency of the nearest station, if within range."""
    if not STATION_FREQUENCIES:
        return None
    nearest = min(STATION_FREQUENCIES, key=lambda f: abs(frequency - f))
    if abs(frequency - nearest) > STATION_TOLERANCE:
        return None
    return nearest


def get_station(frequency: float) -> dict | None:
    """Returns the station when the frequency is within ±STATION_TOLERANCE."""
    nearest = nearest_station_frequency(frequency)
    if nearest is None:
        return None
    return STATIONS[nearest]


def signal_strength(frequency: float) -> float:
    """0 = no signal, 1 = perfect tuning (relative to the nearest station)."""
    nearest = nearest_station_frequency(frequency)
    if nearest is None:
        # Very weak signal when a station is close but outside tuning tolerance
        if not STATION_FREQUENCIES:
            return 0.0
        nearest_any = min(STATION_FREQUENCIES, key=lambda f: abs(frequency - f))
        dist = abs(frequency - nearest_any)
        if dist >= NEAR_TOLERANCE:
            return 0.0
        return max(0.0, 0.15 * (1 - dist / NEAR_TOLERANCE))
    dist = abs(frequency - nearest)
    if dist <= LOCK_TOLERANCE:
        return 1.0
    return max(0.0, 1.0 - (dist - LOCK_TOLERANCE) / (NEAR_TOLERANCE - LOCK_TOLERANCE))


# Opening lines per language. The opening is the strongest language anchor a
# 1B model has: swapping it pulls the whole broadcast into the target language.
STATION_OPENINGS = {
    87.6: {
        "en": "[JINGLE] You're on Night Drive. Go ahead, caller.",
        "es": "[JINGLE] Estás en Night Drive. Adelante, oyente.",
        "fr": "[JINGLE] Vous êtes sur Night Drive. Allez-y, on vous écoute.",
    },
    88.9: {
        "es": "[JINGLE] ¡Quiubo pues, mi gente selenita! Desde el cráter de la 70 les saluda su locutor de confianza.",
        "en": "[JINGLE] What's up, my lunar people! Greeting you from Crater 70, it's your trusted host.",
        "fr": "[JINGLE] Salut la famille sélénite ! Depuis le cratère de la 70, votre animateur préféré vous salue.",
    },
    91.7: {
        "es": "[JINGLE] Buenas noches, queridos oyentes. Les habla Don Aurelio desde el salón municipal.",
        "en": "[JINGLE] Good evening, dear listeners. This is Don Aurelio, live from the municipal hall.",
        "fr": "[JINGLE] Bonsoir, chers auditeurs. Ici Don Aurelio, en direct de la salle municipale.",
    },
    94.1: {
        "en": "[JINGLE] This is The Midnight Archive.",
        "es": "[JINGLE] Esto es El Archivo de Medianoche.",
        "fr": "[JINGLE] Ici Les Archives de Minuit.",
    },
    97.3: {
        "es": "[JINGLE] Servicio Meteorológico de Júpiter, parte de las 19 horas.",
        "en": "[JINGLE] Jupiter Weather Service, 7 p.m. bulletin.",
        "fr": "[JINGLE] Service météorologique de Jupiter, bulletin de 19 heures.",
    },
    99.9: {
        "es": "[JINGLE] Al tercer tono serán las veinticinco horas del martes pasado. ...Piiip.",
        "en": "[JINGLE] At the third tone, the time will be twenty-five o'clock last Tuesday. ...Beep.",
        "fr": "[JINGLE] Au troisième bip, il sera vingt-cinq heures mardi dernier. ...Biiip.",
    },
    101.3: {
        "en": "[JINGLE] Mmm, welcome back to the Deep Sea Lounge.",
        "es": "[JINGLE] Mmm, bienvenidos de nuevo al Deep Sea Lounge.",
        "fr": "[JINGLE] Mmm, bienvenue au Deep Sea Lounge.",
    },
    103.5: {
        "es": "[JINGLE] ¡Amigos! ¿Cansados de cielos vacíos?",
        "en": "[JINGLE] Friends! Tired of empty skies?",
        "fr": "[JINGLE] Chers amis ! Fatigués des ciels vides ?",
    },
    106.1: {
        "es": "[JINGLE] Bienvenidos, mis amores, a Cocina Imposible.",
        "en": "[JINGLE] Welcome, my darlings, to Impossible Kitchen.",
        "fr": "[JINGLE] Bienvenue, mes chéris, dans Cuisine Impossible.",
    },
}


def station_system_prompt(canonical: float, lang: str) -> str:
    """Persona prompt with the opening swapped into the requested language."""
    station = STATIONS[canonical]
    base = station["system_prompt"]
    native = station.get("lang", "es")
    lang = normalize_lang(lang)
    openings = STATION_OPENINGS.get(canonical, {})
    if lang != native and openings.get(native) and openings.get(lang):
        base = base.replace(openings[native], openings[lang])
    return localized_system_prompt(base, native, lang)


# Station names per UI language (brand names stay, descriptors translate).
STATION_NAMES = {
    87.6: {"es": "Night Drive AM", "en": "Night Drive AM", "fr": "Night Drive AM"},
    88.9: {
        "es": "Radio Selenita — Tropical lunar",
        "en": "Radio Selenita — Lunar Tropical",
        "fr": "Radio Sélénite — Tropicale lunaire",
    },
    91.7: {
        "es": "WFifty — Locutor de los 50",
        "en": "WFifty — 1950s Announcer",
        "fr": "WFifty — Speaker des années 50",
    },
    94.1: {
        "es": "The Midnight Archive",
        "en": "The Midnight Archive",
        "fr": "The Midnight Archive",
    },
    97.3: {
        "es": "Júpiter FM — Boletín meteorológico",
        "en": "Jupiter FM — Weather Bulletin",
        "fr": "Jupiter FM — Bulletin météo",
    },
    99.9: {"es": "La Hora Exacta", "en": "The Exact Hour", "fr": "L'Heure Exacte"},
    101.3: {"es": "Deep Sea Lounge", "en": "Deep Sea Lounge", "fr": "Deep Sea Lounge"},
    103.5: {
        "es": "Nube 103 — Comercial interdimensional",
        "en": "Cloud 103 — Interdimensional Commercials",
        "fr": "Nuage 103 — Pub interdimensionnelle",
    },
    106.1: {"es": "Cocina Imposible", "en": "Impossible Kitchen", "fr": "Cuisine Impossible"},
}

# CRT portrait of each frequency's announcer (served from /static/img).
PORTRAITS = {
    87.6: "/static/img/st_876.webp",
    88.9: "/static/img/st_889.webp",
    91.7: "/static/img/st_917.webp",
    94.1: "/static/img/st_941.webp",
    97.3: "/static/img/st_973.webp",
    99.9: "/static/img/st_999.webp",
    101.3: "/static/img/st_1013.webp",
    103.5: "/static/img/st_1035.webp",
    106.1: "/static/img/st_1061.webp",
}


def station_catalog() -> dict:
    """Public station metadata (voice, name, visual theme) for the frontend."""
    from decrypt import ENCRYPTED_FREQUENCY

    catalog = {
        str(freq): {
            "name": data["name"],
            "names": STATION_NAMES.get(freq, {"es": data["name"]}),
            "lang": data.get("lang", "es"),
            "voice": data["voice"],
            "theme": data.get("theme", {"glow": "#7cfc00", "flicker": 0.3}),
            "portrait": PORTRAITS.get(freq),
        }
        for freq, data in STATIONS.items()
    }
    catalog[str(ENCRYPTED_FREQUENCY)] = {
        "name": "· – · · –",
        "names": {"es": "· – · · –", "en": "· – · · –", "fr": "· – · · –"},
        "lang": "es",
        "voice": ENCRYPTED_VOICE,
        "theme": ENCRYPTED_THEME,
        "portrait": "/static/img/st_1047.webp",
        "encrypted": True,
    }
    catalog[str(CURSED_FREQUENCY)] = {
        "name": "· · · ?",
        "names": {"es": "· · · ?", "en": "· · · ?", "fr": "· · · ?"},
        "lang": "es",
        "voice": CURSED_VOICE,
        "theme": CURSED_THEME,
        "portrait": "/static/img/st_955.webp",
        "cursed": True,
    }
    return catalog