import gradio as gr
from transformers import pipeline
from transformers import pipeline

# Initialize the translation pipeline
pipe = pipeline("translation", model="alpha2002/eng_alpha_darija", tokenizer="alpha2002/eng_alpha_darija")

def translate_transformers(from_text):
    translation = pipe(from_text, src_lang="en_XX", tgt_lang="ar_AR")
    return translation[0]['translation_text']

def translate_gradio(input_text):
    return translate_transformers(input_text)

interface = gr.Interface(fn=translate_gradio,
                         title="English to Darija Translator",
                         description="Enter English text to translate it to Darija in Arabic script.",
                         inputs="text", 
                         outputs='text')

interface.launch(share=True)