| | from functools import lru_cache |
| | from typing import Literal, Any |
| |
|
| | import gradio as gr |
| | from transformers import pipeline |
| | from indic_transliteration import sanscript |
| | from indic_transliteration.sanscript import transliterate |
| | import os |
| | os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0" |
| |
|
| | |
| | HI_EN_MODEL = "Helsinki-NLP/opus-mt-hi-en" |
| | EN_HI_MODEL = "Helsinki-NLP/opus-mt-en-hi" |
| |
|
| | TaskType = Literal[ |
| | "Hindi → English", |
| | "English → Hindi", |
| | "Hindi → Hinglish (transliteration)", |
| | ] |
| |
|
| |
|
| | @lru_cache(maxsize=None) |
| | def get_hi_en_pipe() -> pipeline: |
| | """Lazy-load Hindi → English translation pipeline""" |
| | return pipeline("translation", model=HI_EN_MODEL) |
| |
|
| |
|
| | @lru_cache(maxsize=None) |
| | def get_en_hi_pipe() -> pipeline: |
| | """Lazy-load English → Hindi translation pipeline""" |
| | return pipeline("translation", model=EN_HI_MODEL) |
| |
|
| |
|
| | def translate( |
| | text: str | None, |
| | task: TaskType, |
| | max_length: int = 256, |
| | ) -> str: |
| | """ |
| | Main translation function used by the Gradio interface. |
| | Handles empty input gracefully and supports three tasks. |
| | """ |
| | text = (text or "").strip() |
| | if not text: |
| | return "" |
| |
|
| | try: |
| | if task == "Hindi → English": |
| | pipe = get_hi_en_pipe() |
| | result = pipe( |
| | text, |
| | max_length=max_length, |
| | num_beams=4, |
| | early_stopping=True, |
| | ) |
| | return result[0]["translation_text"] |
| |
|
| | if task == "English → Hindi": |
| | pipe = get_en_hi_pipe() |
| | result = pipe( |
| | text, |
| | max_length=max_length, |
| | num_beams=4, |
| | early_stopping=True, |
| | ) |
| | return result[0]["translation_text"] |
| |
|
| | if task == "Hindi → Hinglish (transliteration)": |
| | return transliterate(text, sanscript.DEVANAGARI, sanscript.ITRANS) |
| |
|
| | return "Unsupported task selected" |
| |
|
| | except (ValueError, RuntimeError, OSError) as e: |
| | return f"Translation error: {str(e)}" |
| | except Exception as e: |
| | return f"Unexpected error during translation: {str(e)}" |
| |
|
| |
|
| | def build_demo() -> gr.Blocks: |
| | """Build and configure the Gradio interface""" |
| | with gr.Blocks(title="Hindi ↔ English + Hinglish Translator") as demo: |
| | gr.Markdown( |
| | """ |
| | # Hindi ↔ English + Hinglish Translator |
| | |
| | - **Hindi → English** & **English → Hindi**: Neural machine translation |
| | - **Hindi → Hinglish**: Roman transliteration (Devanagari → ITRANS) |
| | """ |
| | ) |
| |
|
| | with gr.Row(): |
| | task = gr.Dropdown( |
| | choices=[ |
| | "Hindi → English", |
| | "English → Hindi", |
| | "Hindi → Hinglish (transliteration)", |
| | ], |
| | value="Hindi → English", |
| | label="Task", |
| | interactive=True, |
| | ) |
| | max_len = gr.Slider( |
| | 32, 512, value=256, step=16, |
| | label="Max output length", |
| | info="Higher values allow longer translations (slower)", |
| | ) |
| |
|
| | input_text = gr.Textbox( |
| | label="Input text", |
| | lines=5, |
| | placeholder="नमस्ते दुनिया! या Hello world...", |
| | ) |
| |
|
| | translate_btn = gr.Button("Translate", variant="primary") |
| |
|
| | output_text = gr.Textbox( |
| | label="Output", |
| | lines=5, |
| | interactive=False, |
| | ) |
| |
|
| | translate_btn.click( |
| | fn=translate, |
| | inputs=[input_text, task, max_len], |
| | outputs=output_text, |
| | ) |
| |
|
| | gr.Examples( |
| | examples=[ |
| | ["नमस्ते! सब ठीक है?", "Hindi → English", 128], |
| | ["How are you today?", "English → Hindi", 128], |
| | ["नमस्ते भाई, क्या हाल है?", "Hindi → Hinglish (transliteration)", 128], |
| | ["खुश रहो और मुस्कुराते रहो", "Hindi → Hinglish (transliteration)", 128], |
| | ], |
| | inputs=[input_text, task, max_len], |
| | label="Quick examples", |
| | ) |
| |
|
| | return demo |
| |
|
| |
|
| | demo = build_demo() |
| |
|
| |
|
| | if __name__ == "__main__": |
| | |
| | demo.launch( |
| | |
| | |
| | |
| | |
| | ) |
