Update app.py

app.py

@@ -142,164 +142,12 @@
 
 
 
-import streamlit as st
-import yt_dlp
-import os
-from pathlib import Path
-import random
-import time
-
-# 페이지 설정
-st.set_page_config(page_title="Simple YouTube Downloader", page_icon="📺")
-st.title("Simple YouTube Downloader 📺")
-
-# 스타일 설정
-st.markdown("""
-    <style>
-        .stButton > button {
-            width: 100%;
-            height: 60px;
-        }
-    </style>
-""", unsafe_allow_html=True)
-
-# 다운로드 폴더 생성
-output_dir = Path("downloads")
-output_dir.mkdir(exist_ok=True)
-
-def download_video(url, progress_bar, status_text):
-    try:
-        user_agents = [
-            'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
-            'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.2.1 Safari/605.1.15',
-            'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:122.0) Gecko/20100101 Firefox/122.0'
-        ]
-        
-        ydl_opts = {
-            'format': 'best',
-            'outtmpl': str(output_dir / '%(title)s.%(ext)s'),
-            'quiet': True,
-            'no_warnings': True,
-            'cookiefile': 'youtube.com_cookies.txt',
-            'user_agent': random.choice(user_agents),
-            'referer': 'https://www.youtube.com/',
-            'http_chunk_size': random.randint(10000000, 15000000),
-            'retries': 5,  # 재시도 횟수를 5로 늘림
-            'sleep_interval': 3,
-            'max_sleep_interval': 8,
-            'headers': {
-                'Accept-Language': 'en-US,en;q=0.9',
-                'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
-            },
-        }
-
-        def progress_hook(d):
-            if d['status'] == 'downloading':
-                try:
-                    progress = d['downloaded_bytes'] / d['total_bytes']
-                    progress_bar.progress(progress)
-                    status_text.text(f"다운로드 중: {progress:.1%}")
-                except:
-                    status_text.text("다운로드 중...")
-            elif d['status'] == 'finished':
-                progress_bar.progress(1.0)
-                status_text.text("처리 완료!")
-
-        ydl_opts['progress_hooks'] = [progress_hook]
-        
-        for attempt in range(5):  # 재시도 횟수를 5로 설정
-            try:
-                with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-                    info = ydl.extract_info(url, download=True)
-                    filename = ydl.prepare_filename(info)
-                    return filename
-            except yt_dlp.utils.ExtractorError as e:
-                if "Sign in to confirm you're not a bot" in str(e):
-                    status_text.text(f"재시도 중... ({attempt + 1}/5)")
-                    time.sleep(random.uniform(3, 5))
-                    continue
-                status_text.text(f"오류 발생: {str(e)}")
-                st.error(f"다운로드 실패: {str(e)}")  # 실패 메시지 표시
-                st.info("문제가 지속될 경우 URL이 올바른지 확인하고 다시 시도하세요.")  # 추가 메모
-                return None
-            except Exception as e:
-                if attempt < 4:
-                    status_text.text(f"재시도 중... ({attempt + 1}/5)")
-                    time.sleep(2)
-                    continue
-                status_text.text(f"오류 발생: {str(e)}")
-                st.error(f"다운로드 실패: {str(e)}")  # 실패 메시지 표시
-                st.info("문제가 지속될 경우 URL이 올바른지 확인하고 다시 시도하세요.")  # 추가 메모
-                return None
-
-    except Exception as e:
-        st.error(f"오류 발생: {str(e)}")
-        return None
-
-# 쿠키 파일 체크
-if not os.path.exists('youtube.com_cookies.txt'):
-    st.warning("⚠️ 'youtube.com_cookies.txt' 파일이 필요합니다. YouTube에 로그인된 쿠키 파일을 업로드해주세요.")
-    uploaded_file = st.file_uploader("쿠키 파일 업로드", type=['txt'])
-    if uploaded_file is not None:
-        with open('youtube.com_cookies.txt', 'wb') as f:
-            f.write(uploaded_file.getvalue())
-        st.success("✅ 쿠키 파일이 업로드되었습니다!")
-        st.rerun()
-
-# 메인 UI
-if 'video_url' not in st.session_state:
-    st.session_state.video_url = ""
-
-video_url = st.text_input("YouTube URL 입력:", 
-                         value=st.session_state.video_url,
-                         placeholder="https://www.youtube.com/watch?v=...")
-
-# 버튼을 같은 줄에 배치
-col1, col2 = st.columns(2)
-
-with col1:
-    download_button = st.button("다운로드", type="primary", key="download_btn", use_container_width=True)
-
-with col2:
-    reset_button = st.button("초기화", key="reset_btn", use_container_width=True)
-
-# 다운로드 로직
-if download_button:
-    if video_url:
-        progress_bar = st.progress(0)
-        status_text = st.empty()
-        
-        downloaded_file = download_video(video_url, progress_bar, status_text)
-        
-        if downloaded_file and os.path.exists(downloaded_file):
-            with open(downloaded_file, 'rb') as file:
-                st.download_button(
-                    label="⬇️ 파일 다운로드",
-                    data=file,
-                    file_name=os.path.basename(downloaded_file),
-                    mime="video/mp4",
-                    use_container_width=True
-                )
-    else:
-        st.warning("⚠️ YouTube URL을 입력해주세요.")
-
-# 초기화 로직
-if reset_button:
-    st.session_state.video_url = ""
-    st.rerun()
-
-
-
 # import streamlit as st
 # import yt_dlp
 # import os
 # from pathlib import Path
 # import random
 # import time
