"""
Hugging Face Space: speak Italian → **Italian transcription + English translation**.

Whisper (Italian) + Marian IT→EN. Run locally: ``python app.py``
"""

from __future__ import annotations

import os
from pathlib import Path

os.environ.setdefault("ASR_REALTIME_MODE", "quality")
os.environ.setdefault("ASR_WHISPER_LANGUAGE", "italian")
os.environ.setdefault("ASR_TRANSLATE", "1")

import numpy as np
import gradio as gr

from italian_en_pipeline import ItalianEnglishPipeline

_SPACE_ROOT = Path(__file__).resolve().parent

_pipeline: ItalianEnglishPipeline | None = None


def _get_pipeline() -> ItalianEnglishPipeline:
    global _pipeline
    if _pipeline is None:
        _pipeline = ItalianEnglishPipeline(project_root=str(_SPACE_ROOT))
    return _pipeline


def transcribe(audio: tuple[int, np.ndarray] | None) -> tuple[str, str]:
    """Gradio Audio (numpy) → (italian, english)."""
    if audio is None:
        return "", ""
    sr, data = audio
    if data is None or len(data) == 0:
        return "", ""
    x = np.asarray(data, dtype=np.float32)
    if x.ndim > 1:
        x = x.mean(axis=-1)
    floats = x.reshape(-1).tolist()
    return _get_pipeline().transcribe_chunk(floats, int(sr))


with gr.Blocks(title="Italian speech → Italian + English") as demo:
    gr.Markdown(
        "### Italian → Italian + English\n"
        "Speak or upload **Italian** audio. Output is **recognized Italian** and **English** translation "
        "(Whisper + Marian). Optional fine-tuned Whisper in `models/whisper_finetuned_it/`."
    )
    audio_in = gr.Audio(
        sources=["microphone", "upload"],
        type="numpy",
        label="Audio (Italian)",
    )
    run_btn = gr.Button("Transcribe", variant="primary")
    out_it = gr.Textbox(label="Italian (ASR)", lines=4)
    out_en = gr.Textbox(label="English (translation)", lines=4)

    run_btn.click(fn=transcribe, inputs=[audio_in], outputs=[out_it, out_en])


if __name__ == "__main__":
    port = int(os.environ.get("PORT", "7860"))
    demo.launch(server_name="0.0.0.0", server_port=port)