Houmeres/SageTube
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 4 Wiki Activity

Add SponsorBlock category selection to video info UI

Browse Source 
Introduces a SponsorBlock section to the video info UI, allowing users to select categories via a dialog. Refactors print statements to use the logger for consistency. Updates imports and improves UI feedback for subtitle and SponsorBlock selections.
This commit is contained in:
Your Name
2025-08-16 15:07:26 +03:00
parent a352371c45
commit 73e662148c
1 changed files with 96 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ytsage_gui_video_info.py → src/gui/ytsage_gui_video_info.py
+96 -10
View File
@@ -16,13 +16,14 @@ from pathlib import Path
from packaging import version
import subprocess
import re
from ..core.ytsage_logging import logger
try:
import yt_dlp
YT_DLP_AVAILABLE = True
except ImportError:
YT_DLP_AVAILABLE = False
print("Warning: yt-dlp not available at startup, will be downloaded at runtime")
from ytsage_gui_dialogs import SubtitleSelectionDialog
logger.warning("yt-dlp not available at startup, will be downloaded at runtime")
from .ytsage_gui_dialogs import SubtitleSelectionDialog, SponsorBlockCategoryDialog
class VideoInfoMixin:
def setup_video_info_section(self):
@@ -124,6 +125,53 @@ class VideoInfoMixin:
video_info_layout.addLayout(subtitle_layout)
# --- End Subtitle Section ---
# Add small spacing between subtitle and sponsorblock sections
video_info_layout.addSpacing(4)
# --- SponsorBlock Section ---
sponsorblock_layout = QHBoxLayout()
self.sponsorblock_select_btn = QPushButton("SponsorBlock Categories...")
self.sponsorblock_select_btn.setFixedHeight(30)
self.sponsorblock_select_btn.clicked.connect(self.open_sponsorblock_dialog)
self.sponsorblock_select_btn.setStyleSheet("""
QPushButton {
background-color: #1d1e22;
border: 2px solid #1d1e22;
border-radius: 4px;
padding: 5px 10px;
}
QPushButton:hover {
background-color: #2a2d36;
}
QPushButton[sponsorBlockSelected="true"] {
border-color: #c90000;
}
QPushButton:disabled {
background-color: #3d3d3d;
color: #888888;
border-color: #3d3d3d;
}
""")
self.sponsorblock_select_btn.setProperty("sponsorBlockSelected", False)
self.sponsorblock_select_btn.setProperty("sponsorBlockSelected", False)
sponsorblock_layout.addWidget(self.sponsorblock_select_btn)
# Label to show selection count
self.selected_sponsorblock_label = QLabel("0 selected")
self.selected_sponsorblock_label.setStyleSheet("color: #cccccc; padding-left: 5px;")
sponsorblock_layout.addWidget(self.selected_sponsorblock_label)
sponsorblock_layout.addStretch()
# Add the sponsorblock layout to the main video info layout
video_info_layout.addLayout(sponsorblock_layout)
# --- End SponsorBlock Section ---
# Initialize SponsorBlock categories as empty initially (will be set to defaults when user opens dialog)
self.selected_sponsorblock_categories = []
self._update_sponsorblock_display()
# Add stretch at the bottom
video_info_layout.addStretch()
@@ -208,7 +256,7 @@ class VideoInfoMixin:
def open_subtitle_dialog(self):
if not hasattr(self, 'available_subtitles') or not hasattr(self, 'available_automatic_subtitles'):
print("Subtitle info not loaded yet.")
logger.warning("Subtitle info not loaded yet.")
return
if not hasattr(self, 'selected_subtitles'):
@@ -227,7 +275,7 @@ class VideoInfoMixin:
if not isinstance(main_window, QMainWindow):
# If the structure is different, this might need adjustment
# Maybe self.parentWidget() or similar depending on how Mixin is used
print("Warning: Cannot find main window to access merge checkbox.")
logger.warning("Cannot find main window to access merge checkbox.")
merge_checkbox = None
else:
merge_checkbox = getattr(main_window, 'merge_subs_checkbox', None)
@@ -235,7 +283,7 @@ class VideoInfoMixin:
if dialog.exec(): # If user clicks OK
self.selected_subtitles = dialog.get_selected_subtitles()
print(f"Selected subtitles: {self.selected_subtitles}")
logger.info(f"Selected subtitles: {self.selected_subtitles}")
# Update UI to reflect selection
count = len(self.selected_subtitles)
self.selected_subs_label.setText(f"{count} selected")
@@ -249,13 +297,51 @@ class VideoInfoMixin:
should_enable = count > 0 and not is_audio_only
merge_checkbox.setEnabled(should_enable)
else:
print("Warning: merge_subs_checkbox not found on parent window.")
logger.warning("merge_subs_checkbox not found on parent window.")
# Re-apply stylesheet to update button border if property changed
self.subtitle_select_btn.style().unpolish(self.subtitle_select_btn)
self.subtitle_select_btn.style().polish(self.subtitle_select_btn)
# No else needed for cancel, state remains unchanged
def open_sponsorblock_dialog(self):
"""Open the SponsorBlock category selection dialog."""
# Initialize selected categories if not exists or empty (first time opening)
if not hasattr(self, 'selected_sponsorblock_categories') or not self.selected_sponsorblock_categories:
# Use None to let the dialog set its own defaults
dialog_categories = None
else:
dialog_categories = self.selected_sponsorblock_categories
dialog = SponsorBlockCategoryDialog(dialog_categories, self)
if dialog.exec():
self.selected_sponsorblock_categories = dialog.get_selected_categories()
logger.info(f"SponsorBlock categories selected: {self.selected_sponsorblock_categories}")
self._update_sponsorblock_display()
def _update_sponsorblock_display(self):
"""Update the SponsorBlock button and label to reflect current selection."""
if not hasattr(self, 'selected_sponsorblock_categories'):
self.selected_sponsorblock_categories = []
count = len(self.selected_sponsorblock_categories)
# Update label text
if count == 0:
self.selected_sponsorblock_label.setText("0 selected")
elif count == 1:
self.selected_sponsorblock_label.setText("1 category selected")
else:
self.selected_sponsorblock_label.setText(f"{count} categories selected")
# Update button property for styling
self.sponsorblock_select_btn.setProperty("sponsorBlockSelected", count > 0)
# Force style refresh
self.sponsorblock_select_btn.style().unpolish(self.sponsorblock_select_btn)
self.sponsorblock_select_btn.style().polish(self.sponsorblock_select_btn)
def download_thumbnail(self, url):
try:
# Store both thumbnail URL and video URL
@@ -274,7 +360,7 @@ class VideoInfoMixin:
pixmap.loadFromData(img_byte_arr.getvalue())
self.thumbnail_label.setPixmap(pixmap)
except Exception as e:
print(f"Error loading thumbnail: {str(e)}")
logger.error(f"Error loading thumbnail: {str(e)}")
def download_thumbnail_file(self, video_url, path):
if not self.save_thumbnail:
@@ -284,7 +370,7 @@ class VideoInfoMixin:
from yt_dlp import YoutubeDL
import requests # Use requests instead of urlopen
print(f"Attempting to save thumbnail for URL: {video_url}")
logger.debug(f"Attempting to save thumbnail for URL: {video_url}")
ydl_opts = {
'quiet': True,
@@ -323,13 +409,13 @@ class VideoInfoMixin:
with open(thumbnail_path, 'wb') as f:
f.write(response.content)
print(f"Thumbnail saved to: {thumbnail_path}")
logger.info(f"Thumbnail saved to: {thumbnail_path}")
self.signals.update_status.emit(f"✅ Thumbnail saved: {filename}")
return True
except Exception as e:
error_msg = f"❌ Thumbnail error: {str(e)}"
print(f"Thumbnail Save Error: {str(e)}")
logger.error(f"Thumbnail Save Error: {str(e)}")
self.signals.update_status.emit(error_msg)
return False