import os
import gradio as gr
from gradio_client import Client

MY_HF_TOKEN_KEY = os.environ["MY_HF_TOKEN_KEY"]

client = Client("Ghana-NLP/test-text-models",hf_token=MY_HF_TOKEN_KEY)

def generate(text, direction):
    output = client.predict(text, direction)
    return  output

title = "GhanaNLP: Test Text Translation Models"

description = """
<b>How to use:</b> Translate Twi, Ga or Ewe text to/from English.
"""

examples = [
    ["Yaaba jogbaŋŋ mi Misuɔmɔ.","Ga->En"],
    ["ha mi gbe fioo","Ga->En"],
    ["kɛ osuɔmɔ aha mi","Ga->En"],
]

gr.Interface(
    fn=generate,
    inputs=[
        gr.Text(label="Input Text"),
        gr.Dropdown(
            ["Twi->En", "En->Twi","En->Ga", "Ga->En","En->Ewe", "Ewe->En",
            "Dagbani->En", "En->Dagbani","En->Gurene", "Gurene->En","En->Kusaal", "Kusaal->En"],
            label="Direction", info="Select From (Input) -> To (Output) languages"
        ), 
        ],
    outputs=[
        gr.Text(label="Output Text"),
    ],
    title=title,
    description=description,
    examples=examples,
).launch()