Managed Memory

streamlit_app.py

@@ -5,50 +5,60 @@ import string
 import streamlit as st
 
 
-# Callbacks
+# Config
 if "videos" not in st.session_state:
     st.session_state["videos"] = ""
 
 if "downloaded" not in st.session_state:
     st.session_state["downloaded"] = []
 
-def download_video(url:str, save_path:str , download:bool, **kwargs):
+
+# Callbacks
+def _remove_video():
+    if st.session_state["videos"]:
+        try:
+            os.remove(st.session_state["videos"])
+            st.session_state["videos"] = ""
+        except FileNotFoundError:
+            print("Couldn't delete file")
+
+def download_video(url:str, save_path:str ,  **kwargs):
     video_id =url.split("/")[-1]
     ydl_opts = {
         'outtmpl': f'{save_path}/{video_id}.%(ext)s',
     }
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        info_dict = ydl.extract_info(url, download=download)
-        return video_id
+        ydl.download(url)
 
+    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"))
 
-st.title("Download Any Video")
+    print("videos found:", videos)
+    for video in videos:
+        if video_id in video:
+            video_path = video
+            return video_path
 
-st.text_input("Video URL", key="url", placeholder="Enter Video URL and Hit Enter")
+
+st.title("Download Any* Video")
+
+st.text_input("Video URL", key="url", placeholder="Enter Video URL and Hit Enter", on_change=_remove_video)
 
 
 if st.session_state["url"]:
-    download = False if st.session_state["url"] in st.session_state["downloaded"] else True
-    video_path = download_video(st.session_state["url"], save_path=os.path.join(".", "data"), download=download)
-    st.session_state["videos"] = video_path
-    if not download:
-        st.session_state["downloaded"].append(st.session_state["url"])
+    download = st.button("Download")
+    if download:
+        video_path = download_video(st.session_state["url"], save_path=os.path.join(".", "data"))
+        st.session_state["videos"] = video_path
 
 if st.session_state["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"))
-    print("videos found:", videos)
-    for video in videos:
-        if st.session_state["videos"] in video:
-            st.write("**Title**:", st.session_state["videos"])
-            st.video(video)
-            video_file = open(video, "rb")
-            st.download_button("**Download Video 📁**",
-                               data=video_file,
-                               file_name=f"{st.session_state['videos'].lower().translate(str.maketrans('','', string.punctuation)).replace(' ', '_')}.mp4",
-                               mime="video/mp4",
-                               type="primary")
-            break
+    st.write("**Title**:", st.session_state["videos"])
+    st.video(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",
+                       mime="video/mp4",
+                       type="primary")