import gradio as gr
from transformers import pipeline

# Load English-to-Japanese translation model
translator = pipeline("translation", model="staka/fugumt-en-ja")

def translate(text):
    if not text.strip():
        return "Please enter some English text."
    return translator(text)[0]["translation_text"]

# Build a better UI
with gr.Blocks(title="English to Japanese Translator") as demo:
    gr.Markdown("## 🌐 English to Japanese Translator")
    gr.Markdown("Enter English text below and get the Japanese translation.")
    
    with gr.Row():
        input_text = gr.Textbox(label="English Input", placeholder="Type something in English...", lines=3)
        output_text = gr.Textbox(label="Japanese Translation", lines=3)
    
    translate_btn = gr.Button("Translate")
    
    translate_btn.click(fn=translate, inputs=input_text, outputs=output_text)

demo.launch(share=True)