Update app.py

app.py

@@ -1241,20 +1241,39 @@ def get_audio_download_link(file_path):
 
 
 
-
-
 # 🎵 Wav Audio files - Transcription History in Wav
-all_files = glob.glob("*.wav")
-all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 10]  # exclude files with short names
-all_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)  # sort by file type and file name in descending order
+audio_files = glob.glob("*.wav")
+audio_files = [file for file in audio_files if len(os.path.splitext(file)[0]) >= 10]  # exclude files with short names
+audio_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)  # sort by file type and file name in descending order
+
+# 🖼 PNG Image files
+image_files = glob.glob("*.png")
+image_files = [file for file in image_files if len(os.path.splitext(file)[0]) >= 10]  # exclude files with short names
+image_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)  # sort by file type and file name in descending order
+
+# 🎥 MP4 Video files
+video_files = glob.glob("*.mp4")
+video_files = [file for file in video_files if len(os.path.splitext(file)[0]) >= 10]  # exclude files with short names
+video_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)  # sort by file type and file name in descending order
+
+# Delete All button for each file type
+if st.sidebar.button("🗑 Delete All Audio"):
+    for file in audio_files:
+        os.remove(file)
+    st.rerun()
 
-filekey = 'delall'
-if st.sidebar.button("🗑 Delete All Audio", key=filekey):
-    for file in all_files:
+if st.sidebar.button("🗑 Delete All Images"):
+    for file in image_files:
+        os.remove(file)
+    st.rerun()
+
+if st.sidebar.button("🗑 Delete All Videos"):
+    for file in video_files:
         os.remove(file)
     st.rerun()
 
-for file in all_files:
+# Display and handle audio files
+for file in audio_files:
     col1, col2 = st.sidebar.columns([6, 1])  # adjust the ratio as needed
     with col1:
         st.markdown(file)
@@ -1262,8 +1281,33 @@ for file in all_files:
             audio_file = open(file, 'rb')
             audio_bytes = audio_file.read()
             st.audio(audio_bytes, format='audio/wav')
-            #st.markdown(get_audio_download_link(file), unsafe_allow_html=True)
-            #st.text_input(label="", value=file)
+    with col2:
+        if st.button("🗑", key="delete_" + file):
+            os.remove(file)
+            st.rerun()
+
+# Display and handle image files
+for file in image_files:
+    col1, col2 = st.sidebar.columns([6, 1])  # adjust the ratio as needed
+    with col1:
+        st.markdown(file)
+        if st.button("🖼", key="show_" + file):  # show emoji button
+            image = open(file, 'rb').read()
+            st.image(image)
+    with col2:
+        if st.button("🗑", key="delete_" + file):
+            os.remove(file)
+            st.rerun()
+
+# Display and handle video files
+for file in video_files:
+    col1, col2 = st.sidebar.columns([6, 1])  # adjust the ratio as needed
+    with col1:
+        st.markdown(file)
+        if st.button("🎥", key="play_" + file):  # play emoji button
+            video_file = open(file, 'rb')
+            video_bytes = video_file.read()
+            st.video(video_bytes)
     with col2:
         if st.button("🗑", key="delete_" + file):
             os.remove(file)