add files upload feature

pages/main.py

@@ -17,7 +17,7 @@ from typing import Tuple
 from typing import Union
 from io import BytesIO
 from copy import deepcopy
-
+import hashlib
 
 # Third-party libraries
 import requests
@@ -57,6 +57,13 @@ from core.core import init_process_mode
 from core.moderation import api_moderation_openai_text
 from core.audio_isolation import isolate_audio
 
+def hash_file(file):
+    hasher = hashlib.md5()
+    buf = file.read()
+    hasher.update(buf)
+    file.seek(0)
+    return hasher.hexdigest()
+
 
 def save_attachment(attachment):
     """Sauvegarde la pièce jointe et retourne le chemin."""
@@ -267,6 +274,18 @@ def recorder_released():
 def main_page():
     """Page principale de l'application."""
 
+    # Dictionnaire pour stocker les fichiers modifiés
+    if 'changed_uploaded_files' not in st.session_state:
+        st.session_state.changed_uploaded_files = {}
+
+    # Dictionnaire pour stocker le contenu modifié des fichiers
+    if 'edited_texts' not in st.session_state:
+        st.session_state.edited_texts = {}
+
+    # Liste pour stocker les fichiers audio
+    if 'audio_files' not in st.session_state:
+        st.session_state.audio_files = []
+
     # Initialisation des variables d'état de session
     if "ui_loaded" not in st.session_state:
         st.session_state["ui_loaded"] = False
@@ -364,66 +383,92 @@ def main_page():
                 get_translation("entrez_message")
             )
 
-            def uploaded_files__on_change():
-                if st.session_state.uploaded_files is not None:
-                    st.session_state.edited_texts = {}
-                    st.session_state.audio_files = []
-                    for uploaded_file in st.session_state.uploaded_files:
-                        with st.container(border=True):
-                            bytes_data = uploaded_file.read()
-                            file_name = uploaded_file.name
-                            filetype = os.path.splitext(uploaded_file.name)[1].lstrip('.')
-                            if filetype in ["txt"]:
-                                st.write("**filename:**", uploaded_file.name)
-                                st.write("**filetype:**", filetype)
-                                file_content = bytes_data.decode()
-                                with st.expander("View file content"):
-                                    edited_text = st.text_area(f"Edit text - {file_name}", value=file_content, key=file_name)
-                                    st.session_state.edited_texts[file_name] = edited_text
-                            elif filetype in ["wav", "mp3"]:
-                                st.write(f"**Filename:** {file_name}")
-                                audio = AudioSegment.from_file(io.BytesIO(bytes_data))
-                                st.write(f"Frame rate: {audio.frame_rate}, Frame width: {audio.frame_width}, Duration: {audio.duration_seconds} seconds")
-                                st.audio(bytes_data, format=f"audio/{filetype}", autoplay=False)
-                                st.session_state.audio_files.append(audio)
-
-                if len(st.session_state.uploaded_files) > 0:
-                    filetypes_uploaded_list = [ os.path.splitext(uploaded_file.name)[1].lstrip('.') for uploaded_file in st.session_state.uploaded_files ]
-                    if st.button("Submit All"):
-                        if ("txt" in filetypes_uploaded_list):
-                            # Concaténer les textes modifiés pour le traitement
-                            concatenated_text = "\n".join(st.session_state.edited_texts.values())
-                            st.write("Processing concatenated text...")
-                            # Ajoutez ici le code pour traiter le texte concaténé
-                            st.write(concatenated_text)
-                        if ( "wav" in filetypes_uploaded_list or "mp3" in filetypes_uploaded_list ):
-                            # Concaténer les fichiers audio pour le traitement
-                            concatenated_audio = sum(st.session_state.audio_files)
-                            audio_buffer = io.BytesIO()
-                            concatenated_audio.export(audio_buffer, format="wav")
-                            st.write("Processing concatenated audio...")
-                            st.audio(audio_buffer.getvalue(), format="audio/wav", autoplay=False)
-
 
             # 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",
-                on_change=uploaded_files__on_change
+                key="chat_files_upload"
             )
-            
-            # Dictionnaire pour stocker le contenu modifié des fichiers
-            if 'edited_texts' not in st.session_state:
-                st.session_state.edited_texts = {}
-
-            # Liste pour stocker les fichiers audio
-            if 'audio_files' not in st.session_state:
-                st.session_state.audio_files = []
-
-
 
+            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()
+                        }
+                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():
+                    with st.container(border=True):
+                        if value["type"] in ["txt"]:
+                            st.write(f"**filename:** {value['name']}")
+
+                    #st.session_state.edited_texts = {}
+                    #st.session_state.audio_files = []
+                    #for uploaded_file in st.session_state.uploaded_files:
+                    #    with st.container(border=True):
+                    #        bytes_data = uploaded_file.read()
+                    #        file_name = uploaded_file.name
+                    #        filetype = os.path.splitext(uploaded_file.name)[1].lstrip('.')
+                    #        if filetype in ["txt"]:
+                    #            st.write("**filename:**", uploaded_file.name)
+                    #            st.write("**filetype:**", filetype)
+                    #            file_content = bytes_data.decode()
+                                #with st.expander("View file content"):
+                                #    edited_text = st.text_area(f"Edit text - {file_name}", value=file_content, key=file_name)
+                                #    st.session_state.edited_texts[file_name] = edited_text
+                            #elif filetype in ["wav", "mp3"]:
+                            #    st.write(f"**Filename:** {file_name}")
+                            #    audio = AudioSegment.from_file(io.BytesIO(bytes_data))
+                            #    st.write(f"Frame rate: {audio.frame_rate}, Frame width: {audio.frame_width}, Duration: {audio.duration_seconds} seconds")
+                            #    st.audio(bytes_data, format=f"audio/{filetype}", autoplay=False)
+                            #    st.session_state.audio_files.append(audio)
+            
+  
+            # Afficher et modifier le contenu de chaque fichier texte
+            #for file_name, edited_text in st.session_state.edited_texts.items():
+            #    with st.container():
+            #        st.write(f"**Filename:** {file_name}")
+            #        with st.expander("View and edit file content"):
+            #            edited_text = st.text_area(f"Edit text - {file_name}", value=edited_text, key=file_name)
+            #            st.session_state.edited_texts[file_name] = edited_text
+
+            # Afficher les fichiers audio
+            #for audio in st.session_state.audio_files:
+            #    with st.container():
+            #        st.write(f"Frame rate: {audio.frame_rate}, Frame width: {audio.frame_width}, Duration: {audio.duration_seconds} seconds")
+            #        audio_buffer = io.BytesIO()
+            #        audio.export(audio_buffer, format="wav")
+            #        st.audio(audio_buffer.getvalue(), format="audio/wav", autoplay=False)
+
+            # Concaténer les textes modifiés pour le traitement
+            #if st.button("Submit All Texts"):
+            #    concatenated_text = "\n".join(st.session_state.edited_texts.values())
+            #    st.write("Processing concatenated text...")
+            #    # Ajoutez ici le code pour traiter le texte concaténé
+            #    st.write(concatenated_text)
+
+            # Concaténer les fichiers audio pour le traitement
+            #if st.button("Submit All Audios"):
+            ##    if st.session_state.audio_files:
+             #       concatenated_audio = sum(st.session_state.audio_files)
+             #       audio_buffer = io.BytesIO()
+             #       concatenated_audio.export(audio_buffer, format="mp3")
+             #       st.write("Processing concatenated audio...")
+             #       st.audio(audio_buffer.getvalue(), format="audio/mp3", autoplay=False)
 
 
             # Interface utilisateur pour l'enregistrement audio