import gradio as gr
import pandas as pd
from jiwer import wer
import re
import os

REGEX_YAML_BLOCK = re.compile(r"---[\n\r]+([\S\s]*?)[\n\r]+---[\n\r]")


def parse_readme(filepath):
    """Parses a repositories README and removes"""
    if not os.path.exists(filepath):
        return "No README.md found."
    with open(filepath, "r") as f:
        text = f.read()
        match = REGEX_YAML_BLOCK.search(text)
        if match:
            text = text[match.end() :]
    return text


def compute(input):
    preds = input['prediction'].tolist()
    truths = input['truth'].tolist()
    print(truths, preds, type(truths))
    err = wer(truths, preds)
    print(err)
    return err


description = "This is a calculator that you can use to find WER"

demo = gr.Interface(
        fn=compute,
        inputs=gr.components.Dataframe(
            headers=["prediction", "truth"],
            col_count=2,
            row_count=1,
            label="Input"
        ),
        outputs=gr.components.Textbox(label="WER"),
        description=description,
        title="WER Calculator",
        article=parse_readme("README.md")
    )

demo.launch()