import base64
import os
import time

import banana_dev as banana
import gradio as gr
from loguru import logger

api_key = os.environ["BANANA_API_KEY"]
model_key = os.environ["BANANA_MODEL_KEY"]
password = os.environ["PASSWORD"]


def transcribe(email: str, file=None, url=None):
    if file:
        with open(file, "rb") as f:
            audio = f.read()
            audio_b64 = base64.b64encode(audio).decode("ISO-8859-1")
        payload = {"audio_b64": audio_b64}
    else:
        payload = {"url": url}
    payload["email"] = email
    response = banana.run(api_key, model_key, payload)
    print(response)

    if "error" in response:
        raise gr.Error(response["error"])

    # TODO: not sure why response dict contains multiple model outputs
    return response["modelOutputs"][0]


def run_demo(email, password, microphone, file_upload):
    if password != os.environ["PASSWORD"]:
        raise gr.Error("Das Kennwort ist falsch.")

    if (microphone is not None) and (file_upload is not None):
        logger.warning(
            "Achtung: Sie haben sowohl eine Datei hochgeladen als auch über das Mikrofon aufgenommen."
            " Wir verwenden nur die Datei, die Sie hochgeladen haben."
        )

    elif (microphone is None) and (file_upload is None):
        raise gr.Error(
            "Sie müssen entweder eine Datei hochladen oder über das Mikrofon aufnehmen."
        )

    file = microphone if microphone is not None else file_upload

    start = time.time()
    transcription = transcribe(email, file)
    logger.info(f"transcription took {time.time()-start:.3f}s")
    return "\n\n".join([seg["text"].strip() for seg in transcription["segments"]])


demo = gr.Interface(
    fn=run_demo,
    inputs=[
        gr.Textbox(label="Email", type="email"),
        gr.Textbox(label="Zugriffs-Kennwort (siehe oben)"),
        gr.Audio(source="microphone", type="filepath", label="Aufnehmen"),
        gr.Audio(source="upload", type="filepath", label="Datei hochladen"),
    ],
    outputs=gr.Textbox(label="Transkription"),
    title="Transkriptionsservice",
    description="<p>Die untenstehende, automatische Transkription wurde von uns für interne Zwecke entwickelt."
    " Sie liefert die bei weitem beste Transkriptionsqualität bei Aufnahmen nicht-fachspezifischer Gespräche.</p>"
    "<p><strong>Bitte geben Sie Ihre Email-Adresse an, damit wir Ihnen die fertige Transkription per Email zusenden können.</strong>"
    " Wir werden Ihre Email-Adresse nicht für andere Zwecke verwenden.</p>"
    "",
    article="",
    allow_flagging="never",
    css="footer {visibility: hidden}",
)

demo.launch(share=True)