import gradio as gr                   # UI library
from transformers import pipeline     # Transformers pipeline

model_checkpoint_en2fil = "SalamaThanks/SalamaThanksTransformer_en2fil_v1"
model_checkpoint_fil2en = "SalamaThanks/SalamaThanksTransformer_fil2en_v1"

translator_en2fil = pipeline("translation", model = model_checkpoint_en2fil)
translator_fil2en = pipeline("translation", model = model_checkpoint_fil2en)

def transformer_en2fil(from_text):
  results = translator_en2fil(from_text)
  return results[0]['translation_text']

def transformer_fil2en(from_text):
  results = translator_fil2en(from_text)
  return results[0]['translation_text']

def check_lang(lang, from_text):
  if lang == "English-to-Filipino":
    return transformer_en2fil(from_text)
  elif lang == "Filipino-to-English":
    return transformer_fil2en(from_text)

interface = gr.Interface(
  fn=check_lang, 
  inputs=[
    gr.inputs.Radio(["English-to-Filipino", "Filipino-to-English"]),
    gr.inputs.Textbox(lines=4, placeholder='Input Text to Translate:')],
  outputs='text'
)

interface.launch(debug=True)