Update app.py

app.py

@@ -1,61 +1,38 @@
-
 import streamlit as st
 import yt_dlp
 import os
 from pathlib import Path
 
-# Set page config
-st.set_page_config(page_title="YouTube Video Downloader", page_icon="📺")
-
-# Set the title of the app
-st.title("YouTube Video Downloader 📺")
+# 페이지 설정
+st.set_page_config(page_title="Simple YouTube Downloader", page_icon="📺")
+st.title("Simple YouTube Downloader 📺")
 
-# Create output directory if it doesn't exist
+# 다운로드 폴더 생성
 output_dir = Path("downloads")
 output_dir.mkdir(exist_ok=True)
 
-# Input field for YouTube video URL
-video_url = st.text_input("Enter YouTube Video URL:", placeholder="https://www.youtube.com/watch?v=...")
-
-# Function to download video
-def download_video(url):
+def reset_form():
+    st.session_state.url = ""
+    
+def download_video(url, progress_bar, status_text):
     try:
-        
         ydl_opts = {
-            'format': 'bestvideo+bestaudio/best',
+            'format': 'best',  # 최고 품질의 통합 포맷
             'outtmpl': str(output_dir / '%(title)s.%(ext)s'),
-            'merge_output_format': 'webm',
-            'quiet': False,  # Change to False to see more detailed error output
-            'no_warnings': False,
-            'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
-            'referer': 'https://www.youtube.com/',
-            'cookiefile': 'youtube.com_cookies.txt',  # Ensure this file path is correct
-            'socket_timeout': 30,
-            'http_chunk_size': 10485760,  # 10MB
-            'retries': 10,  # Increase retry attempts
-            'headers': {
-                'Accept-Language': 'en-US,en;q=0.9',
-                'Accept-Encoding': 'gzip, deflate, br',
-                'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8',
-                'Upgrade-Insecure-Requests': '1',
-            }
+            'quiet': True
         }
-
-        # Progress placeholder
-        progress_bar = st.progress(0)
-        status_text = st.empty()
         
         def progress_hook(d):
             if d['status'] == 'downloading':
                 try:
                     progress = d['downloaded_bytes'] / d['total_bytes']
                     progress_bar.progress(progress)
-                    status_text.text(f"Downloading: {progress:.1%}")
+                    status_text.text(f"다운로드 중: {progress:.1%}")
                 except:
-                    status_text.text("Downloading... (size unknown)")
+                    status_text.text("다운로드 중...")
             elif d['status'] == 'finished':
                 progress_bar.progress(1.0)
-                status_text.text("Processing...")
+                status_text.text("처리 완료!")
 
         ydl_opts['progress_hooks'] = [progress_hook]
         
@@ -65,46 +42,145 @@ def download_video(url):
             return filename
 
     except Exception as e:
-        st.error(f"An error occurred: {str(e)}")
+        st.error(f"오류 발생: {str(e)}")
         return None
 
-# Download button
-if st.button("Download"):
-    if video_url:
-        try:
-            with st.spinner("Preparing download..."):
-                downloaded_file_path = download_video(video_url)
-                
-            if downloaded_file_path and os.path.exists(downloaded_file_path):
-                with open(downloaded_file_path, 'rb') as file:
+# 메인 UI
+video_url = st.text_input("YouTube URL 입력:", key="url", 
+                         placeholder="https://www.youtube.com/watch?v=...")
+
+col1, col2 = st.columns(2)
+
+with col1:
+    if st.button("다운로드", type="primary"):
+        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="Click here to download",
+                        label="⬇️ 파일 다운로드",
                         data=file,
-                        file_name=os.path.basename(downloaded_file_path),
-                        mime="application/octet-stream"
+                        file_name=os.path.basename(downloaded_file),
+                        mime="video/mp4"
                     )
