app.py

"""Create entry."""
# pylint: disbale=invalid-name
import gradio as gr
import pandas as pd

from about_time import about_time
from aset2pairs import aset2pairs
from cmat2aset import cmat2aset
from logzero import logger
from typing import List, Optional, Union
from hf_model_s_cpu import model_s

from radio_mlbee import __version__
from radio_mlbee.gen_cmat import gen_cmat
from radio_mlbee.utils import text1, text2


def greet(name):
    """Greet."""
    if not name:
        name = "world"
    return "Hello " + name + "!! (coming sooooon...)"


def ml_fn(
    text1: str,
    text2: str,
    # split_to_sents: bool = False
) -> pd.DataFrame:
    """Align text1 text2"""
    text1 = str(text1)
    text2 = str(text2)
    try:
        paras1 = text1.splitlines()
        paras1 = [_.strip() for _ in paras1 if _.strip()]
    except Exception as exc:
        logger.error(" praras.slpitlines() erros: %s, setting to ['']", exc)
        paras1 = [""]
    try:
        paras2 = text2.splitlines()
        paras2 = [_.strip() for _ in paras2 if _.strip()]
    except Exception as exc:
        logger.error(" praras slpitlines erros: %s, setting to ['']", exc)
        paras2 = [""]

    # if split_to_sents: ...  # TODO

    with about_time() as t:
        cmat = gen_cmat(paras1, paras2)
        aset = cmat2aset(cmat)

    _ = len(paras1) + len(paras2)
    av = t.duration / _ * 1000
    logger.info(" %s blocks, took %s, av. %s s/1000 blk", _, t.duration_human, av)

    pairs = aset2pairs(paras1, paras2, aset)
    df = pd.DataFrame(pairs, columns=["text1", "text2", "llh"])

    # return pd.DataFrame([["", "", ""]])
    # return df.to_html()
    return df


mlbee = gr.Interface(
    fn=ml_fn,
    inputs=[
        "textarea",
        "textarea",
        # gr.Checkbox(label="Split to sents?"),
    ],
    outputs="dataframe",
    # outputs="html",
    title=f"radio-mlbee {__version__}",
    description="mlbee rest api ",
    examples=[
        # ["a b", "cd", False],
        # [text1, text2, False],
        [text1, text2],
    ]
)

mlbee.launch(
    show_error=True,
    enable_queue=True,
)