import torch
from transformers import MarianMTModel, MarianTokenizer
import gradio as gr

def translate_hebrew_to_english(hebrew_text):
    # Load the tokenizer and model
    tokenizer = MarianTokenizer.from_pretrained("model_files", source_lang="he", target_lang="en")
    model = MarianMTModel.from_pretrained("model_files")

    # Tokenize the input text
    inputs = tokenizer.encode(hebrew_text, return_tensors="pt")

    # Translate the input text
    outputs = model.generate(inputs, max_length=512)

    # Decode the translated text
    translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
    return translated_text

# Define the translation function for Gradio
def translate(input_text):
    translated_text = translate_hebrew_to_english(input_text)
    return translated_text

# Create the Gradio interface
iface = gr.Interface(fn=translate, 
                     inputs=gr.inputs.Textbox(lines=10, placeholder="Paste your Hebrew text here"),
                     outputs=gr.outputs.Textbox(),
                     title="Hebrew to English Translator",
                     layout="vertical",
                     css=".input textarea { height: 300px; } .output textarea { height: 300px; }")
iface.launch()