mirror of
https://github.com/denizsafak/abogen.git
synced 2026-07-22 07:10:28 +02:00
Imrpovements in queue manager gui and code
This commit is contained in:
Deniz Şafak
+285
-23
@@ -5,35 +5,83 @@
|
||||
from PyQt5.QtWidgets import (
|
||||
QDialog,
|
||||
QVBoxLayout,
|
||||
QHBoxLayout, # added
|
||||
QDialogButtonBox,
|
||||
QPushButton,
|
||||
QListWidget,
|
||||
QListWidgetItem,
|
||||
QFileIconProvider,
|
||||
QLabel,
|
||||
)
|
||||
|
||||
from PyQt5.QtCore import QFileInfo, Qt
|
||||
from constants import COLORS
|
||||
|
||||
class QueueManager(QDialog):
|
||||
def __init__(self, parent, queue: list, title="Queue Manager", size=(400, 300)):
|
||||
def __init__(self, parent, queue: list, title="Queue Manager", size=(600, 700)):
|
||||
super().__init__()
|
||||
self.queue = queue
|
||||
self.parent = parent
|
||||
layout = QVBoxLayout()
|
||||
layout.setContentsMargins(15, 15, 15, 15) # set main layout margins
|
||||
layout.setSpacing(12) # set spacing between widgets in main layout
|
||||
# list of queued items
|
||||
self.listwidget = QListWidget()
|
||||
self.listwidget.setSelectionMode(QListWidget.SingleSelection)
|
||||
self.listwidget.setSelectionMode(QListWidget.ExtendedSelection)
|
||||
self.listwidget.setAlternatingRowColors(True)
|
||||
self.listwidget.setContextMenuPolicy(Qt.CustomContextMenu)
|
||||
self.listwidget.customContextMenuRequested.connect(self.show_context_menu)
|
||||
# Add informative instructions at the top
|
||||
instructions = QLabel(
|
||||
"<h2>How Queue Works?</h2>"
|
||||
"You can add text files (.txt) directly using the '<b>Add files</b>' button below. "
|
||||
"To add PDF or EPUB files, use the input box in the main window and click the <b>'Add to Queue'</b> button. "
|
||||
"Each file in the queue keeps the configuration settings active when it was added. "
|
||||
"Changing the main window configuration afterward <b>does not</b> affect files already in the queue. "
|
||||
"You can view each file's configuration by hovering over them."
|
||||
)
|
||||
instructions.setAlignment(Qt.AlignLeft)
|
||||
instructions.setWordWrap(True)
|
||||
instructions.setStyleSheet("margin-bottom: 8px;")
|
||||
layout.addWidget(instructions)
|
||||
# Overlay label for empty queue
|
||||
self.empty_overlay = QLabel(
|
||||
"No items in the queue.",
|
||||
self.listwidget
|
||||
)
|
||||
self.empty_overlay.setAlignment(Qt.AlignCenter)
|
||||
self.empty_overlay.setStyleSheet(f"color: {COLORS['LIGHT_DISABLED']}; background: transparent;")
|
||||
self.empty_overlay.setWordWrap(True)
|
||||
self.empty_overlay.hide()
|
||||
# add queue items to the list
|
||||
self.process_queue()
|
||||
|
||||
layout.addWidget(self.listwidget)
|
||||
# add remove item button
|
||||
remove_button = QPushButton("Remove")
|
||||
remove_button.clicked.connect(self.remove_item)
|
||||
layout.addWidget(remove_button)
|
||||
# add clear queue button
|
||||
clear_button = QPushButton("Clear Queue")
|
||||
clear_button.clicked.connect(self.clear_queue)
|
||||
layout.addWidget(clear_button)
|
||||
button_row = QHBoxLayout()
|
||||
button_row.setContentsMargins(0, 0, 0, 0) # optional: no margins for button row
|
||||
button_row.setSpacing(7) # set spacing between buttons
|
||||
# Add files button
|
||||
add_files_button = QPushButton("Add files")
|
||||
add_files_button.setFixedHeight(40)
|
||||
add_files_button.clicked.connect(self.add_more_files)
|
||||
button_row.addWidget(add_files_button)
|
||||
|
||||
# Remove button
|
||||
self.remove_button = QPushButton("Remove selected")
|
||||
self.remove_button.setFixedHeight(40)
|
||||
self.remove_button.clicked.connect(self.remove_item)
|
||||
button_row.addWidget(self.remove_button)
|
||||
|
||||
# Clear button
|
||||
self.clear_button = QPushButton("Clear Queue")
|
||||
self.clear_button.setFixedHeight(40)
|
||||
self.clear_button.clicked.connect(self.clear_queue)
|
||||
button_row.addWidget(self.clear_button)
|
||||
|
||||
layout.addLayout(button_row)
|
||||
layout.addWidget(self.listwidget)
|
||||
|
||||
# Connect selection change to update button state
|
||||
self.listwidget.currentItemChanged.connect(self.update_button_states)
|
||||
self.listwidget.itemSelectionChanged.connect(self.update_button_states)
|
||||
|
||||
buttons = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel, self)
|
||||
buttons.accepted.connect(self.accept)
|
||||
@@ -44,30 +92,244 @@ class QueueManager(QDialog):
|
||||
self.setLayout(layout)
|
||||
|
||||
self.setWindowTitle(title)
|
||||
self.setMinimumSize(*size)
|
||||
self.resize(*size)
|
||||
|
||||
self.update_button_states()
|
||||
|
||||
def process_queue(self):
|
||||
"""Process the queue items."""
|
||||
self.listwidget.clear()
|
||||
if not self.queue:
|
||||
self.empty_overlay.resize(self.listwidget.size())
|
||||
self.empty_overlay.show()
|
||||
self.update_button_states()
|
||||
return
|
||||
else:
|
||||
self.empty_overlay.hide()
|
||||
icon_provider = QFileIconProvider()
|
||||
for item in self.queue:
|
||||
# TODO take just the file name from the item, no directories
|
||||
self.listwidget.addItem(item.file_name)
|
||||
# Only show the file name, not the full path
|
||||
file_name = item.file_name
|
||||
display_name = file_name
|
||||
import os
|
||||
if os.path.sep in file_name:
|
||||
display_name = os.path.basename(file_name)
|
||||
# Get icon for the file
|
||||
icon = icon_provider.icon(QFileInfo(file_name))
|
||||
list_item = QListWidgetItem(icon, display_name)
|
||||
# Set tooltip with detailed info
|
||||
output_folder = getattr(item, 'output_folder', '')
|
||||
tooltip = (
|
||||
f"<b>Path:</b> {file_name}<br>"
|
||||
f"<b>Language:</b> {getattr(item, 'lang_code', '')}<br>"
|
||||
f"<b>Speed:</b> {getattr(item, 'speed', '')}<br>"
|
||||
f"<b>Voice:</b> {getattr(item, 'voice', '')}<br>"
|
||||
f"<b>Save Option:</b> {getattr(item, 'save_option', '')}<br>"
|
||||
)
|
||||
if output_folder not in (None, '', 'None'):
|
||||
tooltip += f"<b>Output Folder:</b> {output_folder}<br>"
|
||||
tooltip += (
|
||||
f"<b>Subtitle Mode:</b> {getattr(item, 'subtitle_mode', '')}<br>"
|
||||
f"<b>Output Format:</b> {getattr(item, 'output_format', '')}<br>"
|
||||
f"<b>Characters:</b> {getattr(item, 'total_char_count', '')}"
|
||||
)
|
||||
list_item.setToolTip(tooltip)
|
||||
self.listwidget.addItem(list_item)
|
||||
|
||||
self.update_button_states()
|
||||
|
||||
def remove_item(self):
|
||||
item = self.listwidget.currentItem()
|
||||
if not item:
|
||||
items = self.listwidget.selectedItems()
|
||||
if not items:
|
||||
return
|
||||
found_item = [q for q in self.queue if q.file_name==item.text()][0]
|
||||
self.queue.remove(found_item)
|
||||
# rebind the queue to the listwidget
|
||||
self.listwidget.clear()
|
||||
import os
|
||||
display_names = [item.text() for item in items]
|
||||
to_remove = []
|
||||
for q in self.queue:
|
||||
if os.path.basename(q.file_name) in display_names:
|
||||
to_remove.append(q)
|
||||
for item in to_remove:
|
||||
self.queue.remove(item)
|
||||
self.process_queue()
|
||||
pass
|
||||
self.update_button_states()
|
||||
|
||||
def clear_queue(self):
|
||||
from PyQt5.QtWidgets import QMessageBox
|
||||
if len(self.queue) > 1:
|
||||
reply = QMessageBox.question(
|
||||
self,
|
||||
"Confirm Clear Queue",
|
||||
f"Are you sure you want to clear {len(self.queue)} items from the queue?",
|
||||
QMessageBox.Yes | QMessageBox.No,
|
||||
QMessageBox.No
|
||||
)
|
||||
if reply != QMessageBox.Yes:
|
||||
return
|
||||
self.queue.clear()
|
||||
self.listwidget.clear()
|
||||
self.empty_overlay.resize(self.listwidget.size()) # Ensure overlay is sized correctly
|
||||
self.empty_overlay.show() # Show the overlay when queue is empty
|
||||
self.update_button_states()
|
||||
|
||||
def get_queue(self):
|
||||
return self.queue
|
||||
return self.queue
|
||||
|
||||
def get_current_attributes(self):
|
||||
# Fetch current attribute values from the parent abogen GUI
|
||||
attrs = {}
|
||||
parent = self.parent
|
||||
if parent is not None:
|
||||
# lang_code: use parent's get_voice_formula and get_selected_lang
|
||||
if hasattr(parent, 'get_voice_formula') and hasattr(parent, 'get_selected_lang'):
|
||||
voice_formula = parent.get_voice_formula()
|
||||
attrs['lang_code'] = parent.get_selected_lang(voice_formula)
|
||||
attrs['voice'] = voice_formula
|
||||
else:
|
||||
attrs['lang_code'] = getattr(parent, 'selected_lang', '')
|
||||
attrs['voice'] = getattr(parent, 'selected_voice', '')
|
||||
# speed
|
||||
if hasattr(parent, 'speed_slider'):
|
||||
attrs['speed'] = parent.speed_slider.value() / 100.0
|
||||
else:
|
||||
attrs['speed'] = getattr(parent, 'speed', 1.0)
|
||||
# save_option
|
||||
attrs['save_option'] = getattr(parent, 'save_option', '')
|
||||
# output_folder
|
||||
attrs['output_folder'] = getattr(parent, 'selected_output_folder', '')
|
||||
# subtitle_mode
|
||||
if hasattr(parent, 'get_actual_subtitle_mode'):
|
||||
attrs['subtitle_mode'] = parent.get_actual_subtitle_mode()
|
||||
else:
|
||||
attrs['subtitle_mode'] = getattr(parent, 'subtitle_mode', '')
|
||||
# output_format
|
||||
attrs['output_format'] = getattr(parent, 'selected_format', '')
|
||||
# total_char_count
|
||||
attrs['total_char_count'] = getattr(parent, 'char_count', '')
|
||||
else:
|
||||
# fallback: empty values
|
||||
attrs = {k: '' for k in [
|
||||
'lang_code', 'speed', 'voice', 'save_option',
|
||||
'output_folder', 'subtitle_mode', 'output_format', 'total_char_count']}
|
||||
return attrs
|
||||
|
||||
def add_more_files(self):
|
||||
from PyQt5.QtWidgets import QFileDialog, QMessageBox
|
||||
import os
|
||||
from utils import calculate_text_length # import the function
|
||||
# Only allow .txt files
|
||||
files, _ = QFileDialog.getOpenFileNames(self, "Select .txt files", "", "Text Files (*.txt)")
|
||||
if not files:
|
||||
return
|
||||
# Get current attribute values from GUI
|
||||
current_attrs = self.get_current_attributes()
|
||||
duplicates = []
|
||||
for file_path in files:
|
||||
# Create a dummy item with the current GUI attributes
|
||||
class QueueItem:
|
||||
pass
|
||||
item = QueueItem()
|
||||
item.file_name = file_path
|
||||
for attr, value in current_attrs.items():
|
||||
setattr(item, attr, value)
|
||||
# Read file content and calculate total_char_count using calculate_text_length
|
||||
try:
|
||||
with open(file_path, 'r', encoding='utf-8', errors='ignore') as f:
|
||||
file_content = f.read()
|
||||
item.total_char_count = calculate_text_length(file_content)
|
||||
except Exception:
|
||||
item.total_char_count = 0
|
||||
# Prevent adding duplicate items to the queue (check all attributes)
|
||||
is_duplicate = False
|
||||
for queued_item in self.queue:
|
||||
if (
|
||||
getattr(queued_item, 'file_name', None) == getattr(item, 'file_name', None) and
|
||||
getattr(queued_item, 'lang_code', None) == getattr(item, 'lang_code', None) and
|
||||
getattr(queued_item, 'speed', None) == getattr(item, 'speed', None) and
|
||||
getattr(queued_item, 'voice', None) == getattr(item, 'voice', None) and
|
||||
getattr(queued_item, 'save_option', None) == getattr(item, 'save_option', None) and
|
||||
getattr(queued_item, 'output_folder', None) == getattr(item, 'output_folder', None) and
|
||||
getattr(queued_item, 'subtitle_mode', None) == getattr(item, 'subtitle_mode', None) and
|
||||
getattr(queued_item, 'output_format', None) == getattr(item, 'output_format', None) and
|
||||
getattr(queued_item, 'total_char_count', None) == getattr(item, 'total_char_count', None)
|
||||
):
|
||||
is_duplicate = True
|
||||
break
|
||||
if is_duplicate:
|
||||
duplicates.append(os.path.basename(file_path))
|
||||
continue
|
||||
self.queue.append(item)
|
||||
if duplicates:
|
||||
QMessageBox.warning(self, "Duplicate Item(s)", "The following file(s) with the same attributes are already in the queue and were not added:\n" + '\n'.join(duplicates))
|
||||
self.process_queue()
|
||||
self.update_button_states()
|
||||
|
||||
def resizeEvent(self, event):
|
||||
super().resizeEvent(event)
|
||||
if hasattr(self, 'empty_overlay'):
|
||||
self.empty_overlay.resize(self.listwidget.size())
|
||||
|
||||
def update_button_states(self):
|
||||
# Enable Remove if at least one item is selected, else disable
|
||||
if hasattr(self, 'remove_button'):
|
||||
selected_count = len(self.listwidget.selectedItems())
|
||||
self.remove_button.setEnabled(selected_count > 0)
|
||||
if selected_count > 1:
|
||||
self.remove_button.setText(f"Remove selected ({selected_count})")
|
||||
else:
|
||||
self.remove_button.setText("Remove selected")
|
||||
# Disable Clear if queue is empty
|
||||
if hasattr(self, 'clear_button'):
|
||||
self.clear_button.setEnabled(bool(self.queue))
|
||||
|
||||
def show_context_menu(self, pos):
|
||||
from PyQt5.QtWidgets import QMenu, QAction
|
||||
from PyQt5.QtGui import QDesktopServices
|
||||
from PyQt5.QtCore import QUrl
|
||||
import os
|
||||
global_pos = self.listwidget.viewport().mapToGlobal(pos)
|
||||
selected_items = self.listwidget.selectedItems()
|
||||
menu = QMenu(self)
|
||||
if len(selected_items) == 1:
|
||||
remove_action = QAction("Remove this item", self)
|
||||
remove_action.triggered.connect(self.remove_item)
|
||||
menu.addAction(remove_action)
|
||||
go_to_folder_action = QAction("Go to folder", self)
|
||||
def go_to_folder():
|
||||
from PyQt5.QtWidgets import QMessageBox
|
||||
item = selected_items[0]
|
||||
display_name = item.text()
|
||||
for q in self.queue:
|
||||
if os.path.basename(q.file_name) == display_name:
|
||||
if not os.path.exists(q.file_name):
|
||||
QMessageBox.warning(self, "File Not Found", f"The file does not exist.")
|
||||
return
|
||||
folder = os.path.dirname(q.file_name)
|
||||
if os.path.exists(folder):
|
||||
QDesktopServices.openUrl(QUrl.fromLocalFile(folder))
|
||||
break
|
||||
go_to_folder_action.triggered.connect(go_to_folder)
|
||||
menu.addAction(go_to_folder_action)
|
||||
# Add Open file action
|
||||
open_file_action = QAction("Open file", self)
|
||||
def open_file():
|
||||
from PyQt5.QtWidgets import QMessageBox
|
||||
item = selected_items[0]
|
||||
display_name = item.text()
|
||||
for q in self.queue:
|
||||
if os.path.basename(q.file_name) == display_name:
|
||||
if not os.path.exists(q.file_name):
|
||||
QMessageBox.warning(self, "File Not Found", f"The file does not exist.")
|
||||
return
|
||||
QDesktopServices.openUrl(QUrl.fromLocalFile(q.file_name))
|
||||
break
|
||||
open_file_action.triggered.connect(open_file)
|
||||
menu.addAction(open_file_action)
|
||||
elif len(selected_items) > 1:
|
||||
remove_action = QAction(f"Remove selected ({len(selected_items)})", self)
|
||||
remove_action.triggered.connect(self.remove_item)
|
||||
menu.addAction(remove_action)
|
||||
# Always add Clear Queue
|
||||
clear_action = QAction("Clear Queue", self)
|
||||
clear_action.triggered.connect(self.clear_queue)
|
||||
menu.addAction(clear_action)
|
||||
menu.exec_(global_pos)
|
||||
Reference in New Issue
Block a user