Added cookies

streamlit_app.py

@@ -24,9 +24,17 @@ def _remove_video():
 
 def download_video(url:str, save_path:str ,  **kwargs):
     video_id =url.split("/")[-1]
+
+    if len(video_id) < 1:
+        video_id = url.split("/")[-2]
+
     ydl_opts = {
         'outtmpl': f'{save_path}/{video_id}.%(ext)s',
     }
+
+    if st.session_state["ig_cookie"]:
+        ydl_opts['cookies'] = st.session_state["ig_cookie"]
+
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         ydl.download(url)
 
@@ -44,6 +52,14 @@ def download_video(url:str, save_path:str ,  **kwargs):
 
 st.title("Download Any* Video")
 
+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.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.text_input("Video URL", key="url", placeholder="Enter Video URL and Hit Enter", on_change=_remove_video)