-                st.success("✅ Download ready!")
-            else:
-                st.error("❌ Download failed. Please try again.")
-        except Exception as e:
-            st.error(f"❌ An error occurred: {str(e)}")
-    else:
-        st.warning("⚠️ Please enter a valid YouTube URL.")
-
-# Help section
-with st.expander("ℹ️ Help & Troubleshooting"):
-    st.markdown("""
-    **If you're experiencing download issues:**
+        else:
+            st.warning("⚠️ YouTube URL을 입력해주세요.")
+
+with col2:
+    if st.button("초기화"):
+        reset_form()
+
+
+# import streamlit as st
+# import yt_dlp
+# import os
+# from pathlib import Path
+
+# # Set page config
+# st.set_page_config(page_title="YouTube Video Downloader", page_icon="📺")
+
+# # Set the title of the app
+# st.title("YouTube Video Downloader 📺")
+
+# # Create output directory if it doesn't exist
+# output_dir = Path("downloads")
+# output_dir.mkdir(exist_ok=True)
+
+# # Input field for YouTube video URL
+# video_url = st.text_input("Enter YouTube Video URL:", placeholder="https://www.youtube.com/watch?v=...")
+
+# # Function to download video
+# def download_video(url):
+#     try:
+        
+#         ydl_opts = {
+#             'format': 'bestvideo+bestaudio/best',
+#             'outtmpl': str(output_dir / '%(title)s.%(ext)s'),
+#             'merge_output_format': 'webm',
+#             'quiet': False,  # Change to False to see more detailed error output
+#             'no_warnings': False,
+#             'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
+#             'referer': 'https://www.youtube.com/',
+#             'cookiefile': 'youtube.com_cookies.txt',  # Ensure this file path is correct
+#             'socket_timeout': 30,
+#             'http_chunk_size': 10485760,  # 10MB
+#             'retries': 10,  # Increase retry attempts
+#             'headers': {
+#                 'Accept-Language': 'en-US,en;q=0.9',
+#                 'Accept-Encoding': 'gzip, deflate, br',
+#                 'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8',
+#                 'Upgrade-Insecure-Requests': '1',
+#             }
+#         }
+
+#         # Progress placeholder
+#         progress_bar = st.progress(0)
+#         status_text = st.empty()
+        
+#         def progress_hook(d):
+#             if d['status'] == 'downloading':
+#                 try:
+#                     progress = d['downloaded_bytes'] / d['total_bytes']
+#                     progress_bar.progress(progress)
+#                     status_text.text(f"Downloading: {progress:.1%}")
+#                 except:
+#                     status_text.text("Downloading... (size unknown)")
+#             elif d['status'] == 'finished':
+#                 progress_bar.progress(1.0)
+#                 status_text.text("Processing...")
+
+#         ydl_opts['progress_hooks'] = [progress_hook]
+        
+#         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+#             info = ydl.extract_info(url, download=True)
+#             filename = ydl.prepare_filename(info)
+#             return filename
+
+#     except Exception as e:
+#         st.error(f"An error occurred: {str(e)}")
+#         return None
+
+# # Download button
+# if st.button("Download"):
+#     if video_url:
+#         try:
+#             with st.spinner("Preparing download..."):
+#                 downloaded_file_path = download_video(video_url)
+                
+#             if downloaded_file_path and os.path.exists(downloaded_file_path):
+#                 with open(downloaded_file_path, 'rb') as file:
+#                     st.download_button(
+#                         label="Click here to download",
+#                         data=file,
+#                         file_name=os.path.basename(downloaded_file_path),
+#                         mime="application/octet-stream"
+#                     )
+#                 st.success("✅ Download ready!")
+#             else:
+#                 st.error("❌ Download failed. Please try again.")
+#         except Exception as e:
+#             st.error(f"❌ An error occurred: {str(e)}")
+#     else:
+#         st.warning("⚠️ Please enter a valid YouTube URL.")
+
+# # Help section
+# with st.expander("ℹ️ Help & Troubleshooting"):
+#     st.markdown("""
+#     **If you're experiencing download issues:**
     
-    1. Cookie Authentication Method:
-       - Install a browser extension like "Get cookies.txt"
-       - Visit YouTube and ensure you're logged in
-       - Export cookies to 'youtube.com_cookies.txt'
-       - Place the file in the same directory as this app
+#     1. Cookie Authentication Method:
+#        - Install a browser extension like "Get cookies.txt"
+#        - Visit YouTube and ensure you're logged in
+#        - Export cookies to 'youtube.com_cookies.txt'
+#        - Place the file in the same directory as this app
     
-    2. Alternative Solutions:
-       - Try different videos
-       - Check if the video is public/not age-restricted
-       - Try again later if YouTube is blocking requests
-    """)
+#     2. Alternative Solutions:
+#        - Try different videos
+#        - Check if the video is public/not age-restricted
+#        - Try again later if YouTube is blocking requests
+#     """)