1.4.1 - remove langcodes dependency

bpm-ai/bpm_ai/translate/translate.py

@@ -7,7 +7,7 @@
 
 from bpm_ai.common.errors import MissingParameterError, LanguageNotFoundError
 from bpm_ai.common.multimodal import transcribe_audio, prepare_images_for_llm_prompt, ocr_documents
-from bpm_ai.translate.schema import get_translation_output_schema
+from bpm_ai.translate.util import get_translation_output_schema, get_lang_code
 
 
 @trace("bpm-ai-translate", ["llm"])
@@ -68,10 +68,7 @@ async def translate_nmt(
     input_items = await transcribe_audio(input_items, asr)
 
     try:
-        import langcodes
-        target_language_code = langcodes.find(target_language).language
-    except ImportError:
-        raise ImportError('langcodes is not installed')
+        target_language_code = get_lang_code(target_language)
     except LookupError:
         raise LanguageNotFoundError(f"Could not identify target language '{target_language}'.")
 

bpm-ai/bpm_ai/translate/util.py

@@ -0,0 +1,30 @@
+from enum import Enum
+
+
+def get_translation_output_schema(input_data: dict, target_language: str):
+    return {k: f'{k} translated into {target_language}' for k in input_data.keys()}
+
+
+class Language(Enum):
+    ENGLISH = "en"
+    FRENCH = "fr"
+    GERMAN = "de"
+    SPANISH = "es"
+    ITALIAN = "it"
+    PORTUGUESE = "pt"
+    DUTCH = "nl"
+    DANISH = "da"
+    SWEDISH = "sv"
+    NYNORSK = "no"
+    NORWEGIAN = "no"
+    BOKMAL = "no"
+    FINNISH = "fi"
+    POLISH = "pl"
+    UKRAINIAN = "uk"
+
+
+language_dict = {lang.lower(): Language[lang].value for lang in Language.__members__.keys()}
+
+
+def get_lang_code(language: str) -> str:
+    return language_dict[language.strip().lower()]