Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from pytube import YouTube
+from pytube.exceptions import VideoUnavailable
 
 class YouTubeDownloader:
     @staticmethod
@@ -8,27 +9,38 @@ class YouTubeDownloader:
         url = st.text_input("Enter YouTube URL to download:")
         if url:
             YouTubeDownloader.validate_url(url)
-            with st.expander("preview video"):
-                st.video(url)
-            if st.button("Download"):
-                YouTubeDownloader.cleanup()
-                file_ = YouTubeDownloader.download_video(url)
-                st.video(file_)
-                YouTubeDownloader.helper_message()
+            try:
+                with st.expander("preview video"):
+                    st.video(url)
+                if st.button("Download"):
+                    YouTubeDownloader.cleanup()
+                    file_ = YouTubeDownloader.download_video(url)
+                    if file_:  # file_ 이 None이 아닐 때만 비디오 표시
+                        st.video(file_)
+                        YouTubeDownloader.helper_message()
+            except Exception as e:
+                st.error(f"Error: {str(e)}")
         st.markdown("예) https://www.youtube.com/watch?v=hXpNj1ChxRI ")
 
-
     @staticmethod
     def download_video(url):
-        with st.spinner("Downloading..."):
-            local_file = (
-                YouTube(url)
-                .streams.filter(progressive=True, file_extension="mp4")
-                .first()
-                .download()
-            )
-            st.success("Downloaded")
-        return local_file
+        try:
+            with st.spinner("Downloading..."):
+                yt = YouTube(url)
+                stream = yt.streams.filter(progressive=True, file_extension="mp4").first()
+                if stream:
+                    local_file = stream.download()
+                    st.success("Downloaded")
+                    return local_file
+                else:
+                    st.error("No suitable stream found for this video")
+                    return None
+        except VideoUnavailable:
+            st.error("This video is unavailable")
+            return None
+        except Exception as e:
+            st.error(f"An error occurred: {str(e)}")
+            return None
 
     @staticmethod
     def validate_url(url):
@@ -53,6 +65,5 @@ class YouTubeDownloader:
             "영상 화면 우측 하단 ... 버튼을 클릭하면 다운로드 할 수 있습니다."
         )
 
-
 if __name__ == "__main__":
     YouTubeDownloader.run()