-# import logging
-
-# # 로깅 설정
-# logging.basicConfig(level=logging.ERROR, filename='error.log')
 
 # # 페이지 설정
 # st.set_page_config(page_title="Simple YouTube Downloader", page_icon="📺")
@@ -335,10 +183,10 @@ if reset_button:
 #             'cookiefile': 'youtube.com_cookies.txt',
 #             'user_agent': random.choice(user_agents),
 #             'referer': 'https://www.youtube.com/',
-#             'http_chunk_size': random.randint(100000000, 150000000),
-#             'retries': 5,  # 재시도 횟수
+#             'http_chunk_size': random.randint(10000000, 15000000),
+#             'retries': 5,  # 재시도 횟수를 5로 늘림
 #             'sleep_interval': 3,
-#             'max_sleep_interval': 10,
+#             'max_sleep_interval': 8,
 #             'headers': {
 #                 'Accept-Language': 'en-US,en;q=0.9',
 #                 'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
@@ -351,8 +199,7 @@ if reset_button:
 #                     progress = d['downloaded_bytes'] / d['total_bytes']
 #                     progress_bar.progress(progress)
 #                     status_text.text(f"다운로드 중: {progress:.1%}")
-#                 except Exception as e:
-#                     logging.error(f"Progress Hook Error: {e}")
+#                 except:
 #                     status_text.text("다운로드 중...")
 #             elif d['status'] == 'finished':
 #                 progress_bar.progress(1.0)
@@ -371,21 +218,22 @@ if reset_button:
 #                     status_text.text(f"재시도 중... ({attempt + 1}/5)")
 #                     time.sleep(random.uniform(3, 5))
 #                     continue
-#                 logging.error(f"Extractor Error: {str(e)}")
-#                 st.info("문제가 지속될 경우 URL이 올바른지 확인하고 다시 시도하세요.")
+#                 status_text.text(f"오류 발생: {str(e)}")
+#                 st.error(f"다운로드 실패: {str(e)}")  # 실패 메시지 표시
+#                 st.info("문제가 지속될 경우 URL이 올바른지 확인하고 다시 시도하세요.")  # 추가 메모
 #                 return None
 #             except Exception as e:
-#                 logging.error(f"Download Error: {str(e)}")
 #                 if attempt < 4:
 #                     status_text.text(f"재시도 중... ({attempt + 1}/5)")
 #                     time.sleep(2)
 #                     continue
-#                 st.info("문제가 지속될 경우 다른 URL을 시도해주세요.")
+#                 status_text.text(f"오류 발생: {str(e)}")
+#                 st.error(f"다운로드 실패: {str(e)}")  # 실패 메시지 표시
+#                 st.info("문제가 지속될 경우 URL이 올바른지 확인하고 다시 시도하세요.")  # 추가 메모
 #                 return None
 
 #     except Exception as e:
-#         logging.error(f"General Error: {str(e)}")
-#         st.info("문제가 지속될 경우 다른 URL을 시도해주세요.")
+#         st.error(f"오류 발생: {str(e)}")
 #         return None
 
 # # 쿠키 파일 체크
@@ -438,5 +286,157 @@ if reset_button:
 # # 초기화 로직
 # if reset_button:
 #     st.session_state.video_url = ""
