Update app.py

app.py

@@ -3,10 +3,10 @@ LocaleNLP Translation Service
 ============================
 
 A multi-language translation application supporting English, Wolof, Hausa, and Darija.
-Features text, audio, and document translation with a modern web interface.
+Features text, audio, and document translation with automatic chaining for all language pairs.
 
 Author: LocaleNLP Team
-Version: 1.0.0
+Version: 1.0
 """
 
 import os
@@ -280,7 +280,7 @@ class TranslationService:
         target_lang: Language
     ) -> str:
         """
-        Translate text from source to target language.
+        Translate text from source to target language with automatic chaining.
         
         Args:
             text: Input text to translate
@@ -292,11 +292,63 @@ class TranslationService:
         """
         if not text.strip():
             return "No input text to translate."
-            
+        
+        # Direct translation if model exists
+        if (source_lang, target_lang) in TRANSLATION_MODELS:
+            return self._direct_translate(text, source_lang, target_lang)
+        
+        # Automatic chaining through English
+        return self._chained_translate(text, source_lang, target_lang)
+    
+    def _direct_translate(
+        self, 
+        text: str, 
+        source_lang: Language, 
+        target_lang: Language
+    ) -> str:
+        """Perform direct translation using available model."""
         pipeline_obj, lang_tag = self.model_manager.get_translation_pipeline(
             source_lang, target_lang
         )
         
+        return self._process_text_with_pipeline(text, pipeline_obj, lang_tag)
+    
+    def _chained_translate(
+        self, 
+        text: str, 
+        source_lang: Language, 
+        target_lang: Language
+    ) -> str:
+        """
+        Perform chained translation through English as intermediate language.
+        
+        Args:
+            text: Input text to translate
+            source_lang: Source language
+            target_lang: Target language
+            
+        Returns:
+            Translated text through chaining
+        """
+        # First: source_lang -> English
+        intermediate_text = self._direct_translate(
+            text, source_lang, Language.ENGLISH
+        )
+        
+        # Second: English -> target_lang
+        final_text = self._direct_translate(
+            intermediate_text, Language.ENGLISH, target_lang
+        )
+        
+        return final_text
+    
+    def _process_text_with_pipeline(
+        self, 
+        text: str, 
+        pipeline_obj: Any, 
+        lang_tag: str
+    ) -> str:
+        """Process text using translation pipeline."""
         # Process text in paragraphs
         paragraphs = text.splitlines()
         translated_paragraphs = []
@@ -420,7 +472,7 @@ class TranslationApp:
         
         with gr.Blocks(
             title="LocaleNLP Translation Service",
-            theme=gr.themes.Soft()
+            theme=gr.themes.Base(primary_hue=gr.themes.colors.black)
         ) as interface:
             # Header
             gr.Markdown("""
@@ -437,7 +489,7 @@ class TranslationApp:
                 )
                 
                 input_lang = gr.Dropdown(
-                    choices=[lang.value for lang in Language if lang != Language.DARIJA],
+                    choices=[lang.value for lang in Language],
                     label="Input Language",
                     value=Language.ENGLISH.value
                 )
@@ -477,12 +529,12 @@ class TranslationApp:
             
             translate_btn = gr.Button(
                 "🔄 Process & Translate",
-                variant="primary"
+                variant="secondary"
             )
             
             output_text = gr.Textbox(
                 label="Translated Text",
-                lines=8,
+                lines=10,
                 interactive=False
             )