import gradio as gr
from transformers import pipeline

# Fixed model path
MODEL_PATH = "BounharAbdelaziz/Terjman-Large-v2.0"

# Translation function
def translate_text(text):
    # Load the translation pipeline with the fixed model
    translator = pipeline("translation", model=MODEL_PATH)
    # Perform translation
    translated = translator(text, max_length=512)
    return translated[0]["translation_text"]

# Gradio app
def gradio_app():
    with gr.Blocks() as app:
        gr.Markdown("# 🇲🇦 Terjman-Large v2.0:")
        gr.Markdown("Enter the english text you want to translate to moroccan darija.")

        input_text = gr.Textbox(label="Input Text", placeholder="Enter text to translate...")
        output_text = gr.Textbox(label="Translated Text", interactive=False)

        translate_button = gr.Button("Translate")

        # Link input and output
        translate_button.click(
            fn=translate_text,
            inputs=input_text,
            outputs=output_text
        )

    return app

# Run the app
if __name__ == "__main__":
    app = gradio_app()
    app.launch()