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 @@ def download_video(url, progress_bar, status_text):
             '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 @@ def download_video(url, progress_bar, status_text):
                     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 @@ def download_video(url, progress_bar, status_text):
                     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 download_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()으로 변경했습니다.