import os
import subprocess

# Install required dependencies if missing
try:
    import torch
    import transformers
    import sentencepiece
except ImportError:
    subprocess.run(["pip", "install", "torch", "transformers", "sentencepiece"], check=True)
    import torch  # Re-import after installation
    import transformers
    import sentencepiece

from transformers import pipeline
import gradio as gr

# Define source and target language
src_lang = "ne"  # Nepali
tgt_lang = "en"  # English

# Load model with language codes
model = pipeline("translation", model="BeebekBhz/Final-ne-en", src_lang=src_lang, tgt_lang=tgt_lang)

def translate(text):
    return model(text)[0]['translation_text']

iface = gr.Interface(fn=translate, inputs="text", outputs="text")
iface.launch(share=True)