remove useless models

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+import gradio as gr 
 from transformers import pipeline
 
 # Initialize translation pipelines
@@ -12,19 +12,9 @@ en_to_zh = pipeline("translation", model="Helsinki-NLP/opus-mt-en-zh")
 zh_to_en = pipeline("translation", model="Helsinki-NLP/opus-mt-zh-en")
 fr_to_es = pipeline("translation", model="Helsinki-NLP/opus-mt-fr-es")
 es_to_fr = pipeline("translation", model="Helsinki-NLP/opus-mt-es-fr")
-fr_to_zh = pipeline("translation", model="Helsinki-NLP/opus-mt-fr-zh")
-zh_to_fr = pipeline("translation", model="Helsinki-NLP/opus-mt-zh-fr")
-fr_to_ja = pipeline("translation", model="Helsinki-NLP/opus-mt-fr-jap")
-ja_to_fr = pipeline("translation", model="Helsinki-NLP/opus-mt-jap-fr")
-es_to_ja = pipeline("translation", model="Helsinki-NLP/opus-mt-es-jap")
-ja_to_es = pipeline("translation", model="Helsinki-NLP/opus-mt-jap-es")
-es_to_zh = pipeline("translation", model="Helsinki-NLP/opus-mt-es-zh")
-zh_to_es = pipeline("translation", model="Helsinki-NLP/opus-mt-zh-es")
-ja_to_zh = pipeline("translation", model="Helsinki-NLP/opus-mt-jap-zh")
-zh_to_ja = pipeline("translation", model="Helsinki-NLP/opus-mt-zh-jap")
 
 # Dictionary to map language pairs to translation pipelines
-translators = {
+direct_translators = {
     "English-French": en_to_fr,
     "French-English": fr_to_en,
     "English-Japanese": en_to_ja,
@@ -34,23 +24,53 @@ translators = {
     "English-Mandarin": en_to_zh,
     "Mandarin-English": zh_to_en,
     "French-Spanish": fr_to_es,
-    "Spanish-French": es_to_fr,
-    "French-Mandarin": fr_to_zh,
-    "Mandarin-French": zh_to_fr,
-    "French-Japanese": fr_to_ja,
-    "Japanese-French": ja_to_fr,
-    "Spanish-Japanese": es_to_ja,
-    "Japanese-Spanish": ja_to_es,
-    "Spanish-Mandarin": es_to_zh,
-    "Mandarin-Spanish": zh_to_es,
-    "Japanese-Mandarin": ja_to_zh,
-    "Mandarin-Japanese": zh_to_ja
+    "Spanish-French": es_to_fr
 }
 
+# Two-step translation pairs (using English as pivot)
+two_step_pairs = [
+    "French-Mandarin", "Mandarin-French",
+    "French-Japanese", "Japanese-French",
+    "Spanish-Japanese", "Japanese-Spanish",
+    "Spanish-Mandarin", "Mandarin-Spanish",
+    "Japanese-Mandarin", "Mandarin-Japanese"
+]
+
+def translate_two_step(text, source_lang, target_lang):
+    """Perform translation in two steps using English as an intermediate language."""
+    # First step: translate to English
+    source_to_en_key = f"{source_lang}-English"
+    source_to_en = direct_translators.get(source_to_en_key)
+    if not source_to_en:
+        return f"Error: Cannot translate from {source_lang} to English."
+    
+    # Translate source to English
+    english_text = source_to_en(text)[0]["translation_text"]
+    
+    # Second step: translate from English to target
+    en_to_target_key = f"English-{target_lang}"
+    en_to_target = direct_translators.get(en_to_target_key)
+    if not en_to_target:
+        return f"Error: Cannot translate from English to {target_lang}."
+    
+    # Translate English to target
+    return en_to_target(english_text)[0]["translation_text"]
+
 def get_translator(source_lang, target_lang):
-    """Get the appropriate translation pipeline based on source and target languages."""
+    """Get the appropriate translation method based on source and target languages."""
     key = f"{source_lang}-{target_lang}"
-    return translators.get(key)
+    
+    # Direct translation available
+    if key in direct_translators:
+        return lambda text: direct_translators[key](text)[0]["translation_text"]
+    
+    # Two-step translation needed
+    elif key in two_step_pairs:
+        return lambda text: translate_two_step(text, source_lang, target_lang)
+    
+    # No translation path available
+    else:
+        return None
 
 def translate(text, source_lang, target_lang):
     """Translate text from source language to target language."""
@@ -62,7 +82,7 @@ def translate(text, source_lang, target_lang):
     
     translator = get_translator(source_lang, target_lang)
     if translator:
-        return translator(text)[0]["translation_text"]
+        return translator(text)
     else:
         return f"Translation from {source_lang} to {target_lang} is not supported."
 
@@ -78,6 +98,7 @@ examples = {
 # Create Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Multilingual Translation")
+    gr.Markdown("### Some language pairs use two-step translation via English")
     
     with gr.Row():
         source_lang = gr.Dropdown(
@@ -99,6 +120,21 @@ with gr.Blocks() as demo:
         with gr.Column():
             target_text = gr.Textbox(label="Translation Result", lines=5)
     
+    translation_info = gr.Markdown("")
+    
+    # Function to update translation path information
+    def update_translation_info(source, target):
+        if source == target:
+            return "No translation needed (same language)"
+        
+        key = f"{source}-{target}"
+        if key in direct_translators:
+            return f"Using direct translation from {source} to {target}"
+        elif key in two_step_pairs:
+            return f"Using two-step translation: {source} → English → {target}"
+        else:
+            return f"Translation from {source} to {target} is not supported"
+    
     # Set up examples based on source language
     example_list = gr.Examples(
         examples=examples["English"],
@@ -111,6 +147,19 @@ with gr.Blocks() as demo:
     
     source_lang.change(update_examples, inputs=source_lang, outputs=example_list)
     
+    # Update translation path info when languages change
+    source_lang.change(
+        update_translation_info,
+        inputs=[source_lang, target_lang],
+        outputs=translation_info
+    )
+    
+    target_lang.change(
+        update_translation_info,
+        inputs=[source_lang, target_lang],
+        outputs=translation_info
+    )
+    
     # Set up translation function
     translate_btn.click(
         translate,