Dymas/abogen
1
0
Fork 0
mirror of https://github.com/denizsafak/abogen.git synced 2026-07-18 13:40:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add queue summary window, check GPU errors

Browse Source
This commit is contained in:
Deniz Şafak
2025-07-11 05:23:03 +03:00
parent 9b88a1ba55
commit ef1063018f
2 changed files with 66 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
abogen/gui.py
+50 -2
View File
@@ -1536,7 +1536,6 @@ class abogen(QWidget):
QMessageBox.warning(self, "Duplicate Item", "This item is already in the queue.") QMessageBox.warning(self, "Duplicate Item", "This item is already in the queue.")
return return
print("Adding:", item_queue)
self.enqueue(item_queue) self.enqueue(item_queue)
# Clear input after adding to queue # Clear input after adding to queue
self.input_box.clear_input() self.input_box.clear_input()
@@ -1758,6 +1757,48 @@ class abogen(QWidget):
Thread(target=gpu_and_load, daemon=True).start() Thread(target=gpu_and_load, daemon=True).start()
def show_queue_summary(self):
"""Show a styled, resizable summary dialog after queue finishes."""
if not self.queued_items:
return
# Build HTML summary for better styling
summary_html = "<html><body>"
for idx, item in enumerate(self.queued_items, 1):
status = "<span style='color:green;'>Completed</span>"
output = getattr(item, "output_path", None)
if not output:
output = "Unknown"
summary_html += (
f"<b>{idx}) {os.path.basename(item.file_name)}</b><br>"
f"<span style='color:#555;'>Language:</span> {item.lang_code}<br>"
f"<span style='color:#555;'>Voice:</span> {item.voice}<br>"
f"<span style='color:#555;'>Output:</span> <span style='color:#007acc;'>{output}</span><br>"
f"<span style='color:#555;'>Status:</span> {status}"
f"<br><br>"
)
summary_html += "</body></html>"
dialog = QDialog(self)
dialog.setWindowTitle("Queue Summary")
dialog.resize(600, 500) # Make window resizable and larger
layout = QVBoxLayout(dialog)
text_edit = QTextEdit(dialog)
text_edit.setReadOnly(True)
text_edit.setHtml(summary_html)
layout.addWidget(text_edit)
close_btn = QPushButton("Close", dialog)
close_btn.setFixedHeight(36)
close_btn.clicked.connect(dialog.accept)
layout.addWidget(close_btn)
dialog.setLayout(layout)
dialog.setMinimumSize(400, 300)
dialog.setSizeGripEnabled(True) # Allow resizing
dialog.exec_()
def on_conversion_finished(self, message, output_path): def on_conversion_finished(self, message, output_path):
prevent_sleep_end() prevent_sleep_end()
if message == "Cancelled": if message == "Cancelled":
@@ -1786,6 +1827,10 @@ class abogen(QWidget):
self.update_log(message) self.update_log(message)
if output_path: if output_path:
self.last_output_path = output_path self.last_output_path = output_path
# Store output_path in the current queued item if in queue mode
if self.queued_items and self.current_queue_index < len(self.queued_items):
self.queued_items[self.current_queue_index].output_path = output_path
self.etr_label.hide() # Hide ETR label self.etr_label.hide() # Hide ETR label
self.progress_bar.setValue(100) self.progress_bar.setValue(100)
self.progress_bar.hide() self.progress_bar.hide()
@@ -1808,7 +1853,7 @@ class abogen(QWidget):
if self.open_file_btn: if self.open_file_btn:
self.open_file_btn.setVisible(show_open_file_button) self.open_file_btn.setVisible(show_open_file_button)
# Only show finish_widget if queue is done --- # Only show finish_widget if queue is done
if self.current_queue_index + 1 >= len(self.queued_items) or not self.queued_items: if self.current_queue_index + 1 >= len(self.queued_items) or not self.queued_items:
# Queue finished, show finish screen # Queue finished, show finish screen
self.controls_widget.hide() self.controls_widget.hide()
@@ -1816,6 +1861,9 @@ class abogen(QWidget):
self.log_text.moveCursor(QTextCursor.End) self.log_text.moveCursor(QTextCursor.End)
self.log_text.ensureCursorVisible() self.log_text.ensureCursorVisible()
save_config(self.config) save_config(self.config)
# Show queue summary if more than one item
if len(self.queued_items) > 1:
self.show_queue_summary()
else: else:
# More items in queue: clear log and reload for next item # More items in queue: clear log and reload for next item
self.log_text.clear() self.log_text.clear()
abogen/utils.py
+16 -8
View File
@@ -217,14 +217,22 @@ def calculate_text_length(text):
def get_gpu_acceleration(enabled): def get_gpu_acceleration(enabled):
from torch.cuda import is_available try:
import torch
if not enabled: from torch.cuda import is_available
return "CUDA GPU available but using CPU.", False if not enabled:
return "CUDA GPU available but using CPU.", False
if is_available(): if is_available():
return "CUDA GPU available and enabled.", True return "CUDA GPU available and enabled.", True
return "CUDA GPU is not available. Using CPU.", False # Gather more diagnostic info if CUDA is not available
try:
cuda_devices = torch.cuda.device_count()
cuda_error = torch.cuda.get_device_name(0) if cuda_devices > 0 else "No devices found"
except Exception as e:
cuda_error = str(e)
return f"CUDA GPU is not available. Using CPU.\nDetails: {cuda_error}", False
except Exception as e:
return f"Error checking CUDA GPU: {e}", False
def prevent_sleep_start(): def prevent_sleep_start():