up

pages/configuration_ui_lang.py

@@ -74,8 +74,8 @@ translations = load_ui_language()
 
 def language_selection_page():
     st.set_page_config(initial_sidebar_state="collapsed")
-    st.set_page_config(page_icon="👹")
-    st.set_page_config(layout="wide")
+    #st.set_page_config(page_icon="👹")
+    #st.set_page_config(layout="wide")
     # Fonction de rappel pour mettre à jour la langue temporaire lorsque la sélection change
     def update_temp_language():
         st.session_state['temp_interface_language'] = st.session_state['language_selector']

pages/main.py

@@ -155,6 +155,9 @@ def tts_settings(name__tts_voice,
 
 @st.fragment
 def recorder_released():
+    if "audio_list" not in st.session_state:
+        st.session_state.audio_list = []
+    
     if "rec_widget" in st.session_state:
         if st.session_state.rec_widget:
             audio_recorded = True
@@ -285,10 +288,17 @@ def recorder_released():
 
             except Exception as e:
                 st.error(f"[AUDIO] - {get_translation('erreur_importation_audio')}: {str(e)}")
+            #finally:
+            #    st.session_state.ui_chat_input_disabled = False
+            #    st.session_state.ui_audio_input_disabled = False
+            #    st.session_state.ui_filesuploader_disabled = False
 
 def main_page():
     """Page principale de l'application."""
 
+    if "audio_list" not in st.session_state:
+        st.session_state.audio_list = []
+
     if "ui_chat_input_disabled" not in st.session_state:
         st.session_state.ui_chat_input_disabled = False
 
@@ -402,93 +412,44 @@ def main_page():
         pass
     else:
         with st.container(border=True):
-            # Interface utilisateur pour le chat textuel
-            st.session_state.user_input = st.chat_input(
-                get_translation("entrez_message"),
-                disabled=st.session_state.ui_chat_input_disabled
-            )
-
-
-            # Interface utilisateur pour l'upload de fichiers
-            st.session_state.uploaded_files = st.file_uploader(
-                "Choose files to upload", 
-                accept_multiple_files=True,
-                type=CHAT_FILES_UPLOAD_ALLOWED_TYPES,
-                key="chat_files_upload",
-                disabled=st.session_state.ui_filesuploader_disabled
-            )
-
-            if st.session_state.uploaded_files is not None:
-                new_uploaded_files_hashes = []
-                for file in st.session_state.uploaded_files:
-                    uploaded_file_hash = hash_file(file)
-                    new_uploaded_files_hashes.append(uploaded_file_hash)
-                    if uploaded_file_hash not in st.session_state.changed_uploaded_files.keys():
-                        uploaded_file_name = file.name
-                        uploaded_file_type = os.path.splitext(file.name)[1].lstrip('.')
-                        st.session_state.changed_uploaded_files[uploaded_file_hash] = {
-                            "name": uploaded_file_name,
-                            "type": uploaded_file_type,
-                            "bytes_data": file.read()
-                        }
-                        if uploaded_file_type in ["wav", "mp3"]:
-                            audio = AudioSegment.from_wav(io.BytesIO(st.session_state.changed_uploaded_files[uploaded_file_hash]["bytes_data"]))
-                            with tempfile.NamedTemporaryFile(suffix=f".{uploaded_file_type}", delete=False) as tmp_file:
-                                audio.export(tmp_file, format=uploaded_file_type)
-                                tmp_file.close()
-                                st.session_state.changed_uploaded_files[uploaded_file_hash]["audio_transcription"] = transcribe_audio(
-                                    filepath=tmp_file.name
-                                )
-                                audio = AudioSegment.empty()
-
-                changed_uploaded_files = {}
-                for file_hash in st.session_state.changed_uploaded_files.keys():
-                    if file_hash in new_uploaded_files_hashes:
-                        changed_uploaded_files[file_hash] = st.session_state.changed_uploaded_files[file_hash]
-                st.session_state.changed_uploaded_files = changed_uploaded_files
 
-                for key, value in st.session_state.changed_uploaded_files.items():
-                    if key in new_uploaded_files_hashes:
-                        with st.container(border=True):
-                            if value["type"] in ["txt"]:
-                                st.write(f"**filename:** {value['name']}")
-                                st.write(f"**filetype:** {value['type']}")
-                                file_content = value["bytes_data"].decode()
-                                with st.expander("View file content"):
-                                    st.text_area( f"Edit text - {value["name"]}", 
-                                                  value=file_content, 
-                                                  key=value["name"],
-                                                  on_change=callback_change_edited_text, 
-                                                  args=(key, value, value["name"])
-                                                )
-                            elif value["type"] in ["wav", "mp3"]:
-                                st.write(f"**Filename:** {value['name']}")
-                                audio = AudioSegment.from_file(io.BytesIO(value["bytes_data"]))
-                                st.write(f"Frame rate: {audio.frame_rate}, Frame width: {audio.frame_width}, Duration: {audio.duration_seconds} seconds")
-                                st.audio(value["bytes_data"], 
-                                         format=f"audio/{value['type']}", 
-                                         autoplay=False
-                                         )
-                                if "audio_transcription" in value.keys():
-                                    audio_file_transcription = value["audio_transcription"]
-                                    with st.expander("View audio transcription"):
-                                        st.text_area( f"Edit text - {value['name']}", 
-                                                     value=audio_file_transcription, 
-                                                     key=value["name"], 
-                                                     on_change=callback_change_edited_text, 
-                                                     args=(key, value, value["name"])
-                                                    )
+            chat_input_tabs1, chat_input_tabs2, chat_input_tabs3 = st.tabs(["text_input", "audio_input", "file_upload_input"])
+            
+            st.session_state.ui_chat_input_disabled = False
+            st.session_state.ui_audio_input_disabled = False
+            st.session_state.ui_filesuploader_disabled = False
+
+            with chat_input_tabs1:
+                # Interface utilisateur pour le chat textuel
+                st.session_state.user_input = st.chat_input(
+                    get_translation("entrez_message"),
+                    disabled=st.session_state.ui_chat_input_disabled
+                )
 
+            with chat_input_tabs3:
+                # Interface utilisateur pour l'upload de fichiers
+                st.session_state.uploaded_files = st.file_uploader(
+                    "Choose files to upload", 
+                    accept_multiple_files=True,
+                    type=CHAT_FILES_UPLOAD_ALLOWED_TYPES,
+                    key="chat_files_upload",
+                    disabled=st.session_state.ui_filesuploader_disabled
+                )
 
-            # Interface utilisateur pour l'enregistrement audio
-            st.audio_input(
-                    "Record a voice message",
-                    on_change=recorder_released, 
-                    key="rec_widget",
-                    disabled=st.session_state.ui_audio_input_disabled
-            )
+            with chat_input_tabs2:
+                # Interface utilisateur pour l'enregistrement audio
+                st.audio_input(
+                        "Record a voice message",
+                        on_change=recorder_released, 
+                        key="rec_widget",
+                        disabled=st.session_state.ui_audio_input_disabled
+                )
 
             if st.session_state.user_input:
+                #st.session_state.ui_chat_input_disabled = True
+                #st.session_state.ui_audio_input_disabled = True
+                #st.session_state.ui_filesuploader_disabled = True
+
                 # Appeler la fonction de modération
                 moderation_result = api_moderation_openai_text(st.session_state.user_input)
                 if moderation_result.get("flagged"):
@@ -570,37 +531,110 @@ def main_page():
                             else:
                                 response_status.update(label=f'({target_language_name}) - {get_translation("erreur_traduction")}', state="error", expanded=False)
 
-                if st.session_state.audio_list:
-                    with st.status(f"{get_translation('concatenation_audio_en_cours')}", expanded=False) as audio_status:
-                        audio_status.update(label=f"{get_translation('concatenation_audio_en_cours')}", state="running", expanded=False)
-                        try:
-                            st.session_state.final_audio = concatenate_audio_files(st.session_state.audio_list)
-                            with st.container(border=True):
-
-                                # Générer un nom de fichier unique
-                                st.session_state.timestamp = time.strftime("%Y%m%d-%H%M%S")
-                                st.session_state.langues = "_".join([lang["iso-639-1"] for lang in st.session_state.selected_languages])
-                                st.session_state.nom_fichier = f"reponse_audio_{st.session_state.langues}_{st.session_state.timestamp}.mp3"
-
-                                st.audio(st.session_state.final_audio, format="audio/mp3", autoplay=st.session_state.autoplay_tts)
-
-                                st.download_button(
-                                    label=f"📥 {get_translation('telecharger_audio')}",
-                                    data=st.session_state.final_audio,
-                                    file_name=st.session_state.nom_fichier,
-                                    mime="audio/mp3",
-                                    use_container_width=True,
-                                    type="primary",
-                                    key=f"download_button_{st.session_state.langues}_{st.session_state.timestamp}",
-                                )
 
-                                audio_status.update(label=f"{get_translation('concatenation_audio_terminee')}", state="complete", expanded=True)
-                        except Exception as e:
-                            st.error(f"{get_translation('erreur_concatenation_audio')} : {str(e)}")
-                            
-                            audio_status.update(label=f"{get_translation('erreur_concatenation_audio')} : {str(e)}", state="error", expanded=True)
+
+            if st.session_state.audio_list:
+                #st.session_state.ui_chat_input_disabled = True
+                #st.session_state.ui_audio_input_disabled = True
+                #st.session_state.ui_filesuploader_disabled = True
+
+                with st.status(f"{get_translation('concatenation_audio_en_cours')}", expanded=False) as audio_status:
+                    audio_status.update(label=f"{get_translation('concatenation_audio_en_cours')}", state="running", expanded=False)
+                    try:
+                        st.session_state.final_audio = concatenate_audio_files(st.session_state.audio_list)
+                        with st.container(border=True):
+
+                            # Générer un nom de fichier unique
+                            st.session_state.timestamp = time.strftime("%Y%m%d-%H%M%S")
+                            st.session_state.langues = "_".join([lang["iso-639-1"] for lang in st.session_state.selected_languages])
+                            st.session_state.nom_fichier = f"reponse_audio_{st.session_state.langues}_{st.session_state.timestamp}.mp3"
+
+                            st.audio(st.session_state.final_audio, format="audio/mp3", autoplay=st.session_state.autoplay_tts)
+
+                            st.download_button(
+                                label=f"📥 {get_translation('telecharger_audio')}",
+                                data=st.session_state.final_audio,
+                                file_name=st.session_state.nom_fichier,
+                                mime="audio/mp3",
+                                use_container_width=True,
+                                type="primary",
+                                key=f"download_button_{st.session_state.langues}_{st.session_state.timestamp}",
+                            )
+
+                            audio_status.update(label=f"{get_translation('concatenation_audio_terminee')}", state="complete", expanded=True)
+                    except Exception as e:
+                        st.error(f"{get_translation('erreur_concatenation_audio')} : {str(e)}")
                         
+                        audio_status.update(label=f"{get_translation('erreur_concatenation_audio')} : {str(e)}", state="error", expanded=True)
+                    #finally:
+                    #    st.session_state.ui_chat_input_disabled = False
+                    #    st.session_state.ui_audio_input_disabled = False
+                    #    st.session_state.ui_filesuploader_disabled = False
+
+            if st.session_state.uploaded_files is not None:
+                #st.session_state.ui_chat_input_disabled = True
+                #st.session_state.ui_audio_input_disabled = True
+                #st.session_state.ui_filesuploader_disabled = True
 
+                new_uploaded_files_hashes = []
+                for file in st.session_state.uploaded_files:
+                    uploaded_file_hash = hash_file(file)
+                    new_uploaded_files_hashes.append(uploaded_file_hash)
+                    if uploaded_file_hash not in st.session_state.changed_uploaded_files.keys():
+                        uploaded_file_name = file.name
+                        uploaded_file_type = os.path.splitext(file.name)[1].lstrip('.')
+                        st.session_state.changed_uploaded_files[uploaded_file_hash] = {
+                            "name": uploaded_file_name,
+                            "type": uploaded_file_type,
+                            "bytes_data": file.read()
+                        }
+                        if uploaded_file_type in ["wav", "mp3"]:
+                            audio = AudioSegment.from_wav(io.BytesIO(st.session_state.changed_uploaded_files[uploaded_file_hash]["bytes_data"]))
+                            with tempfile.NamedTemporaryFile(suffix=f".{uploaded_file_type}", delete=False) as tmp_file:
+                                audio.export(tmp_file, format=uploaded_file_type)
+                                tmp_file.close()
+                                st.session_state.changed_uploaded_files[uploaded_file_hash]["audio_transcription"] = transcribe_audio(
+                                    filepath=tmp_file.name
+                                )
+                                audio = AudioSegment.empty()
+
+                changed_uploaded_files = {}
+                for file_hash in st.session_state.changed_uploaded_files.keys():
+                    if file_hash in new_uploaded_files_hashes:
+                        changed_uploaded_files[file_hash] = st.session_state.changed_uploaded_files[file_hash]
+                st.session_state.changed_uploaded_files = changed_uploaded_files
+
+                for key, value in st.session_state.changed_uploaded_files.items():
+                    if key in new_uploaded_files_hashes:
+                        with st.container(border=True):
+                            if value["type"] in ["txt"]:
+                                st.write(f"**filename:** {value['name']}")
+                                st.write(f"**filetype:** {value['type']}")
+                                file_content = value["bytes_data"].decode()
+                                with st.expander("View file content"):
+                                    st.text_area( f"Edit text - {value["name"]}", 
+                                                value=file_content, 
+                                                key=value["name"],
+                                                on_change=callback_change_edited_text, 
+                                                args=(key, value, value["name"])
+                                                )
+                            elif value["type"] in ["wav", "mp3"]:
+                                st.write(f"**Filename:** {value['name']}")
+                                audio = AudioSegment.from_file(io.BytesIO(value["bytes_data"]))
+                                st.write(f"Frame rate: {audio.frame_rate}, Frame width: {audio.frame_width}, Duration: {audio.duration_seconds} seconds")
+                                st.audio(value["bytes_data"], 
+                                        format=f"audio/{value['type']}", 
+                                        autoplay=False
+                                        )
+                                if "audio_transcription" in value.keys():
+                                    audio_file_transcription = value["audio_transcription"]
+                                    with st.expander("View audio transcription"):
+                                        st.text_area( f"Edit text - {value['name']}", 
+                                                    value=audio_file_transcription, 
+                                                    key=value["name"], 
+                                                    on_change=callback_change_edited_text, 
+                                                    args=(key, value, value["name"])
+                                                    )
 
 
 main_page()