update text translation stream and progress status

app.py

@@ -15,6 +15,7 @@ from typing import Optional
 from typing import Tuple
 from typing import Union
 
+
 # Third-party libraries
 import requests
 import streamlit as st
@@ -327,7 +328,7 @@ def process_message(
     message: str,
     operation_prompt: str = "",
     tts_enabled: bool = False
-) -> Tuple[Optional[bytes], Optional[float]]:
+) -> str:
     """
     Traite les messages des utilisateurs et génère une réponse.
 
@@ -336,55 +337,27 @@ def process_message(
         operation_prompt (str, optional): Prompt supplémentaire pour l'opération. Par défaut "".
         tts_enabled (bool, optional): Si la synthèse vocale est activée. Par défaut False.
 
-    Returns:
-        Tuple[Optional[bytes], Optional[float]]: Un tuple contenant l'audio TTS et sa durée,
-        ou (None, None) si TTS est désactivé ou en cas d'erreur.
     """
     payload_content = f'{operation_prompt} :\n"""\n{message}\n"""'
-
     st.session_state.messages.append({"role": "user", "content": payload_content})
-    with st.chat_message("user", avatar="👤"):
-        st.markdown(message)
-
-    with st.chat_message("assistant", avatar="👻"):
-        message_placeholder = st.empty()
-        full_response = ""
+    full_response = "" 
+    try:
+        for response in client.chat.completions.create(
+                model="gpt-4o-mini",
+                messages=st.session_state.messages,
+                stream=True,
+                temperature=0.1):
+            full_response += (response.choices[0].delta.content or "")
+            yield full_response + "▌"
         
-        with st.status("Traitement en cours...", expanded=True) as status:
-            status.update(label="Génération de la réponse", state="running", expanded=True)
-            try:
-                for response in client.chat.completions.create(
-                        model="gpt-4o-mini",
-                        messages=st.session_state.messages,
-                        stream=True,
-                        temperature=0.1):
-                    full_response += (response.choices[0].delta.content or "")
-                    message_placeholder.markdown(full_response + "▌")
-                # Utiliser regex pour supprimer les trois premiers et derniers guillemets doubles
-                full_response = re.sub(r'^"{3}|"{3}$', '', full_response.strip())
-                message_placeholder.markdown(full_response)
-                status.update(label="Réponse générée avec succès", state="complete")
-            except Exception as e:
-                st.error(f"Une erreur s'est produite lors de la génération de la réponse : {e}")
-                status.update(label="Erreur lors de la génération de la réponse", state="error")
-                return None, None
-    
-    st.session_state.messages.append(
-        {"role": "assistant", "content": full_response}
-    )
-
-    if tts_enabled:
-        with st.status("Conversion texte-parole en cours...", expanded=True) as status:
-            status.update(label="Conversion texte-parole", state="running", expanded=True)
-            try:
-                tts_audio, tts_duration = text_to_speech(full_response)
-                status.update(label="Conversion texte-parole réussie", state="complete")
-                return tts_audio, tts_duration
-            except Exception as e:
-                st.error(f"Une erreur s'est produite lors de la conversion texte-parole : {e}")
-                status.update(label="Erreur lors de la conversion texte-parole", state="error")
-                return None, None
-    return None, None
+        # Utiliser regex pour supprimer les trois premiers et derniers guillemets doubles
+        full_response = re.sub(r'^"{3}|"{3}$', '', full_response.strip())
+        st.session_state.messages.append({"role": "assistant", "content": full_response})
+        st.session_state.full_response = full_response
+        return full_response
+    except Exception as e:
+        st.error(f"Une erreur s'est produite lors de la génération de la réponse : {e}")
+        return ""    
 
 class GlobalSystemPrompts:
     """Class to store global system prompts."""
@@ -451,6 +424,46 @@ LANGUAGES_EMOJI = {
     "Vietnamese": "🇻🇳", "Welsh": "🏴󠁧󠁢󠁷󠁬󠁳󠁿"
 }
 
