from transformers import MarianMTModel, MarianTokenizer
import gradio as gr

# Load model
model_name = "Helsinki-NLP/opus-mt-en-ur"
tokenizer = MarianTokenizer.from_pretrained(model_name)
model = MarianMTModel.from_pretrained(model_name)

# Translation function
def translate_text(text):
    tokens = tokenizer(text, return_tensors="pt", padding=True)
    translated = model.generate(**tokens)
    result = tokenizer.decode(translated[0], skip_special_tokens=True)
    return result

# Gradio UI
interface = gr.Interface(
    fn=translate_text,
    inputs=gr.Textbox(lines=4, placeholder="Enter English text here..."),
    outputs="text",
    title="English to Urdu Translator",
    description="Translate English text into Urdu instantly."
)

interface.launch()