fix_and_run_ui.sh

#!/usr/bin/env bash
# Fix and run OneTrainer UI with DearPyGui 1.8.0

set -e

echo "==== OneTrainer UI with DearPyGui 1.8.0 ===="
echo "This script will install DearPyGui 1.8.0 and run the UI."

# Use python3 as backup if python command isn't available
if ! command -v python &> /dev/null && command -v python3 &> /dev/null; then
    export OT_PYTHON_CMD="python3"
fi

source "${BASH_SOURCE[0]%/*}/lib.include.sh"
prepare_runtime_environment

# Forcibly install DearPyGui 1.8.0 which is known to work more reliably
echo "Installing DearPyGui 1.8.0..."
run_pip_in_active_env uninstall -y dearpygui
run_pip_in_active_env install dearpygui==1.8.0
echo "DearPyGui 1.8.0 installed successfully"

# Fix the Additional Embeddings tab to use a more compatible implementation
EMBEDDINGS_TAB_PATH="dpg_ui/tabs/additional_embeddings_tab.py"
if [ -f "$EMBEDDINGS_TAB_PATH" ]; then
    # Backup the file if a backup doesn't already exist
    if [ ! -f "${EMBEDDINGS_TAB_PATH}.bak" ]; then
        cp "$EMBEDDINGS_TAB_PATH" "${EMBEDDINGS_TAB_PATH}.bak"
        echo "Created backup of $EMBEDDINGS_TAB_PATH"
    fi
    
    # Replace the file_entry with a simpler implementation
    run_python_in_active_env -c "
import re

# Read the file
with open('$EMBEDDINGS_TAB_PATH', 'r') as f:
    content = f.read()

# Replace the file_entry with a simpler implementation
pattern = r'llama_path\s*=\s*Components\.file_entry\(.*?\n.*?\n.*?\n.*?\n.*?\)'
replacement = \"\"\"                    # Use a simpler approach to avoid file dialog issues
                    input_id = dpg.add_input_text(
                        default_value=\"\",
                        width=250,
                        callback=self.on_llama_path_change
                    )
                    browse_button = dpg.add_button(
                        label=\"...\",
                        width=30,
                        height=30
                    )
                    
                    # Set up button callback
                    def show_file_dialog():
                        with dpg.file_dialog(
                            label=\"Select LLaMA Model\",
                            width=700,
                            height=400,
                            callback=lambda s, a: dpg.set_value(input_id, a.get(\"file_path_name\", \"\")),
                            show=True
                        ):
                            pass
                            
                    dpg.set_item_callback(browse_button, show_file_dialog)\"\"\"

# Use regex with dotall mode to match across multiple lines
new_content = re.sub(pattern, replacement, content, flags=re.DOTALL)

# Write the modified content back to the file
with open('$EMBEDDINGS_TAB_PATH', 'w') as f:
    f.write(new_content)
"
    echo "Fixed $EMBEDDINGS_TAB_PATH"
fi

# Create a simple no-frills launcher for the UI
LAUNCHER_SCRIPT="start-OneTrainerUI.py"
cat > "$LAUNCHER_SCRIPT" << 'EOF'
#!/usr/bin/env python3
"""
Simple launcher for OneTrainer UI
"""

import os
import sys
import importlib.util

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

# Banner
print("=== OneTrainer with DearPyGui ===")

# Check if DearPyGui is installed
try:
    import dearpygui
    print(f"DearPyGui version: {getattr(dearpygui, '__version__', 'unknown')}")
except ImportError:
    print("DearPyGui is not installed. Please run fix_and_run_ui.sh")
    sys.exit(1)

# Very simple error handling
try:
    # Import and run the application
    from dpg_ui.app import OneTrainerApp
    
    app = OneTrainerApp()
    app.run()
except Exception as e:
    print(f"Error: {e}")
    import traceback
    traceback.print_exc()
    sys.exit(1)
EOF

chmod +x "$LAUNCHER_SCRIPT"
echo "Created $LAUNCHER_SCRIPT"

# Run the UI
echo "Starting OneTrainer UI..."
run_python_in_active_env ./"$LAUNCHER_SCRIPT"