medusa_gui.py

#!/usr/bin/env python3

import sys
import os
import resources_rc
from PySide6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout,
                              QPushButton, QLabel, QFileDialog, QMessageBox, QButtonGroup,
                              QRadioButton, QMenuBar, QMenu, QGroupBox, QStatusBar)
from PySide6.QtCore import Qt, QSize, QUrl
from PySide6.QtGui import QPixmap, QDesktopServices
from medusa_core import decompile_wavetable, recompile_wavetable, process_wavs, create_wavetable_bank
from version import __version__ as VERSION, __app_name__ as APP_NAME
from tools.version_manager import check_for_updates

class MedusaApp(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle(f'{APP_NAME} v{VERSION}')
        
        # Set Windows 95 style directly
        self.setStyleSheet("""
            QMainWindow {
                border-image: url(:/images/background.png) 0 0 0 0 stretch stretch;
            }
            QWidget {
                color: #000000;
                font-family: "MS Sans Serif", Arial;
                font-size: 12px;
            }
            QPushButton {
                background-color: rgba(192, 192, 192, 0.7);
                border: 2px solid #808080;
                border-style: solid;
                border-width: 2px;
                border-top-color: #a0a0a0;
                border-left-color: #a0a0a0;
                border-right-color: #303030;
                border-bottom-color: #303030;
                padding: 3px 6px;
                min-width: 85px;
                font-family: "MS Sans Serif", Arial;
                font-size: 12px;
                color: black;
            }
            QPushButton:pressed {
                border-top-color: #303030;
                border-left-color: #303030;
                border-right-color: #a0a0a0;
                border-bottom-color: #a0a0a0;
                background-color: rgba(128, 128, 128, 0.7);
            }
            QGroupBox {
                background-color: rgba(0, 0, 0, 0.5);
                text-transform: uppercase;
                border: 2px solid #808080;
                border-style: solid;
                border-width: 2px;
                border-top-color: #303030;
                border-left-color: #303030;
                border-right-color: #a0a0a0;
                border-bottom-color: #a0a0a0;
                margin: 4px;
                padding: 8px;
                spacing: 4px;
            }
            QGroupBox::title {
                subcontrol-origin: margin;
                subcontrol-position: top left;
                left: 8px;
                padding: 0 3px;
            }
            QRadioButton {
                spacing: 6px;
                font-family: "MS Sans Serif", Arial;
                font-size: 12px;
            }
            QRadioButton::indicator {
                width: 13px;
                height: 13px;
                border: 1px solid #808080;
                border-radius: 7px;
                background-color: #ffffff;
            }
            QRadioButton::indicator:checked {
                background-color: #000000;
                border: 1px solid #808080;
                border-radius: 7px;
            }
            QMenuBar {
                background-color: #c0c0c0;
                border-bottom: 1px solid #808080;
                font-family: "MS Sans Serif", Arial;
                font-size: 12px;
            }
            QMenuBar::item {
                spacing: 4px;
                padding: 2px 6px;
                background: transparent;
            }
            QMenuBar::item:selected {
                background-color: #404080;
                color: #ffffff;
            }
            QMenu {
                background-color: rgba(192, 192, 192, 0.7);
                border: 1px solid #808080;
                border-style: solid;
                border-width: 1px;
                border-top-color: #a0a0a0;
                border-left-color: #a0a0a0;
                border-right-color: #303030;
                border-bottom-color: #303030;
                font-family: "MS Sans Serif", Arial;
                font-size: 12px;
            }
            QMenu::item {
                padding: 2px 25px 2px 20px;
            }
            QMenu::item:selected {
                background-color: #000080;
                color: #ffffff;
            }
            QStatusBar {
                background-color: #c0c0c0;
                border-top: 1px solid #808080;
                border-style: solid;
                border-width: 1px;
                border-top-color: #404040;
                border-left-color: #404040;
                border-right-color: #ffffff;
                border-bottom-color: #ffffff;
                font-family: "MS Sans Serif", Arial;
                font-size: 12px;
                padding: 2px 3px;
            }
            QLabel { background: transparent; }
            QLabel#titleLabel {
                font-size: 12px;
                font-weight: bold;
                color: #000080;
            }
        """)
        
        # Create menu bar
        menubar = QMenuBar()
        self.setMenuBar(menubar)
        
        # File menu
        file_menu = QMenu("&File", self)
        menubar.addMenu(file_menu)
        file_menu.addAction("E&xit", self.close)
        
        # Help menu
        help_menu = QMenu("&Help", self)
        menubar.addMenu(help_menu)
        help_menu.addAction("Check for &Updates", self.check_updates)
        help_menu.addSeparator()
        help_menu.addAction("&About", self.about)
        
        # Create central widget and layout
        central_widget = QWidget()
        central_widget.setStyleSheet("background: transparent;")
        self.setCentralWidget(central_widget)
        # Add header image
        header_label = QLabel()
        pixmap = QPixmap(":/images/header.png")
        scaled_pixmap = pixmap.scaled(450, pixmap.height(), Qt.AspectRatioMode.KeepAspectRatio)
        header_label.setPixmap(scaled_pixmap)
        
        layout = QVBoxLayout(central_widget)
        layout.setSpacing(4)  # Match the spacing from QGroupBox
        layout.setContentsMargins(4, 0, 4, 4)  # Match the margins from QGroupBox
        layout.addWidget(header_label)
        
        # Set fixed window size
        self.setFixedSize(450, 200 + scaled_pixmap.height())
        
        # Create wavetable group
        create_group = QGroupBox("CREATE WAVETABLE")
        create_group.setStyleSheet("QGroupBox { background-color: rgba(255, 255, 255, 0.3); }")
        create_layout = QVBoxLayout(create_group)
        
        create_btn = QPushButton("Create from Audio Files")
        create_btn.clicked.connect(self.select_create_input)
        create_btn.setToolTip("Create wavetable bank directly from audio files (C)")
        create_btn.setShortcut("C")
        create_layout.addWidget(create_btn)
        
        # Selection mode radio buttons
        mode_group = QWidget()
        mode_layout = QHBoxLayout(mode_group)
        mode_layout.setContentsMargins(20, 0, 0, 0)
        
        self.selection_group = QButtonGroup(self)
        alpha_radio = QRadioButton("Alphanumeric")
        random_radio = QRadioButton("Random")
        alpha_radio.setChecked(True)
        
        self.selection_group.addButton(alpha_radio)
        self.selection_group.addButton(random_radio)
        
        mode_layout.addWidget(alpha_radio)
        mode_layout.addWidget(random_radio)
        mode_layout.addStretch()
        
        create_layout.addWidget(mode_group)
        layout.addWidget(create_group)
        
        # Tools group
        tools_group = QGroupBox("TOOLS")
        tools_group.setStyleSheet("QGroupBox { background-color: rgba(255, 255, 255, 0.3); }")
        tools_layout = QVBoxLayout(tools_group)
        
        decompile_btn = QPushButton("Decompile .polyend File")
        decompile_btn.clicked.connect(self.select_decompile_input)
        decompile_btn.setToolTip("Extract wavetables from a .polyend file to WAV files")
        decompile_btn.setShortcut("D")
        tools_layout.addWidget(decompile_btn)
        
        recompile_btn = QPushButton("Recompile Wavetables")
        recompile_btn.clicked.connect(self.select_recompile_input)
        recompile_btn.setToolTip("Create .polyend file from processed WAV files (R)")
        recompile_btn.setShortcut("R")
        tools_layout.addWidget(recompile_btn)
        
        layout.addWidget(tools_group)
        
        # Add status bar with link
        status_bar = QStatusBar()
        self.setStatusBar(status_bar)
        
        # Create link label
        link_label = QLabel()
        link_label.setText('<a href="https://github.com/jkmaxwell/medusa_tools" style="color: #0000EE; text-decoration: underline;">2025 Justin Maxwell</a>')
        link_label.setTextFormat(Qt.TextFormat.RichText)
        link_label.setOpenExternalLinks(True)
        
        # Add permanent widget to right side of status bar
        status_bar.addPermanentWidget(link_label)
        status_bar.showMessage("Ready")
    
    def update_status(self, message):
        """Update status bar message."""
        self.statusBar().showMessage(message)
    
    def select_decompile_input(self):
        self.update_status("Selecting file to decompile...")
        dialog = QFileDialog(self)
        dialog.setWindowTitle("Select Medusa Wavetable File")
        dialog.setNameFilter("Polyend Files (*.polyend)")
        dialog.setFileMode(QFileDialog.FileMode.ExistingFile)
        dialog.setViewMode(QFileDialog.ViewMode.Detail)
        
        if dialog.exec() == QFileDialog.DialogCode.Accepted:
            input_file = dialog.selectedFiles()[0]
            self.update_status(f"Decompiling {os.path.basename(input_file)}...")
            result = decompile_wavetable(input_file)
            if result['success']:
                QMessageBox.information(
                    self,
                    "Success",
                    f"Extracted {result['num_wavetables']} wavetables to {result['output_dir']}"
                )
            else:
                QMessageBox.critical(
                    self,
                    "Error",
                    f"Error decompiling wavetable: {result['error']}"
                )
        self.update_status("Ready")
    
    def select_recompile_input(self):
        self.update_status("Selecting files to recompile...")
        dialog = QFileDialog(self)
        dialog.setWindowTitle("Select Directory Containing Wavetable WAV Files")
        dialog.setFileMode(QFileDialog.FileMode.Directory)
        dialog.setOption(QFileDialog.Option.ShowDirsOnly)
        
        if dialog.exec() == QFileDialog.DialogCode.Accepted:
            input_dir = dialog.selectedFiles()[0]
            
            parent_dir = os.path.dirname(input_dir)
            default_output = os.path.join(parent_dir, "recompiled.polyend")
            
            save_dialog = QFileDialog(self)
            save_dialog.setWindowTitle("Save Recompiled Wavetable File As")
            save_dialog.setNameFilter("Polyend Files (*.polyend)")
            save_dialog.setAcceptMode(QFileDialog.AcceptMode.AcceptSave)
            save_dialog.setDirectory(parent_dir)
            save_dialog.selectFile("recompiled.polyend")
            
            if save_dialog.exec() == QFileDialog.DialogCode.Accepted:
                output_file = save_dialog.selectedFiles()[0]
                self.update_status("Recompiling wavetables...")
                result = recompile_wavetable(input_dir, output_file)
                if result['success']:
                    QMessageBox.information(
                        self,
                        "Success",
                        f"Successfully recompiled {result['num_wavetables']} wavetables to {result['output_file']}"
                    )
                else:
                    QMessageBox.critical(
                        self,
                        "Error",
                        f"Error recompiling wavetables: {result['error']}"
                    )
        self.update_status("Ready")
    
    def select_create_input(self):
        self.update_status("Selecting audio files...")
        input_dir = QFileDialog.getExistingDirectory(
            self,
            "Select Directory Containing Audio Files",
            "",
            QFileDialog.Option.ShowDirsOnly
        )
        
        if not input_dir:
            self.update_status("Ready")
            return
            
        random_mode = self.selection_group.checkedButton().text() == "Random"
        
        output_file, _ = QFileDialog.getSaveFileName(
            self,
            "Save Wavetable Bank As",
            "wavetables.polyend",
            "Polyend Files (*.polyend)"
        )
        
        if not output_file:
            self.update_status("Ready")
            return
            
        self.update_status("Creating wavetable bank...")
        result = create_wavetable_bank(
            input_dir,
            output_file,
            random_order=random_mode
        )
        
        if result['success']:
            QMessageBox.information(
                self,
                "Success",
                f"Successfully created wavetable bank:\n"
                f"- Output file: {result['output_file']}\n"
                f"- Number of wavetables: {result['num_wavetables']}\n"
                f"- Source files: {len(result['source_files'])}"
            )
        else:
            QMessageBox.critical(
                self,
                "Error",
                f"Error creating wavetable bank: {result['error']}"
            )
        self.update_status("Ready")
    
    def check_updates(self):
        """Check for available updates and notify user."""
        self.update_status("Checking for updates...")
        update_info = check_for_updates()
        
        if update_info:
            msg = QMessageBox(self)
            msg.setWindowTitle("Update Available")
            msg.setTextFormat(Qt.TextFormat.RichText)
            msg.setText(
                f"<h3>Update Available!</h3>"
                f"<p>Current version: {update_info['current_version']}<br>"
                f"Latest version: {update_info['latest_version']}</p>"
                f"<p>Release Notes:</p>"
                f"<pre>{update_info['release_notes']}</pre>"
                f"<p>Would you like to download the update?</p>"
            )
            msg.setStandardButtons(QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No)
            msg.setDefaultButton(QMessageBox.StandardButton.Yes)
            
            if msg.exec() == QMessageBox.StandardButton.Yes:
                QDesktopServices.openUrl(QUrl(update_info['download_url']))
        else:
            QMessageBox.information(
                self,
                "No Updates Available",
                f"You're running the latest version ({VERSION})!"
            )
        self.update_status("Ready")
    
    def about(self):
        about_box = QMessageBox(self)
        about_box.setWindowTitle(f"About {APP_NAME}")
        about_box.setTextFormat(Qt.TextFormat.RichText)
        about_box.setText(
            f"<h3>{APP_NAME} v{VERSION}</h3>"
            "<p>A tool for working with Polyend Medusa synthesizer wavetables.</p>"
            "<p><b>Features:</b></p>"
            "<ul>"
            "<li>Extract wavetables from .polyend files</li>"
            "<li>Create wavetable banks from audio files</li>"
            "<li>Support for random or alphabetical file selection</li>"
            "</ul>"
            "<p>© 2024 All rights reserved.</p>"
        )
        about_box.setStyleSheet(self.styleSheet())
        about_box.exec()

def main():
    # Use existing QApplication instance if one exists
    app = QApplication.instance()
    if app is None:
        app = QApplication(sys.argv)
    
    # Set application name and organization for macOS menu bar
    app.setApplicationName(APP_NAME)
    app.setOrganizationName("Justin Maxwell")
    app.setOrganizationDomain("code404.com")
    app.setApplicationDisplayName(APP_NAME)
    
    window = MedusaApp()
    window.show()
    sys.exit(app.exec())

if __name__ == "__main__":
    main()