+
+def convert_iso6391_to_language_name(language_code: str) -> str:
+    """
+    Convertit un code ISO 639-1 en nom de langue.
+
+    Args:
+        language_code (str): Le code ISO 639-1 de la langue.
+
+    Returns:
+        str: Le nom de la langue correspondant au code ISO 639-1, ou 'English' si non trouvé.
+    """
+    # Dictionnaire associant les codes ISO 639-1 aux noms de langues
+    iso_to_language: Dict[str, str] = {
+        "af": "Afrikaans", "ar": "Arabic", "hy": "Armenian", "az": "Azerbaijani",
+        "be": "Belarusian", "bs": "Bosnian", "bg": "Bulgarian", "ca": "Catalan",
+        "zh": "Chinese", "hr": "Croatian", "cs": "Czech", "da": "Danish",
+        "nl": "Dutch", "en": "English", "et": "Estonian", "fi": "Finnish",
+        "fr": "French", "gl": "Galician", "de": "German", "el": "Greek",
+        "he": "Hebrew", "hi": "Hindi", "hu": "Hungarian", "is": "Icelandic",
+        "id": "Indonesian", "it": "Italian", "ja": "Japanese", "kn": "Kannada",
+        "kk": "Kazakh", "ko": "Korean", "lv": "Latvian", "lt": "Lithuanian",
+        "mk": "Macedonian", "ms": "Malay", "mr": "Marathi", "mi": "Maori",
+        "ne": "Nepali", "no": "Norwegian", "fa": "Persian", "pl": "Polish",
+        "pt": "Portuguese", "ro": "Romanian", "ru": "Russian", "sr": "Serbian",
+        "sk": "Slovak", "sl": "Slovenian", "es": "Spanish", "sw": "Swahili",
+        "sv": "Swedish", "tl": "Tagalog", "ta": "Tamil", "th": "Thai",
+        "tr": "Turkish", "uk": "Ukrainian", "ur": "Urdu", "vi": "Vietnamese",
+        "cy": "Welsh"
+    }
+
+    try:
+        # Retourner le nom de la langue correspondant au code ISO 639-1
+        return iso_to_language[language_code]
+    except KeyError:
+        # Gérer spécifiquement l'exception KeyError
+        print(f"Code de langue non trouvé : {language_code}")
+        return "English"
+
+
+
 def convert_language_name_to_iso6391(language_data: Union[str, Dict[str, str]]) -> str:
     """
     Convertit un nom de langue en son code ISO 639-1.
@@ -556,6 +569,10 @@ def main():
 
     # Interface utilisateur pour le chat textuel
     if user_input := st.chat_input(get_translation("entrez_message")):
+        
+        with st.chat_message("user", avatar="👤"):
+            st.markdown(user_input)
+
         # Traitement du message texte de l'utilisateur
         if st.session_state.language_detected is None:
             st.session_state.language_detected = detect_language(
@@ -563,106 +580,30 @@ def main():
             )
         
         audio_list = []
+
         for cursor_selected_lang in st.session_state.selected_languages:
             st.session_state.target_language = cursor_selected_lang["iso-639-1"]
-            
+            st.session_state.full_response = "" 
             # Initialisation du mode de traitement pour la langue cible actuelle
             system_prompt, operation_prompt = init_process_mode()
-
-            # Traitement du message utilisateur pour la langue cible actuelle
-            try:
-                tts_audio, tts_duration = process_message(
-                    user_input,
-                    operation_prompt=f"{operation_prompt}",
-                    tts_enabled=st.session_state.enable_tts_for_input_from_text_field
-                )
-                if tts_audio is not None:
-                    audio_list.append((tts_audio, tts_duration))
-            except Exception as e:
-                st.error(f"Erreur lors du traitement du message : {str(e)}")
-        
-        if audio_list:
-            try:
-                final_audio = concatenate_audio_files(audio_list)
-                with st.container(border=True):
-                    st.audio(final_audio, format="audio/mp3", autoplay=st.session_state.autoplay_tts)
-                    
-                    # Générer un nom de fichier unique
-                    timestamp = time.strftime("%Y%m%d-%H%M%S")
-                    langues = "_".join([lang["iso-639-1"] for lang in st.session_state.selected_languages])
-                    nom_fichier = f"reponse_audio_{langues}_{timestamp}.mp3"
-
-                    st.download_button(
-                        label=f"📥 {get_translation('telecharger_audio')}",
-                        data=final_audio,
-                        file_name=nom_fichier,
-                        mime="audio/mp3",
-                        use_container_width=True,
-                        type="primary",
-                        key=f"download_button_{langues}_{timestamp}",
+            target_language_name = cursor_selected_lang["language"]
+            with st.status(f'({target_language_name}) - {get_translation("traduction_en_cours")}', expanded=True) as response_status:
+                with st.chat_message("assistant", avatar="👻"):
+                    message_placeholder = st.empty()
+                    response_generator = process_message(
+                        user_input, operation_prompt, st.session_state.enable_tts_for_input_from_text_field
                     )
-            except Exception as e:
-                st.error(f"Erreur lors de la concaténation des fichiers audio : {str(e)}")
-
-    with st.container(border=True):
-        # Interface utilisateur pour l'enregistrement audio
-        st.write(f"🗣️ {get_translation('enregistrez_message')}")
-        if audio := audiorecorder(
-            start_prompt=get_translation("cliquez_enregistrer"),
-            stop_prompt=get_translation("cliquez_arreter"),
-            pause_prompt=get_translation("cliquez_pause"),
-            show_visualizer=True,
-            key="vocal_chat_input"
-        ):
-            # Traitement de l'entrée audio de l'utilisateur
-            if len(audio) > 0:
-                try:
-                    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_audio:
-                        audio.export(temp_audio.name, format="wav")
-                        transcription = transcribe_audio(temp_audio, language=st.session_state.language_detected)
-                    os.unlink(temp_audio.name)  # Suppression du fichier temporaire
-                    if st.session_state.language_detected is None:
-                        st.session_state.language_detected = detect_language(
-                            input_text=transcription, temperature=0.01
-                        )
-                        st.write(get_translation("langue_detectee").format(st.session_state.language_detected))
-
-                    #st.write(get_translation("transcription").format(transcription))
-
-                    audio_list = []
-                    for cursor_selected_lang in st.session_state.selected_languages:
-                        st.session_state.target_language = cursor_selected_lang["iso-639-1"]
-                        
-                        # Initialisation du mode de traitement pour la langue cible actuelle
-                        system_prompt, operation_prompt = init_process_mode()
-                        
-                        # Traitement du message utilisateur pour la langue cible actuelle
-                        try:
-                            tts_audio, tts_duration = process_message(
-                                transcription,
-                                operation_prompt=f"{operation_prompt}",
-                                tts_enabled=st.session_state.enable_tts_for_input_from_audio_record
-                            )
-                            if tts_audio is not None:
-                                audio_list.append((tts_audio, tts_duration))
-                        except Exception as e:
-                            st.error(f"Erreur lors du traitement du message audio : {str(e)}")
-                    if audio_list:
-                        try:
-                            final_audio = concatenate_audio_files(audio_list)
-                            with st.container(border=True):
-                                st.audio(final_audio, format="audio/mp3", autoplay=st.session_state.autoplay_tts)                    
-                                # Ajout d'un bouton de téléchargement pour l'audio final
-                                st.download_button(
-                                    label=f"📥 {get_translation('telecharger_audio')}",
-                                    data=final_audio,
-                                    file_name="audio_concatene.mp3",
-                                    mime="audio/mp3"
-                                )
-                        except Exception as e:
-                            st.error(f"Erreur lors de la concaténation des fichiers audio : {str(e)}")
-                except Exception as e:
-                    st.error(f"Erreur lors du traitement de l'audio : {str(e)}")
+                    response_status.update(label=f'({target_language_name}) - {get_translation("traduction_en_cours")}', state="running", expanded=True)
+                    for response_chunk in response_generator:
+                        message_placeholder.markdown(response_chunk)
+
+                    full_response = response_generator.close()  # Obtenir la réponse complète à la fin
+                    if st.session_state.full_response != "":
+                        message_placeholder.markdown(st.session_state.full_response)
+                        response_status.update(label=f'({target_language_name}) - {get_translation("traduction_terminee")}', state="complete", expanded=False)
+                    else:
+                        response_status.update(label=f'({target_language_name}) - {get_translation("erreur_traduction")}', state="error", expanded=False)
+
 
     # Configuration de la barre latérale
     with st.sidebar:

ui_lang_support.json

@@ -32,7 +32,13 @@
         "chargement": "Chargement en cours...",
         "aucun_message": "Aucun message pour le moment. Commencez la conversation !",
         "telecharger_audio": "Télécharger l'audio",
-        "lecture_auto_tts": "Lecture audio automatique de la synthese vocale"
+        "lecture_auto_tts": "Lecture audio automatique de la synthese vocale",
+	"traitement_en_cours": "Traitement en cours...",
+	"erreur_traitement": "Une erreur s'est produite lors du traitement de votre message : {}",
+	"traitement_termine": "Traitement terminé !",
+	"traduction_en_cours": "Traduction en cours...",
+	"erreur_traduction": "Une erreur s'est produite lors de la traduction de votre message : {}",
+	"traduction_terminee": "Traduction terminée !"
     },
     "Afrikaans": {
         "titre": "DEMORRHA APP - weergawe 1",
@@ -1994,4 +2000,4 @@
         "telecharger_audio": "Lawrlwytho'r sain",
         "lecture_auto_tts": "Darllen sain automatig o'r syntheseis lafar"
     }
-}
+}