Update app.py

app.py

@@ -1,49 +1,66 @@
-import streamlit as st
-from pytube import YouTube
-import os
-
-# Streamlit 제목
-st.title('YouTube Video Downloader')
-
-# 유저로부터 유튜브 링크 입력받기
-youtube_url = st.text_input('Enter YouTube video URL')
-
-if youtube_url:
-    try:
-        # YouTube 객체 생성
-        yt = YouTube(youtube_url)
-
-        # 썸네일 이미지 표시
-        thumbnail_url = yt.thumbnail_url
-        st.image(thumbnail_url, caption='Video Thumbnail', use_column_width=True)
-
-        # 해상도 선택
-        resolution = st.selectbox('Select resolution', ['1080p', '720p'])
-
-        # 선택된 해상도의 비디오 스트림 선택
-        stream = yt.streams.filter(res=resolution, progressive=True, file_extension='mp4').first()
-
-        # 1080p가 존재하지 않으면 720p 스트림 선택
-        if resolution == '1080p' and stream is None:
-            st.warning("1080p가 존재하지 않습니다. 720p를 다운로드합니다.")
-            stream = yt.streams.filter(res='720p', progressive=True, file_extension='mp4').first()
-
-        if stream is None:
-            st.error(f"{resolution} 해상도의 스트림을 찾을 수 없습니다.")
-        else:
-            # 다운로드 버튼
-            if st.button('Download Video'):
-                download_path = stream.download()
-                st.success('Video downloaded successfully!')
-                st.video(download_path)
-
-                # 다운로드된 파일 제공 (optional)
-                with open(download_path, 'rb') as file:
-                    btn = st.download_button(
-                        label="Download Video File",
-                        data=file,
-                        file_name=os.path.basename(download_path),
-                        mime='video/mp4'
-                    )
-    except Exception as e:
-        st.error(f"An error occurred: {e}")
+import streamlit as st
+from pytube import YouTube
+import os
+import time
+
+# Streamlit 제목
+st.title('YouTube Video Downloader')
+
+# 유저로부터 유튜브 링크 입력받기
+youtube_url = st.text_input('Enter YouTube video URL')
+
+# "확인" 버튼 추가
+if st.button('확인'):
+    if youtube_url:
+        try:
+            yt = YouTube(youtube_url)
+            # 썸네일 이미지 표시
+            thumbnail_url = yt.thumbnail_url
+            st.image(thumbnail_url, caption='Video Thumbnail', use_column_width=True)
+            
+            # 해상도 선택
+            resolution = st.selectbox('Select resolution', ['1080p', '1080p60', '1080pHD', '720p'])
+
+            def fetch_video_stream(youtube_url, resolution):
+                yt = YouTube(youtube_url)
+
+                # 딜레이 추가
+                time.sleep(2)  # 2초 딜레이
+
+                # 선택된 해상도의 비디오 스트림 선택
+                stream = yt.streams.filter(res=resolution.replace('pHD', 'p'), progressive=True, file_extension='mp4').first()
+
+                # 1080p 변형들이 존재하지 않으면 720p 스트림 선택
+                if '1080p' in resolution and stream is None:
+                    st.warning(f"{resolution}가 존재하지 않습니다. 720p를 다운로드합니다.")
+                    stream = yt.streams.filter(res='720p', progressive=True, file_extension='mp4').first()
+
+                return yt, stream
+
+            # "추출" 버튼 추가
+            if st.button('추출'):
+                yt, stream = fetch_video_stream(youtube_url, resolution)
+
+                if stream is None:
+                    st.error(f"{resolution} 해상도의 스트림을 찾을 수 없습니다.")
+                else:
+                    # 다운로드
+                    download_path = stream.download()
+                    st.success('Video downloaded successfully!')
+                    st.video(download_path)
+
+                    # 다운로드된 파일 제공 (optional)
+                    with open(download_path, 'rb') as file:
+                        btn = st.download_button(
+                            label="Download Video File",
+                            data=file,
+                            file_name=os.path.basename(download_path),
+                            mime='video/mp4'
+                        )
+        except Exception as e:
+            if '429' in str(e):
+                st.error("Too many requests. Please wait a moment and try again.")
+            else:
+                st.error(f"An error occurred: {e}")
+    else:
+        st.warning('Please enter a valid YouTube video URL')