-#     st.rerun()  # st.experimental_rerun()을 st.rerun()으로 변경했습니다.
+#     st.rerun()
+
+
+
+import streamlit as st
+import yt_dlp
+import os
+from pathlib import Path
+import random
+import time
+import logging
+
+# 로깅 설정
+logging.basicConfig(level=logging.ERROR, filename='error.log')
+
+# 페이지 설정
+st.set_page_config(page_title="Simple YouTube Downloader", page_icon="📺")
+st.title("Simple YouTube Downloader 📺")
+
+# 스타일 설정
+st.markdown("""
+    <style>
+        .stButton > button {
+            width: 100%;
+            height: 60px;
+        }
+    </style>
+""", unsafe_allow_html=True)
+
+# 다운로드 폴더 생성
+output_dir = Path("downloads")
+output_dir.mkdir(exist_ok=True)
+
+def download_video(url, progress_bar, status_text):
+    try:
+        user_agents = [
+            'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
+            'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.2.1 Safari/605.1.15',
+            'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:122.0) Gecko/20100101 Firefox/122.0'
+        ]
+        
+        ydl_opts = {
+            'format': 'best',
+            'outtmpl': str(output_dir / '%(title)s.%(ext)s'),
+            'quiet': True,
+            'no_warnings': True,
+            'cookiefile': 'youtube.com_cookies.txt',
+            'user_agent': random.choice(user_agents),
+            'referer': 'https://www.youtube.com/',
+            'http_chunk_size': random.randint(100000000, 150000000),
+            'retries': 5,  # 재시도 횟수
+            'sleep_interval': 3,
+            'max_sleep_interval': 10,
+            'headers': {
+                'Accept-Language': 'en-US,en;q=0.9',
+                'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
+            },
+        }
+
+        def progress_hook(d):
+            if d['status'] == 'downloading':
+                try:
+                    progress = d['downloaded_bytes'] / d['total_bytes']
+                    progress_bar.progress(progress)
+                    status_text.text(f"다운로드 중: {progress:.1%}")
+                except Exception as e:
+                    logging.error(f"Progress Hook Error: {e}")
+                    status_text.text("다운로드 중...")
+            elif d['status'] == 'finished':
+                progress_bar.progress(1.0)
+                status_text.text("처리 완료!")
+
+        ydl_opts['progress_hooks'] = [progress_hook]
+        
+        for attempt in range(5):  # 재시도 횟수를 5로 설정
+            try:
+                with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                    info = ydl.extract_info(url, download=True)
+                    filename = ydl.prepare_filename(info)
+                    return filename
+            except yt_dlp.utils.ExtractorError as e:
+                if "Sign in to confirm you're not a bot" in str(e):
+                    status_text.text(f"재시도 중... ({attempt + 1}/5)")
+                    time.sleep(random.uniform(3, 5))
+                    continue
+                logging.error(f"Extractor Error: {str(e)}")
+                st.info("문제가 지속될 경우 URL이 올바른지 확인하고 다시 시도하세요.")
+                return None
+            except Exception as e:
+                logging.error(f"Download Error: {str(e)}")
+                if attempt < 4:
+                    status_text.text(f"재시도 중... ({attempt + 1}/5)")
+                    time.sleep(2)
+                    continue
+                st.info("문제가 지속될 경우 다른 URL을 시도해주세요.")
+                return None
+
+    except Exception as e:
+        logging.error(f"General Error: {str(e)}")
+        st.info("문제가 지속될 경우 다른 URL을 시도해주세요.")
+        return None
+
+# 쿠키 파일 체크
+if not os.path.exists('youtube.com_cookies.txt'):
+    st.warning("⚠️ 'youtube.com_cookies.txt' 파일이 필요합니다. YouTube에 로그인된 쿠키 파일을 업로드해주세요.")
+    uploaded_file = st.file_uploader("쿠키 파일 업로드", type=['txt'])
+    if uploaded_file is not None:
+        with open('youtube.com_cookies.txt', 'wb') as f:
+            f.write(uploaded_file.getvalue())
+        st.success("✅ 쿠키 파일이 업로드되었습니다!")
+        st.rerun()
+
+# 메인 UI
+if 'video_url' not in st.session_state:
+    st.session_state.video_url = ""
+
+video_url = st.text_input("YouTube URL 입력:", 
+                         value=st.session_state.video_url,
+                         placeholder="https://www.youtube.com/watch?v=...")
+
+# 버튼을 같은 줄에 배치
+col1, col2 = st.columns(2)
+
+with col1:
+    download_button = st.button("다운로드", type="primary", key="download_btn", use_container_width=True)
+
+with col2:
+    reset_button = st.button("초기화", key="reset_btn", use_container_width=True)
+
+# 다운로드 로직
+if download_button:
+    if video_url:
+        progress_bar = st.progress(0)
+        status_text = st.empty()
+        
+        downloaded_file = download_video(video_url, progress_bar, status_text)
+        
+        if downloaded_file and os.path.exists(downloaded_file):
+            with open(downloaded_file, 'rb') as file:
+                st.download_button(
+                    label="⬇️ 파일 다운로드",
+                    data=file,
+                    file_name=os.path.basename(downloaded_file),
+                    mime="video/mp4",
+                    use_container_width=True
+                )
+    else:
+        st.warning("⚠️ YouTube URL을 입력해주세요.")
+
+# 초기화 로직
+if reset_button:
+    st.session_state.video_url = ""
+    st.rerun()  # st.experimental_rerun()을 st.rerun()으로 변경했습니다.