"""Hum research Space: record, choose a rendering path, hear the result."""

from __future__ import annotations

import gradio as gr

from logic import INSTRUMENTS
from pipeline import generate_all


SOLO_STYLE = (
    "instrumental solo {instrument}, clear single melody, intimate dry recording, "
    "natural instrumental tone"
)
ARRANGEMENT_STYLE = (
    "instrumental chamber pop, {instrument} melody, acoustic bass, "
    "brushed drums, soft strings, warm intimate mood"
)
DRUM_STYLE = (
    "instrumental {instrument} lead melody, steady acoustic drum kit, "
    "kick and snare, closed hi-hat, dry studio recording"
)


def suggest_prompts(instrument: str) -> tuple[str, str, str]:
    name = INSTRUMENTS[instrument][1]
    return tuple(template.format(instrument=name)
                 for template in (SOLO_STYLE, DRUM_STYLE, ARRANGEMENT_STYLE))


def run(audio: str | None, instrument: str, solo_style: str, drum_style: str,
        arrangement_style: str, seed: int, include_yue: bool, save_recording: bool,
        progress: gr.Progress = gr.Progress(track_tqdm=False)):
    try:
        styles = {"yue_solo": solo_style, "yue_drums": drum_style,
                  "yue_arrangement": arrangement_style}
        for result in generate_all(audio, instrument, styles, int(seed), include_yue,
                                   progress, save_recording=save_recording):
            yield result
            if result[-1]["complete"] and "recording_archive" in result[-1]["errors"]:
                gr.Warning("Запись не сохранилась в приватном архиве; см. детали запуска.")
    except Exception as error:
        raise gr.Error(str(error)) from error


CSS = """
.gradio-container {max-width: 1000px !important; margin: auto !important;}
#hero {padding: 20px 4px 10px 4px;}
#hero h1 {font-size: clamp(2.5rem, 6vw, 4.5rem); letter-spacing: -.045em; margin: 0;}
#hero p {font-size: 1.08rem; max-width: 720px; line-height: 1.55; color: #566179;}
.note {border: 1px solid #d9e0ed; border-radius: 16px; padding: 14px 18px;
       background: linear-gradient(120deg,#f5f7ff,#f1fbfc); color: #3e4b64;}
"""


with gr.Blocks(title="Hum · напев в инструмент", css=CSS, theme=gr.themes.Soft()) as demo:
    gr.HTML("""<div id='hero'><h1>Hum <span style='color:#6474db'>♪</span></h1>
    <p>Напойте фразу в микрофон или загрузите короткую запись. Один запуск
    создаст все варианты: сэмплерные инструменты с исходным временем и три
    эксперимента YuE2. Первые результаты появятся до окончания YuE2.</p></div>""")
    with gr.Row():
        with gr.Column(scale=4):
            audio = gr.Audio(label="Ваш напев", sources=["microphone", "upload"],
                             type="filepath", format=None)
            save_recording = gr.Checkbox(
                label="Сохранить исходную запись в приватный архив владельца Space",
                value=False)
            gr.Markdown("Если включить сохранение, исходный файл с микрофона или загрузки "
                        "останется в приватном архиве Hugging Face после перезапуска Space. "
                        "Без галочки исходник остаётся лишь во временном хранилище Space "
                        "до очистки или перезапуска.")
            instrument = gr.Dropdown(list(INSTRUMENTS), value="Флейта", label="Инструмент")
            include_yue = gr.Checkbox(label="Запустить YuE2 · три генерации (включено по умолчанию)",
                                      value=True)
            with gr.Accordion("Настройки YuE2", open=False):
                defaults = suggest_prompts("Флейта")
                solo_style = gr.Textbox(label="Соло · style", lines=2, value=defaults[0])
                drum_style = gr.Textbox(label="Мелодия + барабаны · style", lines=2,
                                        value=defaults[1])
                arrangement_style = gr.Textbox(label="Композиция · style", lines=2,
                                               value=defaults[2])
                seed = gr.Number(label="Seed", value=4096, precision=0)
            go = gr.Button("Создать все варианты", variant="primary", size="lg")
            gr.HTML("""<div class='note'><b>Текущая цель:</b> один инструмент и
            приблизительно исходное время. Электрогитара и бас используют
            записанные сэмплы; регистр баса выбирается автоматически. Обработанные версии
            добавляют усиление и лёгкий эффект, но не новые ноты. Барабаны
            привязаны к долям SheetSage2. YuE2 может изменить
            длительность, добавить человеческие звуки или другие инструменты;
            отсутствие вокала не гарантируется. Это публичный Space: записи
            обрабатываются на сервере Hugging Face.</div>""")
        with gr.Column(scale=6):
            with gr.Row():
                with gr.Column():
                    instrument_only = gr.Audio(label="Инструмент · исходный ритм",
                                               interactive=False)
                    instrument_drums = gr.Audio(label="Инструмент + синтезаторные барабаны",
                                                interactive=False)
                    contour = gr.Audio(label="Непрерывная высота · synth",
                                       interactive=False)
                    guitar_clean = gr.Audio(label="Электрогитара · чистая",
                                            interactive=False)
                    guitar_amp = gr.Audio(label="Электрогитара · усилитель и эффекты",
                                          interactive=False)
                    bass_clean = gr.Audio(label="Бас · чистый",
                                          interactive=False)
                    bass_amp = gr.Audio(label="Бас · тёплый усилитель",
                                        interactive=False)
                with gr.Column():
                    yue_solo = gr.Audio(label="YuE2 · соло", interactive=False)
                    yue_drums = gr.Audio(label="YuE2 · мелодия + барабаны",
                                         interactive=False)
                    yue_arrangement = gr.Audio(label="YuE2 · композиция",
                                               interactive=False)
            with gr.Row():
                midi = gr.File(label="Мелодия · MIDI")
                drum_midi = gr.File(label="Барабаны · MIDI")
                beat_lab = gr.File(label="Доли · LAB")
                abc = gr.File(label="Мелодия для YuE2 · ABC")
            details = gr.JSON(label="Детали запуска")
    instrument.change(suggest_prompts, inputs=instrument,
                      outputs=[solo_style, drum_style, arrangement_style])
    go.click(run, inputs=[audio, instrument, solo_style, drum_style,
                          arrangement_style, seed, include_yue, save_recording],
             outputs=[instrument_only, instrument_drums, contour,
                      guitar_clean, guitar_amp, bass_clean, bass_amp,
                      yue_solo, yue_drums, yue_arrangement,
                      midi, drum_midi, beat_lab, abc, details], show_progress="full")


if __name__ == "__main__":
    demo.queue(max_size=8, default_concurrency_limit=1)
    demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True)