Update language_directions.py

language_directions.py

@@ -1,72 +1,72 @@
-from helsinki_models import helsinki_models, get_clearly_formatted_langauge_directions
-from iso639_wrapper import get_name_from_iso_code
-from language_detection import detect_language
-from collections import OrderedDict
-from utils import convert_keys_to_lowercase, match_in_keys, match_in_values
-
-
-def get_all_source_languages():
-    """
-    Returns a human-readable `dict source_languages_names:codes` 
-    based on the available models.
-    """
-    source_languages = {}
-    language_directions = get_clearly_formatted_langauge_directions()
-
-    for direction in language_directions:
-        source_lang_code = direction.split('-')[0]
-        source_language_name = get_name_from_iso_code(source_lang_code)
-        if source_language_name:
-            source_languages[source_language_name] = source_lang_code
-    source_languages = OrderedDict(sorted(source_languages.items()))
-    all_source_langs_including_auto_detect = \
-        { **{'Detect' : 'Detect'}, **source_languages}
-    return all_source_langs_including_auto_detect
-
-def update_source_languages_dict(source_languages_dict, auto_detected_language):
-    source_languages_dict[auto_detected_language] = "Detect"
-
-def get_target_languages(source_language_code, input_text=None):
-    """
-    Returns a human-readable `dict of target languages names to codes` 
-    based on the available models and the source language passed.
-    """
-    include_all_languages = False
-    if source_language_code == "Detect":
-        source_language_code, include_all_languages = auto_detect_language_code(input_text)
-        
-    target_languages = {}
-    language_directions = get_clearly_formatted_langauge_directions()
-    for direction in language_directions:
-        if direction.startswith(f"{source_language_code}-") or include_all_languages:
-            target_language = direction.split('-')[1]  # Extracting the last part as the target language
-            target_language_name = get_name_from_iso_code(target_language)
-            if target_language_name:
-                target_languages[target_language_name] = target_language
-    return OrderedDict(sorted(target_languages.items())), source_language_code
-
-def auto_detect_language_code(input_text):
-    DEFAULT_SOURCE_LANGUAGE = "eu"
-    detected_language_string = DEFAULT_SOURCE_LANGUAGE
-    if not input_text:
-        return DEFAULT_SOURCE_LANGUAGE, True
-    language_or_code = detect_language(input_text)
-    if language_or_code == "unknown":
-        return DEFAULT_SOURCE_LANGUAGE, True
-    else:
-        detected_language_string = match_in_keys(get_all_source_languages(), language_or_code)
-        if not detected_language_string:
-            detected_language_string = match_in_values(get_all_source_languages(), language_or_code)
-        if detected_language_string:
-            return detected_language_string, False
-        else:
-            return DEFAULT_SOURCE_LANGUAGE, True
-    
-
-# Example usage:
-# all_source_languages = get_all_source_languages()
-# print("All Source Languages:", all_source_languages)
-
-# source_language_code = "pa"  # Replace with your desired source language
-# target_languages = get_target_languages(source_language_code)
-# print(f"Target Languages for {source_language_code}:", target_languages)
+from helsinki_models import helsinki_models, get_clearly_formatted_language_directions
+from iso639_wrapper import get_name_from_iso_code
+from language_detection import detect_language
+from collections import OrderedDict
+from utils import convert_keys_to_lowercase, match_in_keys, match_in_values
+
+
+def get_all_source_languages():
+    """
+    Returns a human-readable `dict source_languages_names:codes` 
+    based on the available models.
+    """
+    source_languages = {}
+    language_directions = get_clearly_formatted_language_directions()
+
+    for direction in language_directions:
+        source_lang_code = direction.split('-')[0]
+        source_language_name = get_name_from_iso_code(source_lang_code)
+        if source_language_name:
+            source_languages[source_language_name] = source_lang_code
+    source_languages = OrderedDict(sorted(source_languages.items()))
+    all_source_langs_including_auto_detect = \
+        { **{'Detect' : 'Detect'}, **source_languages}
+    return all_source_langs_including_auto_detect
+
+def update_source_languages_dict(source_languages_dict, auto_detected_language):
+    source_languages_dict[auto_detected_language] = "Detect"
+
+def get_target_languages(source_language_code, input_text=None):
+    """
+    Returns a human-readable `dict of target languages names to codes` 
+    based on the available models and the source language passed.
+    """
+    include_all_languages = False
+    if source_language_code == "Detect":
+        source_language_code, include_all_languages = auto_detect_language_code(input_text)
+        
+    target_languages = {}
+    language_directions = get_clearly_formatted_langauge_directions()
+    for direction in language_directions:
+        if direction.startswith(f"{source_language_code}-") or include_all_languages:
+            target_language = direction.split('-')[1]  # Extracting the last part as the target language
+            target_language_name = get_name_from_iso_code(target_language)
+            if target_language_name:
+                target_languages[target_language_name] = target_language
+    return OrderedDict(sorted(target_languages.items())), source_language_code
+
+def auto_detect_language_code(input_text):
+    DEFAULT_SOURCE_LANGUAGE = "eu"
+    detected_language_string = DEFAULT_SOURCE_LANGUAGE
+    if not input_text:
+        return DEFAULT_SOURCE_LANGUAGE, True
+    language_or_code = detect_language(input_text)
+    if language_or_code == "unknown":
+        return DEFAULT_SOURCE_LANGUAGE, True
+    else:
+        detected_language_string = match_in_keys(get_all_source_languages(), language_or_code)
+        if not detected_language_string:
+            detected_language_string = match_in_values(get_all_source_languages(), language_or_code)
+        if detected_language_string:
+            return detected_language_string, False
+        else:
+            return DEFAULT_SOURCE_LANGUAGE, True
+    
+
+# Example usage:
+# all_source_languages = get_all_source_languages()
+# print("All Source Languages:", all_source_languages)
+
+# source_language_code = "pa"  # Replace with your desired source language
+# target_languages = get_target_languages(source_language_code)
+# print(f"Target Languages for {source_language_code}:", target_languages)