Spaces:
Running
Running
Commit
·
6ae0a89
1
Parent(s):
b6d644c
youtube
Browse files- app.py +3 -2
- link_processor.py +20 -25
app.py
CHANGED
|
@@ -31,7 +31,7 @@ def download_file_button(file_path, label, file_name, mime_type):
|
|
| 31 |
):
|
| 32 |
# Reset the processing state after download
|
| 33 |
st.session_state.processing_complete = False
|
| 34 |
-
st.
|
| 35 |
|
| 36 |
|
| 37 |
def reset_processing_state():
|
|
@@ -271,7 +271,8 @@ with tab3:
|
|
| 271 |
if st.session_state.processing_complete:
|
| 272 |
if st.button("Process Another File"):
|
| 273 |
reset_processing_state()
|
| 274 |
-
st.
|
|
|
|
| 275 |
|
| 276 |
# Sidebar
|
| 277 |
with st.sidebar:
|
|
|
|
| 31 |
):
|
| 32 |
# Reset the processing state after download
|
| 33 |
st.session_state.processing_complete = False
|
| 34 |
+
st.experimental_rerun()
|
| 35 |
|
| 36 |
|
| 37 |
def reset_processing_state():
|
|
|
|
| 271 |
if st.session_state.processing_complete:
|
| 272 |
if st.button("Process Another File"):
|
| 273 |
reset_processing_state()
|
| 274 |
+
st.experimental_rerun()
|
| 275 |
+
|
| 276 |
|
| 277 |
# Sidebar
|
| 278 |
with st.sidebar:
|
link_processor.py
CHANGED
|
@@ -24,7 +24,7 @@ class LinkDownloader:
|
|
| 24 |
temp_filename = 'downloaded_video.%(ext)s'
|
| 25 |
temp_filepath = str(self.output_dir / temp_filename)
|
| 26 |
|
| 27 |
-
# Configure yt-dlp options
|
| 28 |
ydl_opts = {
|
| 29 |
'format': 'best',
|
| 30 |
'outtmpl': temp_filepath,
|
|
@@ -33,31 +33,29 @@ class LinkDownloader:
|
|
| 33 |
'extract_audio': False,
|
| 34 |
'ignoreerrors': True, # Continue on download errors
|
| 35 |
'no_check_certificate': True, # Skip HTTPS certificate validation
|
| 36 |
-
'username': None, # Will be filled by user if needed
|
| 37 |
-
'password': None, # Will be filled by user if needed
|
| 38 |
}
|
| 39 |
|
| 40 |
-
#
|
| 41 |
-
|
| 42 |
-
|
| 43 |
info = ydl.extract_info(url, download=False)
|
| 44 |
-
if info:
|
| 45 |
-
|
| 46 |
-
except yt_dlp.utils.DownloadError as e:
|
| 47 |
-
if "Sign in" in str(e):
|
| 48 |
-
# If authentication is required, prompt for credentials
|
| 49 |
-
st.warning("Authentication required. Please enter your YouTube credentials.")
|
| 50 |
-
username = st.text_input("YouTube Email", key="yt_email")
|
| 51 |
-
password = st.text_input("YouTube Password", type="password", key="yt_pass")
|
| 52 |
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
ydl_opts['password'] = password
|
| 56 |
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 61 |
|
| 62 |
# Get the actual extension and filepath
|
| 63 |
ext = info.get('ext', 'mp4')
|
|
@@ -65,13 +63,10 @@ class LinkDownloader:
|
|
| 65 |
|
| 66 |
# Verify file exists
|
| 67 |
if not os.path.exists(actual_filepath):
|
| 68 |
-
print(f"Files in directory: {os.listdir(self.output_dir)}")
|
| 69 |
raise FileNotFoundError(f"Downloaded file not found at {actual_filepath}")
|
| 70 |
|
| 71 |
-
print(f"File size: {os.path.getsize(actual_filepath)} bytes")
|
| 72 |
return actual_filepath
|
| 73 |
|
| 74 |
except Exception as e:
|
| 75 |
st.error(f"Error during download: {str(e)}")
|
| 76 |
-
|
| 77 |
-
return None
|
|
|
|
| 24 |
temp_filename = 'downloaded_video.%(ext)s'
|
| 25 |
temp_filepath = str(self.output_dir / temp_filename)
|
| 26 |
|
| 27 |
+
# Configure yt-dlp options
|
| 28 |
ydl_opts = {
|
| 29 |
'format': 'best',
|
| 30 |
'outtmpl': temp_filepath,
|
|
|
|
| 33 |
'extract_audio': False,
|
| 34 |
'ignoreerrors': True, # Continue on download errors
|
| 35 |
'no_check_certificate': True, # Skip HTTPS certificate validation
|
|
|
|
|
|
|
| 36 |
}
|
| 37 |
|
| 38 |
+
# Try to extract video information
|
| 39 |
+
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
| 40 |
+
try:
|
| 41 |
info = ydl.extract_info(url, download=False)
|
| 42 |
+
if not info:
|
| 43 |
+
raise ValueError("Failed to extract video information.")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
|
| 45 |
+
# Download video
|
| 46 |
+
ydl.download([url])
|
|
|
|
| 47 |
|
| 48 |
+
except yt_dlp.utils.DownloadError as e:
|
| 49 |
+
if "Sign in" in str(e):
|
| 50 |
+
st.warning("Authentication required. Please pass cookies from your browser.")
|
| 51 |
+
st.info(
|
| 52 |
+
"Follow the instructions at "
|
| 53 |
+
"[yt-dlp FAQ](https://github.com/yt-dlp/yt-dlp/wiki/FAQ#how-do-i-pass-cookies-to-yt-dlp) "
|
| 54 |
+
"to pass cookies for authentication."
|
| 55 |
+
)
|
| 56 |
+
return None
|
| 57 |
+
else:
|
| 58 |
+
raise e
|
| 59 |
|
| 60 |
# Get the actual extension and filepath
|
| 61 |
ext = info.get('ext', 'mp4')
|
|
|
|
| 63 |
|
| 64 |
# Verify file exists
|
| 65 |
if not os.path.exists(actual_filepath):
|
|
|
|
| 66 |
raise FileNotFoundError(f"Downloaded file not found at {actual_filepath}")
|
| 67 |
|
|
|
|
| 68 |
return actual_filepath
|
| 69 |
|
| 70 |
except Exception as e:
|
| 71 |
st.error(f"Error during download: {str(e)}")
|
| 72 |
+
return None
|
|
|