129 lines
No EOL
5.3 KiB
Python
129 lines
No EOL
5.3 KiB
Python
# main.py
|
|
import sys
|
|
import os
|
|
from PyQt6.QtWidgets import QMainWindow, QTabWidget, QApplication, QMessageBox, QVBoxLayout, QWidget
|
|
from PyQt6.QtCore import pyqtSlot
|
|
|
|
# Import our unified data layer
|
|
import database
|
|
|
|
# Import the UI components from our sub-package
|
|
from tabs.sandbox_tab import SandboxTab
|
|
from tabs.review_tab import ReviewTab
|
|
from tabs.settings_tab import SettingsTab
|
|
|
|
# Import our background engine runners
|
|
import anki_exporter
|
|
import video_generator
|
|
|
|
class MainWindow(QMainWindow):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.setWindowTitle("Castilian Spanish Voice Trainer")
|
|
self.setMinimumSize(900, 650)
|
|
|
|
# 1. Ensure database structure exists on startup
|
|
database.ensure_database_populated()
|
|
|
|
# 2. In-memory cache for application configurations
|
|
self.app_settings = {}
|
|
self.refresh_settings_cache()
|
|
|
|
# 3. Setup central tabbed layout container
|
|
self.central_widget = QWidget()
|
|
self.setCentralWidget(self.central_widget)
|
|
self.main_layout = QVBoxLayout(self.central_widget)
|
|
|
|
self.tab_widget = QTabWidget()
|
|
self.main_layout.addWidget(self.tab_widget)
|
|
|
|
# 4. Instantiate isolated tab modules
|
|
self.sandbox_tab = SandboxTab()
|
|
self.review_tab = ReviewTab()
|
|
self.settings_tab = SettingsTab()
|
|
|
|
# 5. Mount tabs with Database Sandbox on the far left (Index 0)
|
|
self.tab_widget.addTab(self.sandbox_tab, "Database Sandbox")
|
|
self.tab_widget.addTab(self.review_tab, "Flashcard Review")
|
|
self.tab_widget.addTab(self.settings_tab, "Configuration Settings")
|
|
|
|
# 6. Wire up the cross-module communication channels (Signals)
|
|
self.wire_application_signals()
|
|
|
|
def refresh_settings_cache(self):
|
|
"""Loads all database settings into an easy-to-read dictionary."""
|
|
self.app_settings = database.load_all_settings()
|
|
|
|
def wire_application_signals(self):
|
|
"""Connects tab user-actions to execution managers inside main.py."""
|
|
# Settings Tab Signal Actions
|
|
self.settings_tab.settings_changed.connect(self.handle_settings_update)
|
|
self.settings_tab.export_anki_requested.connect(self.execute_anki_export)
|
|
self.settings_tab.generate_video_requested.connect(self.execute_video_generation)
|
|
|
|
# Sandbox / Review Tab synchronization signals
|
|
self.sandbox_tab.data_mutated.connect(self.refresh_ui_views)
|
|
|
|
@pyqtSlot(str, str)
|
|
def handle_settings_update(self, key, value):
|
|
"""Saves a modified setting value directly down to the database and updates memory cache."""
|
|
database.save_setting_to_db(key, value)
|
|
self.refresh_settings_cache()
|
|
|
|
@pyqtSlot(str)
|
|
def execute_anki_export(self, full_deck_name):
|
|
"""Bridges data coordinates out of SQLite to trigger the text-to-speech anki file compiler."""
|
|
self.refresh_settings_cache()
|
|
output_dir = self.app_settings.get("anki_export_directory", os.path.expanduser("~"))
|
|
output_file = os.path.join(output_dir, f"{full_deck_name.replace('::', '_')}.apkg")
|
|
|
|
# Pull all active translation records out via explicit SQL matching
|
|
records = database.get_all_translations_explicit()
|
|
|
|
if not records:
|
|
QMessageBox.warning(self, "Export Failed", "The database contains no translation records to export.")
|
|
return
|
|
|
|
try:
|
|
anki_exporter.compile_anki_package(records, output_file, full_deck_name)
|
|
QMessageBox.information(self, "Export Success", f"Successfully compiled light-weight TTS deck to:\n{output_file}")
|
|
except Exception as e:
|
|
QMessageBox.critical(self, "Export Error", f"Failed to parse or write Anki package archive:\n{str(e)}")
|
|
|
|
@pyqtSlot()
|
|
def execute_video_generation(self):
|
|
"""Coordinates data streams to invoke the independent video compilation generator."""
|
|
self.refresh_settings_cache()
|
|
records = database.get_all_translations_explicit()
|
|
|
|
if not records:
|
|
QMessageBox.warning(self, "Generation Failed", "No phrases available to construct a training video playlist loops.")
|
|
return
|
|
|
|
# UI updates can block execution loops, so we notify before processing
|
|
self.statusBar().showMessage("Generating training video assets. Please wait...")
|
|
QApplication.processEvents()
|
|
|
|
try:
|
|
video_generator.generate_training_video(records, self.app_settings)
|
|
QMessageBox.information(self, "Success", "MP4 Video compilation completed successfully.")
|
|
self.statusBar().showMessage("Video compilation completed.", 5000)
|
|
except Exception as e:
|
|
QMessageBox.critical(self, "Video Error", f"An error occurred during frame rendering sequences:\n{str(e)}")
|
|
self.statusBar().clearMessage()
|
|
|
|
@pyqtSlot()
|
|
def refresh_ui_views(self):
|
|
"""Instructs distinct active tabs to reload data windows after database changes occur."""
|
|
self.sandbox_tab.reload_table_display()
|
|
self.review_tab.reload_review_pool()
|
|
|
|
if __name__ == "__main__":
|
|
app = QApplication(sys.argv)
|
|
|
|
# Modern clean styling choices for desktop applications
|
|
app.setStyle('Fusion')
|
|
|
|
window = MainWindow()
|
|
window.show()
|
|
sys.exit(app.exec()) |