update files upload feature

pages/main.py

@@ -65,7 +65,11 @@ def hash_file(file):
     return hasher.hexdigest()
 
 def callback_change_edited_text(key, value, modified_text):
-    st.session_state.changed_uploaded_files[key]["bytes_data"] = st.session_state[modified_text].encode()
+    if value["type"] in ["txt"]:
+        st.session_state.changed_uploaded_files[key]["bytes_data"] = st.session_state[modified_text].encode()
+    elif value["type"] in ["wav", "mp3"]:
+        st.session_state.changed_uploaded_files[key]["audio_transcription"] = st.session_state[modified_text]
+    
     #print(st.session_state.changed_uploaded_files[key])
 
 def save_attachment(attachment):
@@ -408,6 +412,16 @@ def main_page():
                             "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:
@@ -436,6 +450,14 @@ def main_page():
                                          format=f"audio/{value['type']}", 
                                          autoplay=False
                                          )
+                                if "audio_transcription" in value.keys():
+                                    with st.expander("View audio transcription"):
+                                        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"])
+                                                    )
 
 
             # Interface utilisateur pour l'enregistrement audio