update demorrha

app.py

@@ -108,8 +108,13 @@ class GlobalSystemPrompts:
     def linguascribe():
         SYSTEM_PROMPT = f"{lire_fichier('linguascribe.prompt')}"
         return SYSTEM_PROMPT
-    
+
+SYSTEM_PROMPT=""
+OP_PROMPT=""
+
 def set_mode_translation(from_lang ,dest_lang):
+    global SYSTEM_PROMPT
+    global OP_PROMPT
     SYSTEM_PROMPT=GlobalSystemPrompts.linguascribe()
     OP_PROMPT = f"Translate({from_lang} to {dest_lang})"
     return {
@@ -119,6 +124,30 @@ def set_mode_translation(from_lang ,dest_lang):
 
 SUPPORTED_LANGUAGES=["Afrikaans", "Arabic", "Armenian", "Azerbaijani", "Belarusian", "Bosnian", "Bulgarian", "Catalan", "Chinese", "Croatian", "Czech", "Danish", "Dutch", "English", "Estonian", "Finnish", "French", "Galician", "German", "Greek", "Hebrew", "Hindi", "Hungarian", "Icelandic", "Indonesian", "Italian", "Japanese", "Kannada", "Kazakh", "Korean", "Latvian", "Lithuanian", "Macedonian", "Malay", "Marathi", "Maori", "Nepali", "Norwegian", "Persian", "Polish", "Portuguese", "Romanian", "Russian", "Serbian", "Slovak", "Slovenian", "Spanish", "Swahili", "Swedish", "Tagalog", "Tamil", "Thai", "Turkish", "Ukrainian", "Urdu", "Vietnamese", "Welsh"]
 
+def convert_language_name_to_iso6391(language_name):
+    # Dictionnaire de correspondance entre les noms de langues et les codes ISO 639-1
+    language_to_iso = {
+        "Afrikaans": "af", "Arabic": "ar", "Armenian": "hy", "Azerbaijani": "az",
+        "Belarusian": "be", "Bosnian": "bs", "Bulgarian": "bg", "Catalan": "ca",
+        "Chinese": "zh", "Croatian": "hr", "Czech": "cs", "Danish": "da",
+        "Dutch": "nl", "English": "en", "Estonian": "et", "Finnish": "fi",
+        "French": "fr", "Galician": "gl", "German": "de", "Greek": "el",
+        "Hebrew": "he", "Hindi": "hi", "Hungarian": "hu", "Icelandic": "is",
+        "Indonesian": "id", "Italian": "it", "Japanese": "ja", "Kannada": "kn",
+        "Kazakh": "kk", "Korean": "ko", "Latvian": "lv", "Lithuanian": "lt",
+        "Macedonian": "mk", "Malay": "ms", "Marathi": "mr", "Maori": "mi",
+        "Nepali": "ne", "Norwegian": "no", "Persian": "fa", "Polish": "pl",
+        "Portuguese": "pt", "Romanian": "ro", "Russian": "ru", "Serbian": "sr",
+        "Slovak": "sk", "Slovenian": "sl", "Spanish": "es", "Swahili": "sw",
+        "Swedish": "sv", "Tagalog": "tl", "Tamil": "ta", "Thai": "th",
+        "Turkish": "tr", "Ukrainian": "uk", "Urdu": "ur", "Vietnamese": "vi",
+        "Welsh": "cy"
+    }
+    
+    # Retourne le code ISO 639-1 correspondant au nom de la langue
+    return language_to_iso.get(language_name, "en")  # Par défaut, retourne 'en' si la langue n'est pas trouvée
+
+
 def main():
     st.title("------- DEMORRHA -------")
 
@@ -129,10 +158,11 @@ def main():
     if "process_mode" not in st.session_state:
         st.session_state["process_mode"] = "translation"
 
-
+    if "target_language" not in st.session_state:
+        st.session_state.target_language = "en"
+    
     if "translation" == st.session_state["process_mode"]:
-        pass
-        #set_mode_translation()
+        set_mode_translation(from_lang=st.session_state.language_detected, dest_lang=st.session_state.target_language)
         
     # Initialisation de l'historique des messages avec le prompt système
     if "messages" not in st.session_state:
@@ -176,14 +206,15 @@ def main():
 
         # Sélection de la langue de destination
         def on_language_change():
-            st.session_state.target_language = st.session_state.language_selector
+            st.session_state.target_language = convert_language_name_to_iso6391(st.session_state.language_selector)
 
         st.selectbox(
             label="Choisissez la langue de destination",
             options=tuple(SUPPORTED_LANGUAGES),
             placeholder="Sélectionnez une langue",
             key="language_selector",
-            on_change=on_language_change
+            on_change=on_language_change,
+            index=SUPPORTED_LANGUAGES.index("English")
         )
 
 if __name__ == "__main__":