Final API implementation for 6 UN languages

app.py

@@ -1,49 +1,19 @@
 import gradio as gr
-from transformers import pipeline
+from deep_translator import GoogleTranslator
 
-# Dictionary mapping language names to their ISO 639-1 codes and corresponding Helsinki-NLP model tags.
-# Note: English (en) is the pivot language for the Helsinki-NLP Opus models.
+# Dictionary mapping language names to their ISO 639-1 codes 
+# GoogleTranslator uses standard codes, including simplified Chinese (zh-CN)
 LANGUAGE_MAP = {
+    "Arabic": "ar",
+    "Chinese (Simplified)": "zh-CN", 
     "English": "en",
     "French": "fr",
-    "Spanish": "es",
     "Russian": "ru",
-    "Chinese (Simplified)": "zh",
-    "Arabic": "ar"
+    "Spanish": "es" 
 }
 LANGUAGE_OPTIONS = list(LANGUAGE_MAP.keys())
-MODEL_CACHE = {}
-
-def get_translator_pipeline(src_lang_code, tgt_lang_code):
-    """
-    Loads or retrieves the correct Helsinki-NLP model based on the source and target codes.
-    It handles two main cases:
-    1. src_lang is English (e.g., en-fr, en-ru)
-    2. tgt_lang is English (e.g., fr-en, ru-en)
-    """
-    # Case 1: English as source language
-    if src_lang_code == 'en':
-        model_name = f"Helsinki-NLP/opus-mt-en-{tgt_lang_code}"
-    # Case 2: English as target language
-    elif tgt_lang_code == 'en':
-        model_name = f"Helsinki-NLP/opus-mt-{src_lang_code}-en"
-    # Case 3: Direct translation between two non-English languages
-    else:
-        return None, "Error: Direct translation between two non-English languages is not supported in this simplified app. Please use English as either the Source or Target language."
-
-    if model_name not in MODEL_CACHE:
-        print(f"Loading new model: {model_name}")
-        try:
-            MODEL_CACHE[model_name] = pipeline("translation", model=model_name)
-        except Exception as e:
-            return None, f"Error loading model {model_name}: {e}"
-   
-    return MODEL_CACHE[model_name], None
 
 def translate_text(input_text, src_lang_name, tgt_lang_name):
-    """
-    Takes text and language names, determines the model, and performs translation.
-    """
     if not input_text:
         return "", "Please enter text to translate."
 
@@ -56,18 +26,13 @@ def translate_text(input_text, src_lang_name, tgt_lang_name):
     if not src_code or not tgt_code:
         return "", "Invalid language selection."
 
-    translator, error = get_translator_pipeline(src_code, tgt_code)
-
-    if error:
-        return "", error
-    if not translator:
-        return "", "Translation service unavailable."
-
     try:
-        result = translator(input_text, max_length=512)
-        translated_text = result[0]['translation_text']
+        translator = GoogleTranslator(source=src_code, target=tgt_code)
+        
+        translated_text = translator.translate(input_text)
+        
         return translated_text, "Translation successful."
-       
+        
     except Exception as e:
         return "", f"Translation failed during execution: {e}"
 
@@ -76,10 +41,9 @@ with gr.Blocks(title="UN Official Language Translator", theme=gr.themes.Soft())
     gr.Markdown(
     """
     # 🌐 UN Official Language Translator
-    This app supports translation between the six official languages of the UN:  
-    ** Arabic, Chinese, English, French, Russian, and Spanish**.
-   
-    Please select your source and target languages below.
+    
+    This app supports translation between the six official languages of the UN: 
+    **Arabic, Chinese, English, French, Russian, and Spanish**.
     """
     )
 
@@ -87,32 +51,34 @@ with gr.Blocks(title="UN Official Language Translator", theme=gr.themes.Soft())
         source_lang = gr.Dropdown(
             label="Source Language",
             choices=LANGUAGE_OPTIONS,
-            value="English"
+            value="English",
+            scale=1
         )
         target_lang = gr.Dropdown(
             label="Target Language",
             choices=LANGUAGE_OPTIONS,
-            value="French"
+            value="French",
+            scale=1
         )
 
     input_text = gr.Textbox(
-        label="Input Text",
-        placeholder="Enter the text you want to translate...",
+        label="Input Text", 
+        placeholder="Enter the text you want to translate...", 
         lines=4
     )
-   
+    
     translate_button = gr.Button("Translate", variant="primary")
-   
+    
     output_text = gr.Textbox(
-        label="Translated Output",
-        interactive=False,
+        label="Translated Output", 
+        interactive=False, 
         lines=4
     )
-   
+    
     status_message = gr.Textbox(label="Status", interactive=False, value="Ready to translate.")
-   
+    
     translate_button.click(
-        fn=translate_text,
-        inputs=[input_text, source_lang, target_lang],
+        fn=translate_text, 
+        inputs=[input_text, source_lang, target_lang], 
         outputs=[output_text, status_message]
     )

requirements.txt

@@ -1,5 +1,2 @@
 gradio
-transformers
-torch
-sentencepiece
-accelerate
+deep-translator