Update app.py

app.py

@@ -67,24 +67,20 @@ def input_changed(source_language_dropdown, input_text=""):
 def translate(input_text, source, target):
     source_readable = source
 
-    # Detect source if needed
     if source == "Detect" or source.startswith("Detected"):
-        source, _ = auto_detect_language_code(input_text)
-
-    # Map from human-readable name to code if needed
-    if source in source_lang_dict:
-        source = source_lang_dict[source]
-    if target in target_lang_dict:
-        target = target_lang_dict[target]
+      source, _ = auto_detect_language_code(input_text)
+    if source in source_lang_dict.keys():
+      source = source_lang_dict[source]
+    target_lang_dict, _ = get_target_languages(source)
 
     try:
-          target = target_lang_dict[target]
-          model = f"Helsinki-NLP/opus-mt-{source}-{target}"
-          pipe = pipeline("translation", model=model)
-          translation = pipe(input_text)
-          return translation[0]['translation_text']
+      target = target_lang_dict[target]
+      model = f"Helsinki-NLP/opus-mt-{source}-{target}"
+      pipe = pipeline("translation", model=model)
+      translation = pipe(input_text)
+      return translation[0]['translation_text']
     except KeyError:
-          return f"Error: Translation from {source_readable} to {target} is not supported"
+      return f"Error: Translation from {source_readable} to {target} is not supported by Translation Models"
 
 with st.container():            
   st.header("Basque (EU) Euskera Multilingual Translator (Basque as the Starting Point)")