import gradio as gr
from transformers import pipeline

models = {
    "Spanish": pipeline("translation", model="Helsinki-NLP/opus-mt-en-es"),
    "German": pipeline("translation", model="Helsinki-NLP/opus-mt-en-de"),
    "Japanese": pipeline("translation", model="staka/fugumt-en-ja"),
    "Ukrainian": pipeline("translation", model="Helsinki-NLP/opus-mt-en-uk"),
    "Russian": pipeline("translation", model="Helsinki-NLP/opus-mt-en-ru"),
}

def translate(text, language):
    if not text:
        return "Please enter an English sentence."
    return models[language](text)[0]["translation_text"]

with gr.Blocks(title="Short Translation") as demo:
    gr.Markdown("# Short Translation")
    with gr.Row():
        inp = gr.Textbox(label="English Sentence", lines=2, placeholder="Type your English sentence here...")
        lang = gr.Radio(["Spanish", "German", "Japanese", "Ukrainian", "Russian"], value="Spanish", label="Translation Language")
    out = gr.Textbox(label="Translation", interactive=False)

    btn = gr.Button("Translate")
    btn.click(translate, inputs=[inp, lang], outputs=out)

demo.launch()