update yt-dlp added video cleanup option

requirements.txt

@@ -1,2 +1,2 @@
 streamlit==1.27.2
-yt-dlp==2023.10.13
+yt-dlp

streamlit_app.py

@@ -4,7 +4,6 @@ import yt_dlp
 import string
 import streamlit as st
 
-
 # Config
 if "videos" not in st.session_state:
     st.session_state["videos"] = ""
@@ -23,8 +22,21 @@ def _remove_video():
             print("Couldn't delete file")
 
 
-def download_video(url: str, save_path: str,  **kwargs):
-    video_id =url.split("/")[-1]
+def _remove_all_videos():
+    videos = glob(os.path.join(".", "data", "*.mp4")) \
+             + glob(os.path.join(".", "data", "*.webm")) \
+             + glob(os.path.join(".", "data", "*.mov")) \
+             + glob(os.path.join(".", "data", "*.m4a"))
+    for video in videos:
+        try:
+            os.remove(video)
+        except FileNotFoundError:
+            print("Couldn't delete file:", video)
+
+
+# Main function
+def download_video(url: str, save_path: str, **kwargs):
+    video_id = url.split("/")[-1]
 
     if len(video_id) < 1:
         video_id = url.split("/")[-2]
@@ -56,16 +68,17 @@ st.title("Download Any* Video")
 # Side bar content
 with st.sidebar:
     st.subheader("Cookies! :cookie:")
-    st.write("For some websites that have a more strict bot control it's necessary to upload a cookie file to download content from them")
+    st.write(
+        "For some websites that have a more strict bot control it's necessary to upload a cookie file to download content from them")
     st.file_uploader(label="**Upload Instagram Cookie** :cookie:", type=["txt", "json"],
                      accept_multiple_files=False, key="ig_cookie")
     st.file_uploader(label="**Upload TikTok Cookie** :cookie:", type=["txt", "json"],
                      accept_multiple_files=False, key="tt_cookie")
+    st.button("Clear Videos", on_click=_remove_all_videos)
 
 # Main content
 st.text_input("Video URL", key="url", placeholder="Enter Video URL and Hit Enter", on_change=_remove_video)
 
-
 if st.session_state["url"]:
     download = st.button("Load Video")
     if download:
@@ -84,8 +97,8 @@ if st.session_state["videos"]:
         video_file = open(st.session_state["videos"], "rb")
         st.download_button("**Download Video 📁**",
                            data=video_file,
-                           file_name=f"{st.session_state['videos'].lower().translate(str.maketrans('','', string.punctuation)).replace(' ', '_')}.mp4",
+                           file_name=f"{st.session_state['videos'].lower().translate(str.maketrans('', '', string.punctuation)).replace(' ', '_')}.mp4",
                            mime="video/mp4",
                            type="primary")
     except Exception as e:
-        st.error("Failed downloading the video", icon="😔")
+        st.error("Failed downloading the video", icon="😔")