# AUTOGENERATED! DO NOT EDIT! File to edit: app.ipynb.

# %% auto 0
__all__ = ['mf_transcribe', 'transcribe_malayalam_speech', 'gr_transcribe_malayalam_speech']

# %% app.ipynb 4
import gradio as gr
from faster_whisper import WhisperModel

# %% app.ipynb 8
def transcribe_malayalam_speech(audio_file, compute_type="int8", device="cpu", folder="vegam-whisper-medium-ml-fp16"):
    
    model = WhisperModel(folder, device=device, compute_type=compute_type)
    segments, info = model.transcribe(audio_file, beam_size=5)

    lst = []
    for segment in segments:
        # print("[%.2fs -> %.2fs] %s" % (segment.start, segment.end, segment.text))
        lst.append(segment.text)

    return(" ".join(lst))

# %% app.ipynb 9
def gr_transcribe_malayalam_speech(microphone, file_upload, compute_type="int8", device="cpu", folder="vegam-whisper-medium-ml-fp16"):
    warn_output = ""
    if (microphone is not None) and (file_upload is not None):
        warn_output = (
            "WARNING: You've uploaded an audio file and used the microphone. "
            "The recorded file from the microphone will be used and the uploaded audio will be discarded.\n"
        )

    elif (microphone is None) and (file_upload is None):
        return "ERROR: You have to either use the microphone or upload an audio file"

    audio_file = microphone if microphone is not None else file_upload
    
    model = WhisperModel(folder, device=device, compute_type=compute_type)
    segments, info = model.transcribe(audio_file, beam_size=5)

    lst = []
    for segment in segments:
        # print("[%.2fs -> %.2fs] %s" % (segment.start, segment.end, segment.text))
        lst.append(segment.text)

    return(" ".join(lst))

# %% app.ipynb 16
mf_transcribe = gr.Interface(
    fn=gr_transcribe_malayalam_speech,
    inputs=[
        gr.inputs.Audio(source="microphone", type="filepath", optional=True),
        gr.inputs.Audio(source="upload", type="filepath", optional=True),
    ],
    outputs="text",
    title="PALLAKKU (പല്ലക്ക്)",
    description=(
        "Pallakku is a Malayalam speech to text demo leveraging the model-weights of [vegam-whisper-medium-ml](https://huggingface.co/kurianbenoy/vegam-whisper-medium-ml-fp16)."
    ),
    article="Please note that this demo now uses CPU only and in my testing for a 5 seconds audio file it can take upto 15 seconds for results to come. If you are interested to use a GPU based API instead, feel free to contact the author @ kurian.bkk@gmail.com",
    allow_flagging="never",
)

# %% app.ipynb 17
mf_transcribe.launch(share=False)