start-dpg-ui.py

#!/usr/bin/env python3
"""
Main entry point for the Dear PyGui version of OneTrainer

This script launches the OneTrainer application with the Dear PyGui UI.
"""

import os
import sys
import traceback

# Add the current directory to the path to allow importing local modules
current_dir = os.path.dirname(os.path.abspath(__file__))
if current_dir not in sys.path:
    sys.path.insert(0, current_dir)

# Create a startup banner
banner = """
======================================================
            OneTrainer with Dear PyGui
======================================================

Starting OneTrainer with the Dear PyGui interface...
"""

def check_dependencies():
    """
    Check if required dependencies are installed
    
    Returns:
        True if all dependencies are installed, False otherwise
    """
    try:
        # Check Dear PyGui
        import dearpygui.dearpygui as dpg
        print("✓ Dear PyGui is installed")
        
        # Check other key dependencies
        import torch
        print("✓ PyTorch is installed")
        
        # Try to import key OneTrainer modules
        from modules.util.config.TrainConfig import TrainConfig
        print("✓ OneTrainer modules are available")
        
        return True
    except ImportError as e:
        print(f"Error: {e}")
        print("\nPlease install the missing dependencies:")
        print("  pip install dearpygui")
        print("  pip install -r requirements.txt")
        return False
        
def main():
    """Main entry point"""
    print(banner)
    
    # Check dependencies
    if not check_dependencies():
        return 1
        
    try:
        # Import and run the application
        from dpg_ui.app import OneTrainerApp
        
        print("Launching OneTrainer with Dear PyGui interface...")
        app = OneTrainerApp()
        app.run()
        
        return 0
    except Exception as e:
        print(f"Error launching the application: {e}")
        traceback.print_exc()
        return 1
        
if __name__ == "__main__":
    sys.exit(main())