Error management

streamlit_app.py

@@ -22,7 +22,8 @@ def _remove_video():
         except FileNotFoundError:
             print("Couldn't delete file")
 
-def download_video(url:str, save_path:str ,  **kwargs):
+
+def download_video(url: str, save_path: str,  **kwargs):
     video_id =url.split("/")[-1]
 
     if len(video_id) < 1:
@@ -52,6 +53,7 @@ def download_video(url:str, save_path:str ,  **kwargs):
 
 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")
@@ -60,6 +62,7 @@ with st.sidebar:
     st.file_uploader(label="**Upload TikTok Cookie** :cookie:", type=["txt", "json"],
                      accept_multiple_files=False, key="tt_cookie")
 
+# Main content
 st.text_input("Video URL", key="url", placeholder="Enter Video URL and Hit Enter", on_change=_remove_video)
 
 
@@ -70,11 +73,19 @@ if st.session_state["url"]:
         st.session_state["videos"] = video_path
 
 if st.session_state["videos"]:
-    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")
+    st.write("**Title**:", st.session_state["videos"].split("/")[-1])
+
+    try:
+        st.video(st.session_state["videos"])
+    except Exception as e:
+        st.write("Couldn't show video")
+
+    try:
+        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")
+    except Exception as e:
+        st.error("Failed downloading the video", icon="😔")