Update. Improved functionality

.gitignore

@@ -33,5 +33,6 @@ ENV/
 .ipynb_checkpoints/
 
 data/*.mp4
+data/*.webm
 
 .idea

streamlit_app.py

@@ -1,4 +1,5 @@
 import os
+from glob import glob
 import yt_dlp
 import string
 import streamlit as st
@@ -12,14 +13,13 @@ if "downloaded" not in st.session_state:
     st.session_state["downloaded"] = []
 
 def download_video(url:str, save_path:str , download:bool, **kwargs):
+    video_id =url.split("/")[-1]
     ydl_opts = {
-        'outtmpl': f'{save_path}/video.%(ext)s',
+        'outtmpl': f'{save_path}/{video_id}.%(ext)s',
     }
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         info_dict = ydl.extract_info(url, download=download)
-        video_title = info_dict.get('title', None)
-        video_title = f"{video_title}"
-        return video_title
+        return video_id
 
 
 st.title("Download Any Video")
@@ -35,11 +35,20 @@ if st.session_state["url"]:
         st.session_state["downloaded"].append(st.session_state["url"])
 
 if st.session_state["videos"]:
-    st.write("**Title**:", st.session_state["videos"])
-    st.video(os.path.join(".", "data", "video.mp4"))
-    video_file = open("data/video.mp4", "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")
+
+    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