ui_elements.py

import tkinter as tk
from tkinter import ttk, scrolledtext, filedialog, messagebox
import tkinter.font as tkfont
import sys
import os
from typing import Optional, Callable, Dict, Any, Union, TYPE_CHECKING

if TYPE_CHECKING:
    from tkinter import _ButtonCommand
else:
    _ButtonCommand = Any

# =============================================================================
# PREMIUM UI DESIGN SYSTEM - MODERN & SCALABLE
# =============================================================================

# --- Enhanced Color Palette ---
class Colors:
    # Premium Brand Colors - Modern gradient-friendly palette
    PRIMARY = "#6366F1"        # Indigo - Premium, professional
    PRIMARY_HOVER = "#4F46E5"  # Darker indigo for hover
    PRIMARY_ACTIVE = "#4338CA" # Even darker for active state
    PRIMARY_LIGHT = "#A5B4FC"  # Light indigo for accents
    PRIMARY_GHOST = "#E0E7FF" # Light indigo instead of transparent
    
    # Semantic Colors
    SUCCESS = "#10B981"        # Emerald green
    SUCCESS_HOVER = "#059669"  # Darker emerald
    SUCCESS_LIGHT = "#A7F3D0"  # Light emerald
    
    WARNING = "#F59E0B"        # Amber
    WARNING_HOVER = "#D97706"  # Darker amber
    WARNING_LIGHT = "#FDE68A"  # Light amber
    
    ERROR = "#EF4444"          # Red
    ERROR_HOVER = "#DC2626"    # Darker red
    ERROR_LIGHT = "#FECACA"    # Light red
    
    INFO = "#3B82F6"          # Blue
    INFO_HOVER = "#2563EB"    # Darker blue
    INFO_LIGHT = "#BFDBFE"    # Light blue
    
    # Premium Background System
    BG_PRIMARY = "#FAFBFC"     # Primary background - slightly off-white
    BG_SECONDARY = "#F7F9FB"   # Secondary background
    BG_TERTIARY = "#F1F4F7"    # Tertiary background
    BG_CARD = "#FFFFFF"        # Card backgrounds
    BG_ELEVATED = "#FFFFFF"    # Elevated surfaces
    
    # Glass/Blur Effects (simulated with lighter colors)
    GLASS_PRIMARY = "#F8F9FA"   # Light gray instead of transparent white
    GLASS_SECONDARY = "#F1F3F4" # Slightly darker gray
    
    # Premium Surface Colors
    SURFACE_DEFAULT = "#FFFFFF"
    SURFACE_HOVER = "#F8FAFC"
    SURFACE_ACTIVE = "#F1F5F9"
    SURFACE_DISABLED = "#F8FAFC"
    
    # Professional Text Hierarchy
    TEXT_PRIMARY = "#1E293B"   # Very dark slate
    TEXT_SECONDARY = "#475569" # Medium slate
    TEXT_TERTIARY = "#64748B"  # Light slate
    TEXT_MUTED = "#94A3B8"     # Very light slate
    TEXT_INVERSE = "#FFFFFF"   # White text
    TEXT_ACCENT = "#6366F1"    # Brand color text
    
    # Sophisticated Border System
    BORDER_SUBTLE = "#F1F5F9"  # Very light borders
    BORDER_DEFAULT = "#E2E8F0" # Default borders
    BORDER_STRONG = "#CBD5E1"  # Strong borders
    BORDER_ACCENT = "#6366F1"  # Accent borders
    
    # Professional Shadow System (using gray colors since tkinter doesn't support alpha)
    SHADOW_SM = "#F1F5F9"      # Very light gray
    SHADOW_MD = "#E2E8F0"      # Light gray
    SHADOW_LG = "#CBD5E1"      # Medium gray
    SHADOW_XL = "#94A3B8"      # Darker gray
    
    # Gradient Colors for Premium Effects
    GRADIENT_PRIMARY = ["#6366F1", "#8B5CF6"]   # Indigo to purple
    GRADIENT_SUCCESS = ["#10B981", "#059669"]   # Emerald gradient
    GRADIENT_SUNSET = ["#F59E0B", "#EF4444"]    # Warm gradient
    GRADIENT_OCEAN = ["#06B6D4", "#3B82F6"]     # Cool gradient

# --- Professional Typography System ---
class Typography:
    # Premium Font Stack
    PRIMARY_FONT = "Inter"     # Primary UI font
    HEADING_FONT = "SF Pro Display" # For headings (fallback to primary)
    MONO_FONT = "SF Mono"      # Monospace font
    
    # Comprehensive Size Scale
    XXS = 9   # Tiny text
    XS = 10   # Extra small
    SM = 11   # Small
    BASE = 12 # Base size
    MD = 13   # Medium
    LG = 14   # Large
    XL = 16   # Extra large
    XXL = 18  # Double extra large
    XXXL = 20 # Triple extra large
    H4 = 22   # Heading 4
    H3 = 24   # Heading 3
    H2 = 28   # Heading 2
    H1 = 32   # Heading 1
    DISPLAY = 36 # Display text
    
    # Professional Font Weights (tkinter-compatible)
    THIN = "normal"
    EXTRALIGHT = "normal"
    LIGHT = "normal"
    NORMAL = "normal"
    MEDIUM = "normal"
    SEMIBOLD = "bold"
    BOLD = "bold"
    EXTRABOLD = "bold"
    BLACK = "bold"

# --- Advanced Spacing System ---
class Spacing:
    # Micro spacing
    XXS = 2
    XS = 4
    SM = 6
    # Standard spacing
    MD = 8
    LG = 12
    XL = 16
    XXL = 20
    XXXL = 24
    # Large spacing
    GIANT = 32
    HUGE = 40
    MASSIVE = 48
    
    # Component-specific spacing
    BUTTON_PADDING_X = 16
    BUTTON_PADDING_Y = 8
    CARD_PADDING = 20
    SECTION_MARGIN = 24
    
# --- Animation & Effects System ---
class Effects:
    # Timing functions
    FAST = 150     # Quick interactions
    NORMAL = 250   # Standard transitions
    SLOW = 350     # Deliberate animations
    
    # Easing curves (for future CSS-like transitions)
    EASE_IN = "ease-in"
    EASE_OUT = "ease-out"
    EASE_IN_OUT = "ease-in-out"
    
    # Shadow definitions
    SHADOW_NONE = "none"
    SHADOW_SM = "0 1px 2px rgba(0, 0, 0, 0.05)"
    SHADOW_MD = "0 4px 6px rgba(0, 0, 0, 0.1)"
    SHADOW_LG = "0 10px 15px rgba(0, 0, 0, 0.1)"
    SHADOW_XL = "0 20px 25px rgba(0, 0, 0, 0.15)"
    
    # Border radius for modern rounded corners
    RADIUS_NONE = 0
    RADIUS_SM = 3
    RADIUS_MD = 6
    RADIUS_LG = 8
    RADIUS_XL = 12
    RADIUS_FULL = 9999

# --- Icon System (Unicode/Text-based for cross-platform compatibility) ---
class Icons:
    # Navigation
    ARROW_LEFT = "←"
    ARROW_RIGHT = "→"
    ARROW_UP = "↑" 
    ARROW_DOWN = "↓"
    
    # Actions
    PLAY = "▶"
    PAUSE = "⏸"
    STOP = "⏹"
    REFRESH = "↻"
    DOWNLOAD = "⬇"
    UPLOAD = "⬆"
    SYNC = "⟲"
    
    # Status
    SUCCESS = "✓"
    ERROR = "✗"
    WARNING = "⚠"
    INFO = "ℹ"
    
    # Files & Folders
    FOLDER = "📁"
    FILE = "📄"
    GEAR = "⚙"
    KEY = "🔑"
    LINK = "🔗"
    
    # Security & Tools
    SECURITY = "🔒"
    COPY = "📋"
    
    # Interface
    MENU = "☰"
    CLOSE = "✕"
    MINIMIZE = "−"
    MAXIMIZE = "□"

# =============================================================================
# ADVANCED UI COMPONENT SYSTEM
# =============================================================================

# Global font configuration
FONT_FAMILY_PRIMARY = "TkDefaultFont"
FONT_FAMILY_MONO = "TkFixedFont"

def get_premium_font_family():
    """Gets the best available font for a premium look."""
    try:
        available_fonts = tkfont.families()
        
        # Premium font preferences by platform
        if sys.platform == "win32":
            preferred = ["Segoe UI Variable", "Segoe UI", "Calibri"]
        elif sys.platform == "darwin":  # macOS
            preferred = ["SF Pro Display", "Helvetica Neue", "Lucida Grande"]
        else:  # Linux
            preferred = ["Inter", "Roboto", "Noto Sans", "Ubuntu", "Cantarell", "DejaVu Sans"]
        
        for font in preferred:
            if font in available_fonts:
                return font
                
        return "TkDefaultFont"
    except Exception:
        return "TkDefaultFont"

def get_premium_mono_font():
    """Gets the best available monospace font."""
    try:
        available_fonts = tkfont.families()
        
        if sys.platform == "win32":
            preferred = ["Consolas", "Courier New"]
        elif sys.platform == "darwin":
            preferred = ["SF Mono", "Menlo", "Monaco"]
        else:
            preferred = ["JetBrains Mono", "Fira Code", "Ubuntu Mono", "DejaVu Sans Mono"]
            
        for font in preferred:
            if font in available_fonts:
                return font
                
        return "TkFixedFont"
    except Exception:
        return "TkFixedFont"

def init_font_config():
    """Initializes premium font configuration."""
    global FONT_FAMILY_PRIMARY, FONT_FAMILY_MONO
    FONT_FAMILY_PRIMARY = get_premium_font_family()
    FONT_FAMILY_MONO = get_premium_mono_font()

# =============================================================================
# PREMIUM COMPONENT LIBRARY
# =============================================================================

class PremiumButton:
    """Enhanced button component with hover effects and styling options."""
    
    @staticmethod
    def create_primary(parent, text: str, command: Optional[Callable[[], Any]] = None, icon: Optional[str] = None, size: str = "md"):
        """Creates a primary button with premium styling."""
        btn_frame = tk.Frame(parent, bg=Colors.BG_PRIMARY)
        
        # Size configurations
        sizes = {
            "sm": {"font_size": Typography.SM, "pad_x": 12, "pad_y": 6, "min_width": 60},
            "md": {"font_size": Typography.MD, "pad_x": 16, "pad_y": 8, "min_width": 80},
            "lg": {"font_size": Typography.LG, "pad_x": 20, "pad_y": 10, "min_width": 100}
        }
        
        size_config = sizes.get(size, sizes["md"])
        
        # Create button text with optional icon
        button_text = f"{icon} {text}" if icon else text
        
        btn = tk.Button(
            btn_frame,
            text=button_text,
            command=command if command is not None else lambda: None,
            font=(FONT_FAMILY_PRIMARY, size_config["font_size"], Typography.MEDIUM),
            bg=Colors.PRIMARY,
            fg=Colors.TEXT_INVERSE,
            activebackground=Colors.PRIMARY_HOVER,
            activeforeground=Colors.TEXT_INVERSE,
            relief=tk.FLAT,
            borderwidth=0,
            cursor="hand2",
            padx=size_config["pad_x"],
            pady=size_config["pad_y"],
            width=10,  # Set minimum width in characters
            height=2   # Set minimum height in text lines
        )
        
        # Add hover effects
        def on_enter(e):
            btn.config(bg=Colors.PRIMARY_HOVER)
        
        def on_leave(e):
            btn.config(bg=Colors.PRIMARY)
            
        btn.bind("<Enter>", on_enter)
        btn.bind("<Leave>", on_leave)
        btn.pack(fill=tk.BOTH, expand=True)
        
        # Use a custom attribute container to avoid Pylance warnings
        setattr(btn_frame, '_button', btn)
        return btn_frame
    
    @staticmethod
    def create_secondary(parent, text: str, command: Optional[Callable[[], Any]] = None, icon: Optional[str] = None, size: str = "md"):
        """Creates a secondary button with outline styling."""
        btn_frame = tk.Frame(parent, bg=Colors.BG_PRIMARY)
        
        sizes = {
            "sm": {"font_size": Typography.SM, "pad_x": 12, "pad_y": 6, "min_width": 60},
            "md": {"font_size": Typography.MD, "pad_x": 16, "pad_y": 8, "min_width": 80},
            "lg": {"font_size": Typography.LG, "pad_x": 20, "pad_y": 10, "min_width": 100}
        }
        
        size_config = sizes.get(size, sizes["md"])
        button_text = f"{icon} {text}" if icon else text
        
        btn = tk.Button(
            btn_frame,
            text=button_text,
            command=command if command is not None else lambda: None,
            font=(FONT_FAMILY_PRIMARY, size_config["font_size"], Typography.MEDIUM),
            bg=Colors.BG_CARD,
            fg=Colors.TEXT_ACCENT,
            activebackground=Colors.SURFACE_HOVER,
            activeforeground=Colors.PRIMARY_HOVER,
            relief=tk.SOLID,
            borderwidth=1,
            cursor="hand2",
            padx=size_config["pad_x"],
            pady=size_config["pad_y"],
            width=10,  # Set minimum width in characters
            height=2   # Set minimum height in text lines
        )
        
        def on_enter(e):
            btn.config(bg=Colors.SURFACE_HOVER, fg=Colors.PRIMARY_HOVER, relief=tk.SOLID)
        
        def on_leave(e):
            btn.config(bg=Colors.BG_CARD, fg=Colors.TEXT_ACCENT, relief=tk.SOLID)
            
        btn.bind("<Enter>", on_enter)
        btn.bind("<Leave>", on_leave)
        btn.pack(fill=tk.BOTH, expand=True)
        
        # Use a custom attribute container to avoid Pylance warnings
        setattr(btn_frame, '_button', btn)
        return btn_frame
    
    @staticmethod
    def create_success(parent, text: str, command: Optional[Callable[[], Any]] = None, icon: Optional[str] = None, size: str = "md"):
        """Creates a success button with success styling."""
        btn_frame = tk.Frame(parent, bg=Colors.BG_PRIMARY)
        
        sizes = {
            "sm": {"font_size": Typography.SM, "pad_x": 12, "pad_y": 6},
            "md": {"font_size": Typography.MD, "pad_x": 16, "pad_y": 8},
            "lg": {"font_size": Typography.LG, "pad_x": 20, "pad_y": 10}
        }
        
        size_config = sizes.get(size, sizes["md"])
        button_text = f"{icon} {text}" if icon else text
        
        btn = tk.Button(
            btn_frame,
            text=button_text,
            command=command if command is not None else lambda: None,
            font=(FONT_FAMILY_PRIMARY, size_config["font_size"], Typography.MEDIUM),
            bg=Colors.SUCCESS,
            fg=Colors.TEXT_INVERSE,
            activebackground=Colors.SUCCESS_HOVER,
            activeforeground=Colors.TEXT_INVERSE,
            relief=tk.FLAT,
            borderwidth=0,
            cursor="hand2",
            padx=size_config["pad_x"],
            pady=size_config["pad_y"]
        )
        
        def on_enter(e):
            btn.config(bg=Colors.SUCCESS_HOVER)
        
        def on_leave(e):
            btn.config(bg=Colors.SUCCESS)
            
        btn.bind("<Enter>", on_enter)
        btn.bind("<Leave>", on_leave)
        btn.pack(fill=tk.BOTH, expand=True)
        
        setattr(btn_frame, '_button', btn)
        return btn_frame
    
    @staticmethod
    def create_danger(parent, text: str, command: Optional[Callable[[], Any]] = None, icon: Optional[str] = None, size: str = "md"):
        """Creates a danger button with error styling."""
        btn_frame = tk.Frame(parent, bg=Colors.BG_PRIMARY)
        
        sizes = {
            "sm": {"font_size": Typography.SM, "pad_x": 12, "pad_y": 6},
            "md": {"font_size": Typography.MD, "pad_x": 16, "pad_y": 8},
            "lg": {"font_size": Typography.LG, "pad_x": 20, "pad_y": 10}
        }
        
        size_config = sizes.get(size, sizes["md"])
        button_text = f"{icon} {text}" if icon else text
        
        btn = tk.Button(
            btn_frame,
            text=button_text,
            command=command if command is not None else lambda: None,
            font=(FONT_FAMILY_PRIMARY, size_config["font_size"], Typography.MEDIUM),
            bg=Colors.ERROR,
            fg=Colors.TEXT_INVERSE,
            activebackground=Colors.ERROR_HOVER,
            activeforeground=Colors.TEXT_INVERSE,
            relief=tk.FLAT,
            borderwidth=0,
            cursor="hand2",
            padx=size_config["pad_x"],
            pady=size_config["pad_y"]
        )
        
        def on_enter(e):
            btn.config(bg=Colors.ERROR_HOVER)
        
        def on_leave(e):
            btn.config(bg=Colors.ERROR)
            
        btn.bind("<Enter>", on_enter)
        btn.bind("<Leave>", on_leave)
        btn.pack(fill=tk.BOTH, expand=True)
        
        setattr(btn_frame, '_button', btn)
        return btn_frame
    
    @staticmethod
    def create_warning(parent, text: str, command: Optional[Callable[[], Any]] = None, icon: Optional[str] = None, size: str = "md"):
        """Creates a warning button with warning styling."""
        btn_frame = tk.Frame(parent, bg=Colors.BG_PRIMARY)
        
        sizes = {
            "sm": {"font_size": Typography.SM, "pad_x": 12, "pad_y": 6},
            "md": {"font_size": Typography.MD, "pad_x": 16, "pad_y": 8},
            "lg": {"font_size": Typography.LG, "pad_x": 20, "pad_y": 10}
        }
        
        size_config = sizes.get(size, sizes["md"])
        button_text = f"{icon} {text}" if icon else text
        
        btn = tk.Button(
            btn_frame,
            text=button_text,
            command=command if command is not None else lambda: None,
            font=(FONT_FAMILY_PRIMARY, size_config["font_size"], Typography.MEDIUM),
            bg=Colors.WARNING,
            fg=Colors.TEXT_INVERSE,
            activebackground=Colors.WARNING_HOVER,
            activeforeground=Colors.TEXT_INVERSE,
            relief=tk.FLAT,
            borderwidth=0,
            cursor="hand2",
            padx=size_config["pad_x"],
            pady=size_config["pad_y"]
        )
        
        def on_enter(e):
            btn.config(bg=Colors.WARNING_HOVER)
        
        def on_leave(e):
            btn.config(bg=Colors.WARNING)
            
        btn.bind("<Enter>", on_enter)
        btn.bind("<Leave>", on_leave)
        btn.pack(fill=tk.BOTH, expand=True)
        
        setattr(btn_frame, '_button', btn)
        return btn_frame

class PremiumCard:
    """Card component with elevated styling and shadows."""
    
    @staticmethod
    def create(parent, title=None, padding=Spacing.CARD_PADDING):
        """Creates a premium card container."""
        # Outer frame for shadow effect (simulation)
        shadow_frame = tk.Frame(parent, bg=Colors.SHADOW_MD, height=2)
        shadow_frame.pack(fill=tk.X, padx=(2, 0), pady=(2, 0))
        
        # Main card frame
        card_frame = tk.Frame(
            parent,
            bg=Colors.BG_CARD,
            relief=tk.FLAT,
            borderwidth=1
        )
        card_frame.pack(fill=tk.BOTH, expand=True, padx=(0, 2), pady=(0, 2))
        
        # Inner content frame with padding
        content_frame = tk.Frame(card_frame, bg=Colors.BG_CARD)
        content_frame.pack(fill=tk.BOTH, expand=True, padx=padding, pady=padding)
        
        # Optional title
        if title:
            title_label = tk.Label(
                content_frame,
                text=title,
                font=(FONT_FAMILY_PRIMARY, Typography.LG, Typography.SEMIBOLD),
                bg=Colors.BG_CARD,
                fg=Colors.TEXT_PRIMARY
            )
            title_label.pack(anchor=tk.W, pady=(0, Spacing.MD))
        
        return content_frame

class PremiumDialog:
    """Enhanced dialog system with modern styling."""
    
    @staticmethod
    def create_base(parent, title, width=400, height=300, resizable=True):
        """Creates a base dialog with premium styling."""
        dialog = tk.Toplevel(parent)
        dialog.title(title)
        dialog.transient(parent)
        dialog.grab_set()
        dialog.resizable(resizable, resizable)  # Allow resizing by default
        dialog.configure(bg=Colors.BG_PRIMARY)
        
        # Set minimum size to prevent content from being cut off
        dialog.minsize(min(width, 350), min(height, 200))
        
        # Center the dialog
        dialog.update_idletasks()
        x = (dialog.winfo_screenwidth() // 2) - (width // 2)
        y = (dialog.winfo_screenheight() // 2) - (height // 2)
        dialog.geometry(f"{width}x{height}+{x}+{y}")
        
        # Ensure dialog appears on top and gets focus (Windows Z-order fix)
        dialog.update()
        dialog.lift()
        dialog.focus_force()
        dialog.attributes('-topmost', True)  # Temporarily make topmost
        dialog.after(100, lambda: dialog.attributes('-topmost', False))  # Remove topmost after showing
        
        # Main container with padding
        main_frame = tk.Frame(dialog, bg=Colors.BG_PRIMARY)
        main_frame.pack(fill=tk.BOTH, expand=True, padx=Spacing.XL, pady=Spacing.XL)
        
        return dialog, main_frame

# --- Premium Styling System ---
def setup_premium_styles():
    """Configures advanced TTK styles for premium appearance."""
    style = ttk.Style()
    
    # Select the best available theme
    available_themes = style.theme_names()
    preferred_themes = ['vista', 'clam', 'alt', 'default']
    
    selected_theme = None
    for theme in preferred_themes:
        if theme in available_themes:
            selected_theme = theme
            break
    
    if selected_theme:
        try:
            style.theme_use(selected_theme)
        except tk.TclError:
            style.theme_use('default')
    
    # Premium Frame Styling
    style.configure(
        "Premium.TFrame",
        background=Colors.BG_CARD,
        relief="flat",
        borderwidth=0
    )
    
    style.configure(
        "Card.TFrame", 
        background=Colors.BG_CARD,
        relief="solid",
        borderwidth=1,
        lightcolor=Colors.BORDER_SUBTLE,
        darkcolor=Colors.BORDER_SUBTLE
    )
    
    # Premium Label Styling
    style.configure(
        "Premium.TLabel",
        font=(FONT_FAMILY_PRIMARY, Typography.BASE, Typography.NORMAL),
        background=Colors.BG_CARD,
        foreground=Colors.TEXT_PRIMARY
    )
    
    style.configure(
        "Heading.TLabel",
        font=(FONT_FAMILY_PRIMARY, Typography.XL, Typography.SEMIBOLD),
        background=Colors.BG_CARD,
        foreground=Colors.TEXT_PRIMARY
    )
    
    style.configure(
        "Title.TLabel",
        font=(FONT_FAMILY_PRIMARY, Typography.XXL, Typography.BOLD),
        background=Colors.BG_CARD,
        foreground=Colors.TEXT_PRIMARY
    )
    
    style.configure(
        "Subtitle.TLabel",
        font=(FONT_FAMILY_PRIMARY, Typography.LG, Typography.MEDIUM),
        background=Colors.BG_CARD,
        foreground=Colors.TEXT_SECONDARY
    )
    
    style.configure(
        "Caption.TLabel",
        font=(FONT_FAMILY_PRIMARY, Typography.SM, Typography.NORMAL),
        background=Colors.BG_CARD,
        foreground=Colors.TEXT_MUTED
    )
    
    # Premium Button Styling
    style.configure(
        "Premium.TButton",
        font=(FONT_FAMILY_PRIMARY, Typography.MD, Typography.MEDIUM),
        padding=(Spacing.BUTTON_PADDING_X, Spacing.BUTTON_PADDING_Y),
        background=Colors.PRIMARY,
        foreground=Colors.TEXT_INVERSE,
        borderwidth=0,
        focuscolor="none"
    )
    
    style.map(
        "Premium.TButton",
        background=[
            ("active", Colors.PRIMARY_HOVER),
            ("pressed", Colors.PRIMARY_ACTIVE)
        ],
        foreground=[
            ("active", Colors.TEXT_INVERSE),
            ("pressed", Colors.TEXT_INVERSE)
        ]
    )
    
    # Success Button
    style.configure(
        "Success.TButton",
        font=(FONT_FAMILY_PRIMARY, Typography.MD, Typography.MEDIUM),
        padding=(Spacing.BUTTON_PADDING_X, Spacing.BUTTON_PADDING_Y),
        background=Colors.SUCCESS,
        foreground=Colors.TEXT_INVERSE,
        borderwidth=0,
        focuscolor="none"
    )
    
    style.map(
        "Success.TButton",
        background=[("active", Colors.SUCCESS_HOVER)]
    )
    
    # Secondary/Outline Button
    style.configure(
        "Secondary.TButton",
        font=(FONT_FAMILY_PRIMARY, Typography.MD, Typography.MEDIUM),
        padding=(Spacing.BUTTON_PADDING_X, Spacing.BUTTON_PADDING_Y),
        background=Colors.BG_CARD,
        foreground=Colors.TEXT_ACCENT,
        borderwidth=1,
        relief="solid",
        focuscolor="none"
    )
    
    style.map(
        "Secondary.TButton",
        background=[
            ("active", Colors.SURFACE_HOVER),
            ("pressed", Colors.SURFACE_ACTIVE)
        ]
    )
    
    # Danger Button
    style.configure(
        "Danger.TButton",
        font=(FONT_FAMILY_PRIMARY, Typography.MD, Typography.MEDIUM),
        padding=(Spacing.BUTTON_PADDING_X, Spacing.BUTTON_PADDING_Y),
        background=Colors.ERROR,
        foreground=Colors.TEXT_INVERSE,
        borderwidth=0,
        focuscolor="none"
    )
    
    style.map(
        "Danger.TButton",
        background=[("active", Colors.ERROR_HOVER)]
    )
    
    # Premium LabelFrame
    style.configure(
        "Premium.TLabelframe",
        background=Colors.BG_CARD,
        borderwidth=1,
        relief="solid",
        lightcolor=Colors.BORDER_DEFAULT,
        darkcolor=Colors.BORDER_DEFAULT
    )
    
    style.configure(
        "Premium.TLabelframe.Label",
        font=(FONT_FAMILY_PRIMARY, Typography.MD, Typography.SEMIBOLD),
        background=Colors.BG_CARD,
        foreground=Colors.TEXT_PRIMARY
    )
    
    # Premium Progressbar
    style.configure(
        "Premium.Horizontal.TProgressbar",
        background=Colors.PRIMARY,
        troughcolor=Colors.BG_TERTIARY,
        borderwidth=0,
        lightcolor=Colors.PRIMARY,
        darkcolor=Colors.PRIMARY
    )

# =============================================================================
# ENHANCED WINDOW CREATION FUNCTIONS
# =============================================================================

def create_premium_main_window():
    """Creates the main application window with premium styling."""
    root = tk.Tk()
    root.title("Ogresync")
    root.geometry("800x600")
    root.configure(bg=Colors.BG_PRIMARY)
    root.minsize(600, 400)
    
    # Set window icon with enhanced fallback support
    try:
        # Check if we're running from PyInstaller bundle
        if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'):
            # Try different icon files in order of preference
            icon_files = ["new_logo_1.ico", "ogrelix_logo.ico", "new_logo_1.png", "logo.png"]
            icon_path = None
            for icon_file in icon_files:
                test_path = os.path.join(sys._MEIPASS, "assets", icon_file)  # type: ignore
                if os.path.exists(test_path):
                    icon_path = test_path
                    break
        else:
            # Development mode - check local assets folder
            icon_files = ["new_logo_1.ico", "ogrelix_logo.ico", "new_logo_1.png", "logo.png"]
            icon_path = None
            for icon_file in icon_files:
                test_path = os.path.join("assets", icon_file)
                if os.path.exists(test_path):
                    icon_path = test_path
                    break
        
        if icon_path:
            if icon_path.endswith('.ico'):
                # Use iconbitmap for .ico files (works better on Windows)
                root.iconbitmap(icon_path)
            else:
                # Use iconphoto for .png files
                img = tk.PhotoImage(file=icon_path)
                root.iconphoto(True, img)
    except Exception:
        pass  # Fallback to default icon if loading fails
    
    # Initialize fonts and styles
    init_font_config()
    setup_premium_styles()
    
    # Create main container with premium styling
    main_container = tk.Frame(root, bg=Colors.BG_PRIMARY)
    main_container.pack(fill=tk.BOTH, expand=True, padx=Spacing.XL, pady=Spacing.LG)
    
    # Header section with title and subtitle
    header_frame = tk.Frame(main_container, bg=Colors.BG_PRIMARY)
    header_frame.pack(fill=tk.X, pady=(0, Spacing.XL))
    
    # App title
    title_label = tk.Label(
        header_frame,
        text="Ogresync",
        font=(FONT_FAMILY_PRIMARY, Typography.H2, Typography.BOLD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.TEXT_PRIMARY
    )
    title_label.pack(anchor=tk.W)
    
    # Subtitle
    subtitle_label = tk.Label(
        header_frame,
        text="Obsidian Vault Synchronization",
        font=(FONT_FAMILY_PRIMARY, Typography.MD, Typography.NORMAL),
        bg=Colors.BG_PRIMARY,
        fg=Colors.TEXT_SECONDARY
    )
    subtitle_label.pack(anchor=tk.W, pady=(2, 0))
    
    # Main content area using PremiumCard
    content_card = PremiumCard.create(main_container, padding=Spacing.XL)
    
    # Activity log section
    log_header = tk.Label(
        content_card,
        text=f"{Icons.FILE} Activity Log",
        font=(FONT_FAMILY_PRIMARY, Typography.LG, Typography.SEMIBOLD),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_PRIMARY
    )
    log_header.pack(anchor=tk.W, pady=(0, Spacing.SM))
    
    # Log text widget with premium styling
    log_frame = tk.Frame(content_card, bg=Colors.BG_CARD)
    log_frame.pack(fill=tk.BOTH, expand=True, pady=(0, Spacing.LG))
    
    log_text_widget = scrolledtext.ScrolledText(
        log_frame,
        wrap=tk.WORD,
        height=15,
        state='disabled',
        font=(FONT_FAMILY_MONO, Typography.SM),
        bg=Colors.BG_SECONDARY,
        fg=Colors.TEXT_PRIMARY,
        insertbackground=Colors.TEXT_PRIMARY,
        selectbackground=Colors.PRIMARY_LIGHT,
        selectforeground=Colors.TEXT_PRIMARY,
        relief=tk.FLAT,
        borderwidth=1,
        highlightthickness=1,
        highlightcolor=Colors.BORDER_ACCENT,
        highlightbackground=Colors.BORDER_DEFAULT
    )
    log_text_widget.pack(fill=tk.BOTH, expand=True)
    
    # Progress section
    progress_header = tk.Label(
        content_card,
        text=f"{Icons.SYNC} Progress",
        font=(FONT_FAMILY_PRIMARY, Typography.MD, Typography.MEDIUM),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_PRIMARY
    )
    progress_header.pack(anchor=tk.W, pady=(0, Spacing.XS))
    
    # Progress bar with premium styling
    progress_frame = tk.Frame(content_card, bg=Colors.BG_CARD)
    progress_frame.pack(fill=tk.X, pady=(0, Spacing.SM))
    
    progress_bar_widget = ttk.Progressbar(
        progress_frame,
        style="Premium.Horizontal.TProgressbar",
        orient="horizontal",
        length=400,
        mode="determinate"
    )
    progress_bar_widget.pack(fill=tk.X)
    
    return root, log_text_widget, progress_bar_widget

# Enhanced conflict resolution dialog with premium styling
def create_premium_conflict_dialog(parent_window, conflict_files_text):
    """Creates a beautifully styled conflict resolution dialog."""
    dialog, main_frame = PremiumDialog.create_base(
        parent_window, 
        "Merge Conflict Detected", 
        width=500, 
        height=350,
        resizable=False
    )
    
    # Header with icon and title
    header_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    header_frame.pack(fill=tk.X, pady=(0, Spacing.LG))
    
    # Warning icon and title
    title_frame = tk.Frame(header_frame, bg=Colors.BG_PRIMARY)
    title_frame.pack(fill=tk.X)
    
    icon_label = tk.Label(
        title_frame,
        text=Icons.WARNING,
        font=(FONT_FAMILY_PRIMARY, Typography.H3),
        bg=Colors.BG_PRIMARY,
        fg=Colors.WARNING
    )
    icon_label.pack(side=tk.LEFT, padx=(0, Spacing.SM))
    
    title_label = tk.Label(
        title_frame,
        text="Merge Conflict Detected",
        font=(FONT_FAMILY_PRIMARY, Typography.XL, Typography.SEMIBOLD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.TEXT_PRIMARY
    )
    title_label.pack(side=tk.LEFT, anchor=tk.W)
    
    # Content card
    content_card = PremiumCard.create(main_frame, padding=Spacing.LG)
    
    # Message text
    message_text = (
        f"Conflicts were found in the following files:\n\n"
        f"{conflict_files_text}\n\n"
        f"Please choose how you'd like to resolve these conflicts:"
    )
    
    message_label = tk.Label(
        content_card,
        text=message_text,
        font=(FONT_FAMILY_PRIMARY, Typography.MD),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_SECONDARY,
        justify=tk.LEFT,
        wraplength=420
    )
    message_label.pack(pady=(0, Spacing.LG), anchor=tk.W)
    
    # Resolution choice storage
    resolution = {"choice": None}
    
    def set_choice(choice):
        resolution["choice"] = choice
        dialog.destroy()
    
    # Action buttons with premium styling
    button_frame = tk.Frame(content_card, bg=Colors.BG_CARD)
    button_frame.pack(fill=tk.X, pady=(Spacing.MD, 0))
    
    # Create premium buttons
    btn_local = PremiumButton.create_secondary(
        button_frame, 
        "Keep Local", 
        lambda: set_choice("ours"),
        Icons.DOWNLOAD
    )
    btn_local.pack(side=tk.LEFT, fill=tk.X, expand=True, padx=(0, Spacing.SM))
    
    btn_remote = PremiumButton.create_secondary(
        button_frame,
        "Keep Remote", 
        lambda: set_choice("theirs"),
        Icons.UPLOAD
    )
    btn_remote.pack(side=tk.LEFT, fill=tk.X, expand=True, padx=(Spacing.SM, Spacing.SM))
    
    btn_manual = PremiumButton.create_primary(
        button_frame,
        "Merge Manually",
        lambda: set_choice("manual"),
        Icons.GEAR
    )
    btn_manual.pack(side=tk.LEFT, fill=tk.X, expand=True, padx=(Spacing.SM, 0))
    
    # Wait for user choice
    parent_window.wait_window(dialog)
    return resolution["choice"]

# Enhanced minimal UI for auto-sync with premium styling
def create_premium_minimal_ui(auto_run=False):
    """Creates a minimal UI with premium styling for auto-sync mode."""
    root = tk.Tk()
    root.title("Ogresync")
    root.geometry("600x400")
    root.configure(bg=Colors.BG_PRIMARY)
    root.resizable(False, False)
    
    # Set window icon with enhanced fallback support
    try:
        # Check if we're running from PyInstaller bundle
        if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'):
            # Try different icon files in order of preference
            icon_files = ["new_logo_1.ico", "ogrelix_logo.ico", "new_logo_1.png", "logo.png"]
            icon_path = None
            for icon_file in icon_files:
                test_path = os.path.join(sys._MEIPASS, "assets", icon_file)  # type: ignore
                if os.path.exists(test_path):
                    icon_path = test_path
                    break
        else:
            # Development mode - check local assets folder
            icon_files = ["new_logo_1.ico", "ogrelix_logo.ico", "new_logo_1.png", "logo.png"]
            icon_path = None
            for icon_file in icon_files:
                test_path = os.path.join("assets", icon_file)
                if os.path.exists(test_path):
                    icon_path = test_path
                    break
        
        if icon_path:
            if icon_path.endswith('.ico'):
                # Use iconbitmap for .ico files (works better on Windows)
                root.iconbitmap(icon_path)
            else:
                # Use iconphoto for .png files
                img = tk.PhotoImage(file=icon_path)
                root.iconphoto(True, img)
    except Exception:
        pass  # Fallback to default icon if loading fails
    
    # Main content area
    main_frame = tk.Frame(root, bg=Colors.BG_PRIMARY)
    main_frame.pack(fill=tk.BOTH, expand=True, padx=Spacing.LG, pady=Spacing.LG)
    
    # Header
    header_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    header_frame.pack(fill=tk.X, pady=(0, Spacing.LG))
    
    title_label = tk.Label(
        header_frame,
        text="🔄 Ogresync Auto-Sync",
        font=(FONT_FAMILY_PRIMARY, Typography.H1, Typography.BOLD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.PRIMARY
    )
    title_label.pack()
    
    # Progress section
    progress_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    progress_frame.pack(fill=tk.X, pady=(0, Spacing.LG))
    
    progress_label = tk.Label(
        progress_frame,
        text="Sync Progress:",
        font=(FONT_FAMILY_PRIMARY, Typography.SM, Typography.BOLD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.TEXT_PRIMARY
    )
    progress_label.pack(anchor=tk.W)
    
    progress_bar = ttk.Progressbar(
        progress_frame,
        mode='determinate',
        length=400,
        style="Premium.Horizontal.TProgressbar",
        maximum=100
    )
    progress_bar.pack(fill=tk.X, pady=(Spacing.SM, 0))
    
    # Initialize progress bar
    progress_bar["value"] = 0
    
    # Log area
    log_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    log_frame.pack(fill=tk.BOTH, expand=True)
    
    log_label = tk.Label(
        log_frame,
        text="Sync Log:",
        font=(FONT_FAMILY_PRIMARY, Typography.SM, Typography.BOLD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.TEXT_PRIMARY
    )
    log_label.pack(anchor=tk.W)
    
    log_text = scrolledtext.ScrolledText(
        log_frame,
        height=12,
        font=(FONT_FAMILY_MONO, Typography.SM),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_PRIMARY,
        relief=tk.FLAT,
        borderwidth=1,
        highlightthickness=1,
        highlightcolor=Colors.BORDER_ACCENT,
        highlightbackground=Colors.BORDER_DEFAULT,
        insertbackground=Colors.PRIMARY,
        selectbackground=Colors.PRIMARY_LIGHT,
        selectforeground=Colors.TEXT_PRIMARY,
        state='disabled'
    )
    log_text.pack(fill=tk.BOTH, expand=True, pady=(Spacing.SM, 0))
    
    # Add initial message to show UI is working
    log_text.config(state='normal')
    log_text.insert(tk.END, "Ogresync sync interface ready. Initializing sync process...\n")
    log_text.config(state='disabled')
    
    # Force UI update to show initial message
    root.update_idletasks()
    root.update()
    
    # Initialize after() callback tracking for cleanup
    if not hasattr(root, '_after_ids'):
        root._after_ids = []  # type: ignore
    
    # Override after() method to track callbacks for cleanup
    original_after = root.after
    def tracked_after(ms, func, *args):  # type: ignore
        after_id = original_after(ms, func, *args)
        if hasattr(root, '_after_ids'):
            root._after_ids.append(after_id)  # type: ignore
        return after_id
    root.after = tracked_after  # type: ignore
    
    return root, log_text, progress_bar

# Enhanced wizard UI for setup with premium styling
def create_premium_wizard_ui():
    """Creates a premium setup wizard interface."""
    root = tk.Tk()
    root.title("Ogresync Setup")
    root.geometry("700x550")
    root.configure(bg=Colors.BG_PRIMARY)
    root.resizable(True, False)
    root.minsize(600, 500)
    
    # Set window icon with enhanced fallback support
    try:
        # Check if we're running from PyInstaller bundle
        if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'):
            # Try different icon files in order of preference
            icon_files = ["new_logo_1.ico", "ogrelix_logo.ico", "new_logo_1.png", "logo.png"]
            icon_path = None
            for icon_file in icon_files:
                test_path = os.path.join(sys._MEIPASS, "assets", icon_file)  # type: ignore
                if os.path.exists(test_path):
                    icon_path = test_path
                    break
        else:
            # Development mode - check local assets folder
            icon_files = ["new_logo_1.ico", "ogrelix_logo.ico", "new_logo_1.png", "logo.png"]
            icon_path = None
            for icon_file in icon_files:
                test_path = os.path.join("assets", icon_file)
                if os.path.exists(test_path):
                    icon_path = test_path
                    break
        
        if icon_path:
            if icon_path.endswith('.ico'):
                # Use iconbitmap for .ico files (works better on Windows)
                root.iconbitmap(icon_path)
            else:
                # Use iconphoto for .png files
                img = tk.PhotoImage(file=icon_path)
                root.iconphoto(True, img)
    except Exception:
        pass  # Fallback to default icon if loading fails
    
    # Main content area with premium styling
    main_frame = tk.Frame(root, bg=Colors.BG_PRIMARY)
    main_frame.pack(fill=tk.BOTH, expand=True, padx=Spacing.LG, pady=Spacing.LG)
    
    # Header
    header_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    header_frame.pack(fill=tk.X, pady=(0, Spacing.LG))
    
    title_label = tk.Label(
        header_frame,
        text="🔄 Ogresync Setup Wizard",
        font=(FONT_FAMILY_PRIMARY, Typography.H1, Typography.BOLD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.PRIMARY
    )
    title_label.pack()
    
    subtitle_label = tk.Label(
        header_frame,
        text="Set up your Obsidian vault synchronization with GitHub",
        font=(FONT_FAMILY_PRIMARY, Typography.MD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.TEXT_SECONDARY
    )
    subtitle_label.pack(pady=(Spacing.SM, 0))
    
    # Progress section
    progress_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    progress_frame.pack(fill=tk.X, pady=(0, Spacing.LG))
    
    progress_label = tk.Label(
        progress_frame,
        text="Setup Progress:",
        font=(FONT_FAMILY_PRIMARY, Typography.SM, Typography.BOLD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.TEXT_PRIMARY
    )
    progress_label.pack(anchor=tk.W)
    
    progress_bar = ttk.Progressbar(
        progress_frame,
        mode='determinate',
        length=400,
        style="Premium.Horizontal.TProgressbar"
    )
    progress_bar.pack(fill=tk.X, pady=(Spacing.SM, 0))
    
    # Log area
    log_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    log_frame.pack(fill=tk.BOTH, expand=True, pady=(0, Spacing.LG))
    
    log_label = tk.Label(
        log_frame,
        text="Setup Log:",
        font=(FONT_FAMILY_PRIMARY, Typography.SM, Typography.BOLD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.TEXT_PRIMARY
    )
    log_label.pack(anchor=tk.W)
    
    log_text = scrolledtext.ScrolledText(
        log_frame,
        height=12,
        font=(FONT_FAMILY_MONO, Typography.SM),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_PRIMARY,
        relief=tk.FLAT,
        borderwidth=1,
        highlightthickness=1,
        highlightcolor=Colors.BORDER_ACCENT,
        highlightbackground=Colors.BORDER_DEFAULT,
        insertbackground=Colors.PRIMARY,
        selectbackground=Colors.PRIMARY_LIGHT,
        selectforeground=Colors.TEXT_PRIMARY,
        state='disabled'
    )
    log_text.pack(fill=tk.BOTH, expand=True, pady=(Spacing.SM, 0))
    
    # Control buttons
    button_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    button_frame.pack(fill=tk.X)
    
    # SSH key generation button
    gen_btn = PremiumButton.create_secondary(
        button_frame,
        "Generate SSH Key",
        None,  # Command will be set later
        icon=Icons.SECURITY,
        size="md"
    )
    gen_btn.pack(side=tk.LEFT, padx=(0, Spacing.SM))
    
    # Copy SSH key button
    copy_btn = PremiumButton.create_secondary(
        button_frame,
        "Copy SSH Key",
        None,  # Command will be set later
        icon=Icons.COPY,
        size="md"
    )
    copy_btn.pack(side=tk.LEFT, padx=(0, Spacing.SM))
    
    # Test SSH button
    test_ssh_btn = PremiumButton.create_primary(
        button_frame,
        "Test SSH Connection",
        None,  # Command will be set later
        icon=Icons.SUCCESS,
        size="md"
    )
    test_ssh_btn.pack(side=tk.LEFT, padx=(0, Spacing.SM))
    
    # Exit button
    exit_btn = PremiumButton.create_danger(
        button_frame,
        "Exit",
        None,  # Command will be set later
        size="md"
    )
    exit_btn.pack(side=tk.RIGHT)
    
    return root, log_text, progress_bar, gen_btn, copy_btn, test_ssh_btn, exit_btn

# =============================================================================
# STAGE 1 & STAGE 2 CONFLICT RESOLUTION DIALOGS  
# =============================================================================

def create_stage1_strategy_dialog(parent_window, remote_ahead_analysis, conflict_analysis, title="Repository Conflict"):
    """
    Stage 1: High-level strategy selection dialog.
    
    Args:
        parent_window: Parent window for the dialog
        remote_ahead_analysis: Analysis of remote changes
        conflict_analysis: Analysis of conflicted files  
        title: Dialog title
        
    Returns:
        String indicating chosen strategy: "keep_local", "keep_remote", "smart_merge", or "cancelled"
    """
    dialog = tk.Toplevel(parent_window)
    dialog.title(title)
    dialog.transient(parent_window)
    dialog.grab_set()
    dialog.resizable(False, False)
    dialog.configure(bg=Colors.BG_PRIMARY)
    
    # Center the dialog
    dialog.update_idletasks()
    width, height = 700, 550
    x = (dialog.winfo_screenwidth() // 2) - (width // 2)
    y = (dialog.winfo_screenheight() // 2) - (height // 2)
    dialog.geometry(f"{width}x{height}+{x}+{y}")
    
    # Main container
    main_frame = tk.Frame(dialog, bg=Colors.BG_PRIMARY)
    main_frame.pack(fill=tk.BOTH, expand=True, padx=20, pady=20)
    
    # Header
    header_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    header_frame.pack(fill=tk.X, pady=(0, 20))
    
    title_label = tk.Label(
        header_frame,
        text=f"⚠️ {title}",
        font=(FONT_FAMILY_PRIMARY, Typography.XL, Typography.BOLD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.WARNING
    )
    title_label.pack()
    
    # Analysis summary card
    summary_card = PremiumCard.create(main_frame, padding=Spacing.LG)
    
    # Conflict summary
    total_conflicts = conflict_analysis.get("total_conflicts", 0)
    auto_mergeable = conflict_analysis.get("auto_mergeable", 0)
    manual_needed = conflict_analysis.get("needs_manual_resolution", 0)
    remote_commits = remote_ahead_analysis.get("commits_ahead", 0)
    
    summary_text = f"Conflict Summary:\n\n"
    if remote_commits > 0:
        summary_text += f"• Remote repository is {remote_commits} commit(s) ahead\n"
    summary_text += f"• {total_conflicts} files have conflicts\n"
    if auto_mergeable > 0:
        summary_text += f"• {auto_mergeable} files can be auto-merged\n"
    if manual_needed > 0:
        summary_text += f"• {manual_needed} files need manual resolution\n"
    
    summary_label = tk.Label(
        summary_card,
        text=summary_text,
        font=(FONT_FAMILY_PRIMARY, Typography.MD),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_SECONDARY,
        justify=tk.LEFT,
        wraplength=600
    )
    summary_label.pack(pady=(0, Spacing.LG), anchor=tk.W)
    
    # Strategy selection
    strategy_title = tk.Label(
        summary_card,
        text="Choose your conflict resolution strategy:",
        font=(FONT_FAMILY_PRIMARY, Typography.LG, Typography.BOLD),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_PRIMARY
    )
    strategy_title.pack(anchor=tk.W, pady=(0, Spacing.MD))
    
    # Choice storage
    choice = {"value": "cancelled"}
    
    def set_choice(selected_choice):
        choice["value"] = selected_choice
        dialog.destroy()
    
    # Strategy buttons
    button_frame = tk.Frame(summary_card, bg=Colors.BG_CARD)
    button_frame.pack(fill=tk.X, pady=(Spacing.MD, 0))
    
    # Smart Merge (Recommended)
    smart_btn = PremiumButton.create_primary(
        button_frame,
        "🔄 Smart Merge (Recommended)",
        lambda: set_choice("smart_merge")
    )
    smart_btn.pack(fill=tk.X, pady=(0, Spacing.SM))
    
    smart_desc = tk.Label(
        button_frame,
        text="Automatically merges non-conflicting files and prompts for manual resolution of conflicts",
        font=(FONT_FAMILY_PRIMARY, Typography.SM),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_MUTED,
        wraplength=600,
        justify=tk.LEFT
    )
    smart_desc.pack(anchor=tk.W, pady=(2, Spacing.MD))
    
    # Keep All Local
    local_btn = PremiumButton.create_secondary(
        button_frame,
        "📁 Keep All Local Changes",
        lambda: set_choice("keep_local")
    )
    local_btn.pack(fill=tk.X, pady=(0, Spacing.SM))
    
    local_desc = tk.Label(
        button_frame,
        text="Keeps all your local changes and overwrites remote changes",
        font=(FONT_FAMILY_PRIMARY, Typography.SM),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_MUTED,
        wraplength=600,
        justify=tk.LEFT
    )
    local_desc.pack(anchor=tk.W, pady=(2, Spacing.MD))
    
    # Keep All Remote
    remote_btn = PremiumButton.create_secondary(
        button_frame,
        "🌐 Keep All Remote Changes",
        lambda: set_choice("keep_remote")
    )
    remote_btn.pack(fill=tk.X, pady=(0, Spacing.SM))
    
    remote_desc = tk.Label(
        button_frame,
        text="Downloads all remote changes and overwrites your local changes",
        font=(FONT_FAMILY_PRIMARY, Typography.SM),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_MUTED,
        wraplength=600,
        justify=tk.LEFT
    )
    remote_desc.pack(anchor=tk.W, pady=(2, Spacing.MD))
    
    # Cancel button
    cancel_btn = tk.Button(
        button_frame,
        text="Cancel",
        command=lambda: set_choice("cancelled"),
        font=(FONT_FAMILY_PRIMARY, Typography.SM),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_MUTED,
        relief=tk.FLAT,
        cursor="hand2",
        padx=Spacing.MD,
        pady=Spacing.XS
    )
    cancel_btn.pack(pady=(Spacing.SM, 0))
    
    # Wait for user choice
    parent_window.wait_window(dialog)
    return choice["value"]


def create_stage2_smart_merge_dialog(parent_window, remote_ahead_analysis, conflict_analysis):
    """
    Stage 2: File-by-file smart merge resolution dialog.
    
    Args:
        parent_window: Parent window for the dialog
        remote_ahead_analysis: Analysis of remote changes
        conflict_analysis: Analysis of conflicted files
        
    Returns:
        Dictionary with file resolutions or "cancelled"
    """
    dialog = tk.Toplevel(parent_window)
    dialog.title("Smart Merge - File Resolution")
    dialog.transient(parent_window)
    dialog.grab_set()
    dialog.resizable(True, True)
    dialog.configure(bg=Colors.BG_PRIMARY)
    
    # Set dialog size
    dialog.update_idletasks()
    width, height = 900, 700
    x = (dialog.winfo_screenwidth() // 2) - (width // 2)
    y = (dialog.winfo_screenheight() // 2) - (height // 2)
    dialog.geometry(f"{width}x{height}+{x}+{y}")
    dialog.minsize(800, 600)
    
    # Main container
    main_frame = tk.Frame(dialog, bg=Colors.BG_PRIMARY)
    main_frame.pack(fill=tk.BOTH, expand=True, padx=15, pady=15)
    
    # Header
    header_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    header_frame.pack(fill=tk.X, pady=(0, 15))
    
    title_label = tk.Label(
        header_frame,
        text="🔄 Smart Merge - File by File Resolution",
        font=(FONT_FAMILY_PRIMARY, Typography.XL, Typography.BOLD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.PRIMARY
    )
    title_label.pack(side=tk.LEFT)
    
    # Progress info
    conflicted_files = conflict_analysis.get("conflicted_files", {})
    total_files = len(conflicted_files)
    
    progress_label = tk.Label(
        header_frame,
        text=f"0 of {total_files} resolved",
        font=(FONT_FAMILY_PRIMARY, Typography.MD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.TEXT_SECONDARY
    )
    progress_label.pack(side=tk.RIGHT)
    
    # File list and resolution area
    content_frame = tk.Frame(main_frame, bg=Colors.BG_CARD, relief=tk.RAISED, borderwidth=1)
    content_frame.pack(fill=tk.BOTH, expand=True, pady=(0, 15))
    
    # Scrollable file list
    list_frame = tk.Frame(content_frame, bg=Colors.BG_CARD)
    list_frame.pack(fill=tk.BOTH, expand=True, padx=15, pady=15)
    
    # File list header
    list_header = tk.Label(
        list_frame,
        text="Conflicted Files - Choose resolution for each:",
        font=(FONT_FAMILY_PRIMARY, Typography.MD, Typography.BOLD),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_PRIMARY
    )
    list_header.pack(anchor=tk.W, pady=(0, 10))
    
    # Scrollable frame for files
    canvas = tk.Canvas(list_frame, bg=Colors.BG_CARD, highlightthickness=0)
    scrollbar = tk.Scrollbar(list_frame, orient="vertical", command=canvas.yview)
    scrollable_frame = tk.Frame(canvas, bg=Colors.BG_CARD)
    
    scrollable_frame.bind(
        "<Configure>",
        lambda e: canvas.configure(scrollregion=canvas.bbox("all"))
    )
    
    canvas.create_window((0, 0), window=scrollable_frame, anchor="nw")
    canvas.configure(yscrollcommand=scrollbar.set)
    
    # Resolution storage
    resolutions = {}
    resolved_count = {"value": 0}
    
    def update_progress():
        progress_label.config(text=f"{resolved_count['value']} of {total_files} resolved")
    
    def create_file_resolution_widget(filename, file_info):
        """Create resolution widget for a single file."""
        file_frame = tk.Frame(scrollable_frame, bg=Colors.BG_SECONDARY, relief=tk.RAISED, borderwidth=1)
        file_frame.pack(fill=tk.X, pady=(0, 5), padx=5)
        
        inner_frame = tk.Frame(file_frame, bg=Colors.BG_SECONDARY)
        inner_frame.pack(fill=tk.X, padx=10, pady=8)
        
        # File info
        file_label = tk.Label(
            inner_frame,
            text=f"📄 {filename}",
            font=(FONT_FAMILY_PRIMARY, Typography.MD, Typography.BOLD),
            bg=Colors.BG_SECONDARY,
            fg=Colors.TEXT_PRIMARY
        )
        file_label.pack(anchor=tk.W)
        
        desc_label = tk.Label(
            inner_frame,
            text=file_info.get("description", "File has conflicts"),
            font=(FONT_FAMILY_PRIMARY, Typography.SM),
            bg=Colors.BG_SECONDARY,
            fg=Colors.TEXT_SECONDARY,
            wraplength=700,
            justify=tk.LEFT
        )
        desc_label.pack(anchor=tk.W, pady=(2, 8))
        
        # Resolution buttons
        btn_frame = tk.Frame(inner_frame, bg=Colors.BG_SECONDARY)
        btn_frame.pack(fill=tk.X)
        
        def set_resolution(resolution):
            resolutions[filename] = resolution
            if filename not in [f for f, _ in resolutions.items() if resolutions[f] is not None]:
                resolved_count["value"] += 1
            update_progress()
            # Visual feedback
            for widget in btn_frame.winfo_children():
                if isinstance(widget, tk.Button):
                    widget.config(bg=Colors.BG_CARD, fg=Colors.TEXT_SECONDARY)
            
            # Highlight selected button
            if resolution == "auto_merge" and file_info.get("auto_mergeable", False):
                auto_btn.config(bg=Colors.SUCCESS_HOVER, fg=Colors.TEXT_INVERSE)
            elif resolution == "manual_merge":
                manual_btn.config(bg=Colors.WARNING_HOVER, fg=Colors.TEXT_INVERSE)
            elif resolution == "use_local":
                local_btn.config(bg=Colors.PRIMARY_HOVER, fg=Colors.TEXT_INVERSE)
            elif resolution == "use_remote":
                remote_btn.config(bg=Colors.PRIMARY_HOVER, fg=Colors.TEXT_INVERSE)

        # Auto merge button (if available)
        if file_info.get("auto_mergeable", False):
            auto_btn = tk.Button(
                btn_frame,
                text="🔄 Auto Merge",
                command=lambda: set_resolution("auto_merge"),
                font=(FONT_FAMILY_PRIMARY, Typography.SM),
                bg=Colors.SUCCESS,
                fg=Colors.TEXT_INVERSE,
                relief=tk.FLAT,
                cursor="hand2",
                padx=12,
                pady=4
            )
            auto_btn.pack(side=tk.LEFT, padx=(0, 5))
        
        # Manual merge button
        manual_btn = tk.Button(
            btn_frame,
            text="✏️ Manual Merge",
            command=lambda: set_resolution("manual_merge"),
            font=(FONT_FAMILY_PRIMARY, Typography.SM),
            bg=Colors.WARNING,
            fg=Colors.TEXT_INVERSE,
            relief=tk.FLAT,
            cursor="hand2",
            padx=12,
            pady=4
        )
        manual_btn.pack(side=tk.LEFT, padx=(0, 5))
        
        # Use local button
        local_btn = tk.Button(
            btn_frame,
            text="📁 Use Local",
            command=lambda: set_resolution("use_local"),
            font=(FONT_FAMILY_PRIMARY, Typography.SM),
            bg=Colors.BG_CARD,
            fg=Colors.PRIMARY,
            relief=tk.SOLID,
            borderwidth=1,
            cursor="hand2",
            padx=12,
            pady=4
        )
        local_btn.pack(side=tk.LEFT, padx=(0, 5))
        
        # Use remote button  
        remote_btn = tk.Button(
            btn_frame,
            text="🌐 Use Remote",
            command=lambda: set_resolution("use_remote"),
            font=(FONT_FAMILY_PRIMARY, Typography.SM),
            bg=Colors.BG_CARD,
            fg=Colors.PRIMARY,
            relief=tk.SOLID,
            borderwidth=1,
            cursor="hand2",
            padx=12,
            pady=4
        )
        remote_btn.pack(side=tk.LEFT)
    
    # Create widgets for each conflicted file
    for filename, file_info in conflicted_files.items():
        create_file_resolution_widget(filename, file_info)
    
    # Pack scrollable components
    canvas.pack(side="left", fill="both", expand=True)
    scrollbar.pack(side="right", fill="y")
    
    # Bottom action buttons
    action_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    action_frame.pack(fill=tk.X)
    
    result: Dict[str, Any] = {"value": "cancelled"}
    
    def apply_resolutions():
        # Check if all files are resolved
        unresolved = [f for f in conflicted_files.keys() if f not in resolutions]
        if unresolved:
            import tkinter.messagebox as messagebox
            response = messagebox.askyesno(
                "Unresolved Files",
                f"Some files are not yet resolved:\n{', '.join(unresolved[:3])}\n\n"
                "Continue anyway? (Unresolved files will use auto-merge if possible)"
            )
            if not response:
                return
            # Auto-resolve remaining files
            for filename in unresolved:
                file_info = conflicted_files[filename]
                if file_info.get("auto_mergeable", False):
                    resolutions[filename] = "auto_merge"
                else:
                    resolutions[filename] = "manual_merge"
        
        result["value"] = dict(resolutions)  # Convert to dict to fix type issue
        dialog.destroy()
    
    def cancel_resolution():
        result["value"] = "cancelled"
        dialog.destroy()
    
    # Action buttons
    apply_btn = PremiumButton.create_success(
        action_frame,
        "Apply Resolutions",
        apply_resolutions
    )
    apply_btn.pack(side=tk.RIGHT, padx=(10, 0))
    
    cancel_btn = PremiumButton.create_secondary(
        action_frame,
        "Cancel",
        cancel_resolution
    )
    cancel_btn.pack(side=tk.RIGHT)
    
    # Wait for user to complete
    parent_window.wait_window(dialog)
    return result["value"]


# =============================================================================
# PROGRESSIVE SETUP WIZARD - COMPATIBILITY FUNCTIONS
# =============================================================================

def create_progressive_setup_wizard(parent=None):
    """
    Compatibility wrapper for the new setup wizard module.
    This function maintains backward compatibility while using the new modular setup wizard.
    """
    try:
        import setup_wizard
        return setup_wizard.create_progressive_setup_wizard(parent)
    except ImportError:
        # Fallback if setup_wizard module is not available
        show_premium_error(
            "Setup Wizard Error", 
            "Setup wizard module not available. Please ensure all required files are present.",
            parent
        )
        return None, {"setup_complete": False}

# =============================================================================
# ENHANCED DIALOG SYSTEM
# =============================================================================

def show_premium_info(title, message, parent=None):
    """Show an info message with premium styling."""
    # Calculate required size based on message length
    message_length = len(message)
    if message_length > 500:
        width, height = 650, 450
    elif message_length > 300:
        width, height = 550, 350
    elif message_length > 150:
        width, height = 450, 250
    else:
        width, height = 400, 200
    
    dialog, main_frame = PremiumDialog.create_base(parent, title, width=width, height=height, resizable=True)
    
    # Icon and message
    content_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    content_frame.pack(fill=tk.BOTH, expand=True, pady=(0, Spacing.LG))
    
    icon_label = tk.Label(
        content_frame,
        text=Icons.INFO,
        font=(FONT_FAMILY_PRIMARY, Typography.XXL),
        bg=Colors.BG_PRIMARY,
        fg=Colors.INFO
    )
    icon_label.pack(pady=(0, Spacing.SM))
    
    message_label = tk.Label(
        content_frame,
        text=message,
        font=(FONT_FAMILY_PRIMARY, Typography.MD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.TEXT_PRIMARY,
        wraplength=width - 80,  # Dynamic wrap length based on dialog width
        justify=tk.LEFT  # Changed to LEFT for better readability of long messages
    )
    message_label.pack(pady=(0, Spacing.XL))
    
    # OK button
    button_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    button_frame.pack()
    
    ok_btn = PremiumButton.create_primary(
        button_frame,
        "OK",
        dialog.destroy,
        size="md"
    )
    ok_btn.pack()
    
    parent.wait_window(dialog) if parent else dialog.mainloop()

def show_premium_error(title, message, parent=None):
    """Show an error message with premium styling."""
    dialog, main_frame = PremiumDialog.create_base(parent, title, width=400, height=200, resizable=False)
    
    # Icon and message
    content_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    content_frame.pack(fill=tk.BOTH, expand=True, pady=(0, Spacing.LG))
    
    icon_label = tk.Label(
        content_frame,
        text=Icons.ERROR,
        font=(FONT_FAMILY_PRIMARY, Typography.XXL),
        bg=Colors.BG_PRIMARY,
        fg=Colors.ERROR
    )
    icon_label.pack(pady=(0, Spacing.SM))
    
    message_label = tk.Label(
        content_frame,
        text=message,
        font=(FONT_FAMILY_PRIMARY, Typography.MD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.TEXT_PRIMARY,
        wraplength=350,
        justify=tk.CENTER
    )
    message_label.pack(pady=(0, Spacing.XL))
    
    # OK button
    button_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    button_frame.pack()
    
    ok_btn = PremiumButton.create_danger(
        button_frame,
        "OK",
        dialog.destroy,
        size="md"
    )
    ok_btn.pack()
    
    parent.wait_window(dialog) if parent else dialog.mainloop()

def ask_premium_yes_no(title, message, parent=None):
    """Ask a yes/no question with premium styling."""
    dialog, main_frame = PremiumDialog.create_base(parent, title, width=450, height=250, resizable=False)
    
    # Content frame
    content_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    content_frame.pack(fill=tk.BOTH, expand=True, pady=(0, Spacing.LG))
    
    # Icon
    icon_label = tk.Label(
        content_frame,
        text=Icons.WARNING,
        font=(FONT_FAMILY_PRIMARY, Typography.XXL),
        bg=Colors.BG_PRIMARY,
        fg=Colors.WARNING
    )
    icon_label.pack(pady=(0, Spacing.SM))
    
    # Message
    message_label = tk.Label(
        content_frame,
        text=message,
        font=(FONT_FAMILY_PRIMARY, Typography.MD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.TEXT_PRIMARY,
        wraplength=400,
        justify=tk.CENTER
    )
    message_label.pack(pady=(0, Spacing.XL))
    
    # Button container with fixed dimensions
    button_container = tk.Frame(content_frame, bg=Colors.BG_PRIMARY, height=60)
    button_container.pack(fill=tk.X, side=tk.BOTTOM)
    button_container.pack_propagate(False)
    
    # Center the button area within the container
    button_frame = tk.Frame(button_container, bg=Colors.BG_PRIMARY)
    button_frame.pack(expand=True)
    
    result: Dict[str, Optional[bool]] = {"value": None}
    
    def yes_clicked():
        result["value"] = True
        dialog.destroy()
    
    def no_clicked():
        result["value"] = False
        dialog.destroy()
    
    # Create buttons with guaranteed minimum size and explicit dimensions
    yes_btn = PremiumButton.create_primary(
        button_frame,
        "Yes",
        yes_clicked,
        size="md"
    )
    yes_btn.pack(side=tk.LEFT, padx=(0, Spacing.SM))
    
    no_btn = PremiumButton.create_secondary(
        button_frame,
        "No", 
        no_clicked,
        size="md"
    )
    no_btn.pack(side=tk.LEFT)
    
    # Wait for response
    parent.wait_window(dialog) if parent else dialog.mainloop()
    return result["value"]

def ask_premium_string(title, prompt, initial_value="", parent=None, icon=None):
    """Ask for string input with premium styling."""
    # Calculate dialog size based on prompt length
    prompt_length = len(prompt)
    lines_count = prompt.count('\n') + 1
    
    # Dynamic sizing based on content
    if prompt_length > 300 or lines_count > 6:
        # Large dialog for complex prompts (like GitHub Repository URL dialog)
        width, height = 650, 450
        wrap_length = 580
    elif prompt_length > 150 or lines_count > 3:
        # Medium dialog for moderate prompts
        width, height = 500, 350
        wrap_length = 450
    else:
        # Standard size for simple prompts
        width, height = 400, 250
        wrap_length = 350
    
    dialog, main_frame = PremiumDialog.create_base(parent, title, width=width, height=height)
    
    # Content frame
    content_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    content_frame.pack(fill=tk.BOTH, expand=True, pady=(0, Spacing.LG))
    
    # Optional icon
    if icon:
        icon_label = tk.Label(
            content_frame,
            text=icon,
            font=(FONT_FAMILY_PRIMARY, Typography.XL),
            bg=Colors.BG_PRIMARY,
            fg=Colors.PRIMARY
        )
        icon_label.pack(pady=(0, Spacing.SM))
    
    # Prompt with proper wrapping and left alignment for long text
    prompt_justify = tk.LEFT if lines_count > 2 else tk.CENTER
    prompt_label = tk.Label(
        content_frame,
        text=prompt,
        font=(FONT_FAMILY_PRIMARY, Typography.MD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.TEXT_PRIMARY,
        wraplength=wrap_length,
        justify=prompt_justify
    )
    prompt_label.pack(pady=(0, Spacing.MD))
    
    # Entry with dynamic width based on dialog size
    entry_width = 50 if width > 500 else 40 if width > 400 else 30
    entry_var = tk.StringVar(value=initial_value)
    entry = tk.Entry(
        content_frame,
        textvariable=entry_var,
        font=(FONT_FAMILY_PRIMARY, Typography.MD),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_PRIMARY,
        relief=tk.SOLID,
        borderwidth=1,
        highlightthickness=2,
        highlightcolor=Colors.PRIMARY,
        highlightbackground=Colors.BORDER_DEFAULT,
        width=entry_width
    )
    entry.pack(pady=(0, Spacing.LG))
    entry.focus_set()
    entry.select_range(0, tk.END)
    
    # Button frame
    button_frame = tk.Frame(content_frame, bg=Colors.BG_PRIMARY)
    button_frame.pack(pady=(Spacing.MD, 0))
    
    result: Dict[str, Optional[str]] = {"value": None}
    
    def submit():
        result["value"] = entry_var.get().strip()
        dialog.destroy()
    
    def cancel():
        result["value"] = None
        dialog.destroy()
    
    # Bind Enter key to submit
    def on_enter(event):
        submit()
    
    dialog.bind('<Return>', on_enter)
    dialog.bind('<KP_Enter>', on_enter)
    
    # Create buttons
    ok_btn = PremiumButton.create_primary(
        button_frame,
        "OK",
        submit,
        icon=Icons.SUCCESS,
        size="md"
    )
    ok_btn.pack(side=tk.LEFT, padx=(0, Spacing.SM))
    
    cancel_btn = PremiumButton.create_secondary(
        button_frame,
        "Cancel",
        cancel,
        size="md"
    )
    cancel_btn.pack(side=tk.LEFT)
    
    # Wait for response
    parent.wait_window(dialog) if parent else dialog.mainloop()
    return result["value"]

# =============================================================================
# BACKWARD COMPATIBILITY WRAPPERS
# =============================================================================

# Legacy function wrappers for backward compatibility
def create_main_window():
    """Legacy wrapper for create_premium_main_window."""
    return create_premium_main_window()

def create_conflict_resolution_dialog(parent_window, conflict_files_text):
    """Legacy wrapper for create_premium_conflict_dialog."""
    return create_premium_conflict_dialog(parent_window, conflict_files_text)

def create_minimal_ui(auto_run=False):
    """Legacy wrapper for create_premium_minimal_ui."""
    return create_premium_minimal_ui(auto_run)

def create_wizard_ui():
    """Legacy wrapper for create_premium_wizard_ui."""
    return create_progressive_setup_wizard()

def create_progressive_wizard():
    """Alias for create_progressive_setup_wizard for backward compatibility."""
    return create_progressive_setup_wizard()

def show_info_message(title, message, parent=None):
    """Legacy wrapper for show_premium_info."""
    return show_premium_info(title, message, parent)

def show_error_message(title, message, parent=None):
    """Legacy wrapper for show_premium_error."""
    return show_premium_error(title, message, parent)

def ask_yes_no(title, message, parent=None):
    """Legacy wrapper for ask_premium_yes_no."""
    return ask_premium_yes_no(title, message, parent)

def ask_string_dialog(title, prompt, initial_value="", parent=None, icon=None):
    """Legacy wrapper for ask_premium_string."""
    return ask_premium_string(title, prompt, initial_value, parent, icon)

def ask_file_dialog(title, filetypes=None, parent=None):
    """Open file selection dialog (fallback to tkinter)."""
    try:
        from tkinter import filedialog
        return filedialog.askopenfilename(title=title, filetypes=filetypes or [])
    except Exception:
        return None

def ask_directory_dialog(title, parent=None):
    """Open directory selection dialog (fallback to tkinter)."""
    try:
        from tkinter import filedialog
        return filedialog.askdirectory(title=title)
    except Exception:
        return None

# Enhanced repository conflict resolution dialog
def create_repository_conflict_dialog(parent_window, message, analysis):
    """Creates a dialog for resolving repository content conflicts during setup."""
    # Create the dialog window
    dialog = tk.Toplevel(parent_window)
    dialog.title("Repository Content Conflict")
    dialog.transient(parent_window)
    dialog.grab_set()
    dialog.resizable(True, True)  # Allow resizing
    dialog.configure(bg="#FAFBFC")
    
    # Center the dialog - larger size for better readability
    dialog.update_idletasks()
    width, height = 850, 750  # Increased from 800x700 for better content display
    x = (dialog.winfo_screenwidth() // 2) - (width // 2)
    y = (dialog.winfo_screenheight() // 2) - (height // 2)
    dialog.geometry(f"{width}x{height}+{x}+{y}")
    dialog.minsize(800, 700)  # Increased minimum size
    
    # Force dialog to appear and initialize properly before adding content
    dialog.update()
    dialog.lift()
    dialog.focus_force()
    dialog.after(10, lambda: None)  # Allow time for window to fully initialize
    
    # Main container with increased padding
    main_frame = tk.Frame(dialog, bg="#FAFBFC")
    main_frame.pack(fill=tk.BOTH, expand=True, padx=25, pady=25)  # Increased padding
    
    # Header with icon and title
    header_frame = tk.Frame(main_frame, bg="#FAFBFC")
    header_frame.pack(fill=tk.X, pady=(0, 25))  # Increased spacing
    
    title_label = tk.Label(
        header_frame,
        text="⚠️ Repository Content Conflict",
        font=("Arial", 18, "bold"),  # Slightly larger title
        bg="#FAFBFC",
        fg="#1E293B"
    )
    title_label.pack()
    
    # Content area with increased padding
    content_frame = tk.Frame(main_frame, bg="#FFFFFF", relief=tk.RAISED, borderwidth=1)
    content_frame.pack(fill=tk.BOTH, expand=True, pady=(0, 25))  # Increased spacing
    
    # Padding inside content with increased margins
    inner_frame = tk.Frame(content_frame, bg="#FFFFFF")
    inner_frame.pack(fill=tk.BOTH, expand=True, padx=25, pady=25)  # Increased padding
    
    # Message with increased wrap length
    message_label = tk.Label(
        inner_frame,
        text=message,
        font=("Arial", 13),  # Slightly larger font
        bg="#FFFFFF",
        fg="#475569",
        wraplength=750,  # Increased wrap length
        justify=tk.LEFT
    )
    message_label.pack(pady=(0, 25))  # Increased spacing
    
    # File details with better formatting
    if analysis["local_files"]:
        local_label = tk.Label(
            inner_frame,
            text=f"📁 Local files ({len(analysis['local_files'])} items):",
            font=("Arial", 12, "bold"),  # Slightly larger font
            bg="#FFFFFF",
            fg="#1E293B"
        )
        local_label.pack(anchor=tk.W, pady=(0, 8))  # Increased spacing
        
        # Create scrollable text widget for better file display
        local_frame = tk.Frame(inner_frame, bg="#F8F9FA", relief=tk.SUNKEN, borderwidth=1)
        local_frame.pack(fill=tk.X, pady=(0, 20))  # Increased spacing
        
        local_text = tk.Text(
            local_frame,
            height=min(8, len(analysis["local_files"]) + 1),  # Increased adaptive height
            wrap=tk.WORD,
            font=("Arial", 10),  # Slightly larger font
            bg="#F8F9FA",
            fg="#64748B",
            relief=tk.FLAT,
            borderwidth=0,
            state=tk.DISABLED
        )
        
        # Add scrollbar if needed
        local_scrollbar = tk.Scrollbar(local_frame, orient=tk.VERTICAL, command=local_text.yview)
        local_text.configure(yscrollcommand=local_scrollbar.set)
        
        # Show actual file names with better formatting
        local_files_text = ""
        for i, file in enumerate(analysis["local_files"][:15]):  # Show up to 15 files
            local_files_text += f"• {file}\n"
        if len(analysis["local_files"]) > 15:
            local_files_text += f"• ... and {len(analysis['local_files']) - 15} more files\n"
        
        local_text.config(state=tk.NORMAL)
        local_text.insert(1.0, local_files_text.strip())
        local_text.config(state=tk.DISABLED)
        
        local_text.pack(side=tk.LEFT, fill=tk.BOTH, expand=True, padx=10, pady=6)  # Increased padding
        if len(analysis["local_files"]) > 8:  # Adjusted threshold
            local_scrollbar.pack(side=tk.RIGHT, fill=tk.Y)
    
    if analysis["remote_files"]:
        remote_label = tk.Label(
            inner_frame,
            text=f"🌐 Remote files ({len(analysis['remote_files'])} items):",
            font=("Arial", 12, "bold"),  # Slightly larger font
            bg="#FFFFFF",
            fg="#1E293B"
        )
        remote_label.pack(anchor=tk.W, pady=(0, 8))  # Increased spacing
        
        # Create scrollable text widget for better file display
        remote_frame = tk.Frame(inner_frame, bg="#F8F9FA", relief=tk.SUNKEN, borderwidth=1)
        remote_frame.pack(fill=tk.X, pady=(0, 20))  # Increased spacing
        
        remote_text = tk.Text(
            remote_frame,
            height=min(8, len(analysis["remote_files"]) + 1),  # Increased adaptive height
            wrap=tk.WORD,
            font=("Arial", 10),  # Slightly larger font
            bg="#F8F9FA",
            fg="#64748B",
            relief=tk.FLAT,
            borderwidth=0,
            state=tk.DISABLED
        )
        
        # Add scrollbar if needed
        remote_scrollbar = tk.Scrollbar(remote_frame, orient=tk.VERTICAL, command=remote_text.yview)
        remote_text.configure(yscrollcommand=remote_scrollbar.set)
        
        # Show actual file names with better formatting
        remote_files_text = ""
        for i, file in enumerate(analysis["remote_files"][:15]):  # Show up to 15 files
            remote_files_text += f"• {file}\n"
        if len(analysis["remote_files"]) > 15:
            remote_files_text += f"• ... and {len(analysis['remote_files']) - 15} more files\n"
        
        remote_text.config(state=tk.NORMAL)
        remote_text.insert(1.0, remote_files_text.strip())
        remote_text.config(state=tk.DISABLED)
        
        remote_text.pack(side=tk.LEFT, fill=tk.BOTH, expand=True, padx=10, pady=6)  # Increased padding
        if len(analysis["remote_files"]) > 8:  # Adjusted threshold
            remote_scrollbar.pack(side=tk.RIGHT, fill=tk.Y)
    
    # Choice storage
    choice = {"value": None}
    
    def set_choice(selected_choice):
        choice["value"] = selected_choice
        dialog.destroy()
    
    # Button area with increased spacing
    button_frame = tk.Frame(inner_frame, bg="#FFFFFF")
    button_frame.pack(fill=tk.X, pady=(25, 0))  # Increased spacing
    
    # Button 1: Smart Merge (Recommended) with better styling
    merge_btn = tk.Button(
        button_frame,
        text="🔄 Smart Merge (Recommended)",
        command=lambda: set_choice("merge"),
        font=("Arial", 12, "bold"),  # Slightly larger font
        bg="#6366F1",
        fg="#FFFFFF",
        activebackground="#4F46E5",
        activeforeground="#FFFFFF",
        relief=tk.FLAT,
        borderwidth=0,
        cursor="hand2",
        padx=25,  # Increased padding
        pady=15   # Increased padding
    )
    merge_btn.pack(fill=tk.X, pady=(0, 8))  # Increased spacing
    
    merge_desc = tk.Label(
        button_frame,
        text="Combines both local and remote files. Conflicts will be marked for manual review.",
        font=("Arial", 10),  # Slightly larger font
        bg="#FFFFFF",
        fg="#64748B",
        wraplength=700  # Increased wrap length
    )
    merge_desc.pack(anchor=tk.W, pady=(3, 18))  # Increased spacing
    
    # Button 2: Keep Local Files Only with better styling
    local_btn = tk.Button(
        button_frame,
        text="📁 Keep Local Files Only",
        command=lambda: set_choice("local"),
        font=("Arial", 12),  # Slightly larger font
        bg="#FFFFFF",
        fg="#6366F1",
        activebackground="#F1F5F9",
        activeforeground="#4F46E5",
        relief=tk.SOLID,
        borderwidth=1,
        cursor="hand2",
        padx=25,  # Increased padding
        pady=15   # Increased padding
    )
    local_btn.pack(fill=tk.X, pady=(0, 8))  # Increased spacing
    
    local_desc = tk.Label(
        button_frame,
        text="Overwrites remote repository with your local files.",
        font=("Arial", 10),  # Slightly larger font
        bg="#FFFFFF",
        fg="#64748B",
        wraplength=700  # Increased wrap length
    )
    local_desc.pack(anchor=tk.W, pady=(3, 18))  # Increased spacing
    
    # Button 3: Use Remote Files Only with better styling
    remote_btn = tk.Button(
        button_frame,
        text="🌐 Use Remote Files Only", 
        command=lambda: set_choice("remote"),
        font=("Arial", 12),  # Slightly larger font
        bg="#FFFFFF",
        fg="#6366F1",
        activebackground="#F1F5F9",
        activeforeground="#4F46E5",
        relief=tk.SOLID,
        borderwidth=1,
        cursor="hand2",
        padx=25,  # Increased padding
        pady=15   # Increased padding
    )
    remote_btn.pack(fill=tk.X, pady=(0, 8))  # Increased spacing
    
    remote_desc = tk.Label(
        button_frame,
        text="Downloads remote files and backs up your local files.",
        font=("Arial", 10),  # Slightly larger font
        bg="#FFFFFF",
        fg="#64748B",
        wraplength=700  # Increased wrap length
    )
    remote_desc.pack(anchor=tk.W)
    
    # Final initialization and rendering - ensure dialog is fully loaded before showing
    dialog.update_idletasks()
    dialog.after(50, lambda: None)  # Additional delay to ensure proper rendering
    
    # Wait for user choice
    parent_window.wait_window(dialog)
    return choice["value"]

# =============================================================================
# VAULT RECOVERY DIALOG
# =============================================================================

def create_vault_recovery_dialog(parent_window, vault_path):
    """Creates a dialog for vault recovery when the vault directory is missing."""
    dialog = tk.Toplevel(parent_window)
    dialog.title("Vault Directory Not Found")
    dialog.transient(parent_window) 
    dialog.grab_set()
    dialog.resizable(False, False)
    dialog.configure(bg=Colors.BG_PRIMARY)
    
    # Center the dialog
    dialog.update_idletasks()
    width, height = 500, 400
    x = (dialog.winfo_screenwidth() // 2) - (width // 2)
    y = (dialog.winfo_screenheight() // 2) - (height // 2)
    dialog.geometry(f"{width}x{height}+{x}+{y}")
    
    # Main container
    main_frame = tk.Frame(dialog, bg=Colors.BG_PRIMARY)
    main_frame.pack(fill=tk.BOTH, expand=True, padx=20, pady=20)
    
    # Header
    header_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
    header_frame.pack(fill=tk.X, pady=(0, 20))
    
    title_label = tk.Label(
        header_frame,
        text="⚠️ Vault Directory Not Found",
        font=(FONT_FAMILY_PRIMARY, Typography.XL, Typography.BOLD),
        bg=Colors.BG_PRIMARY,
        fg=Colors.ERROR
    )
    title_label.pack()
    
    # Content card
    content_card = PremiumCard.create(main_frame, padding=Spacing.LG)
    
    # Message
    message_text = (
        f"The configured vault directory could not be found:\n\n"
        f"{vault_path}\n\n"
        f"This might happen if the directory was moved, deleted, or is on a removable drive that's not connected.\n\n"
        f"Please choose how you'd like to proceed:"
    )
    
    message_label = tk.Label(
        content_card,
        text=message_text,
        font=(FONT_FAMILY_PRIMARY, Typography.MD),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_SECONDARY,
        justify=tk.LEFT,
        wraplength=420
    )
    message_label.pack(pady=(0, Spacing.LG), anchor=tk.W)
    
    # Choice storage
    choice = {"value": None}
    
    def set_choice(selected_choice):
        choice["value"] = selected_choice
        dialog.destroy()
    
    # Buttons
    button_frame = tk.Frame(content_card, bg=Colors.BG_CARD)
    button_frame.pack(fill=tk.X, pady=(Spacing.MD, 0))
    
    # Recreate directory button
    recreate_btn = PremiumButton.create_primary(
        button_frame,
        "Recreate Directory",
        lambda: set_choice("recreate"),
        Icons.FOLDER
    )
    recreate_btn.pack(fill=tk.X, pady=(0, Spacing.SM))
    
    # Select new directory button
    select_btn = PremiumButton.create_secondary(
        button_frame,
        "Select Different Directory",
        lambda: set_choice("select_new"),
        Icons.FOLDER
    )
    select_btn.pack(fill=tk.X, pady=(0, Spacing.SM))
    
    # Run setup again button
    setup_btn = PremiumButton.create_secondary(
        button_frame,
        "Run Setup Wizard Again",
        lambda: set_choice("setup"),
        Icons.GEAR
    )
    setup_btn.pack(fill=tk.X, pady=(0, Spacing.SM))
    
    # Cancel button
    cancel_btn = tk.Button(
        button_frame,
        text="Cancel",
        command=lambda: set_choice(None),
        font=(FONT_FAMILY_PRIMARY, Typography.SM),
        bg=Colors.BG_CARD,
        fg=Colors.TEXT_MUTED,
        relief=tk.FLAT,
        cursor="hand2",
        padx=Spacing.MD,
        pady=Spacing.XS
    )
    cancel_btn.pack(pady=(Spacing.SM, 0))
    
    # Wait for user choice
    parent_window.wait_window(dialog)
    return choice["value"]

# =============================================================================
# SSH KEY SUCCESS DIALOG
# =============================================================================

def show_ssh_key_success_dialog(public_key_preview, parent=None):
    """Shows SSH key generation success dialog with public key preview."""
    try:
        dialog = tk.Toplevel(parent if parent else tk.Tk())
        dialog.title("SSH Key Generated Successfully")
        dialog.transient(parent)
        dialog.grab_set()
        dialog.resizable(True, True)  # Allow resizing
        dialog.configure(bg=Colors.BG_PRIMARY)
        
        # Center the dialog
        dialog.update_idletasks()
        width, height = 800, 650  # Increased size for better display
        x = (dialog.winfo_screenwidth() // 2) - (width // 2)
        y = (dialog.winfo_screenheight() // 2) - (height // 2)
        dialog.geometry(f"{width}x{height}+{x}+{y}")
        
        # Main container
        main_frame = tk.Frame(dialog, bg=Colors.BG_PRIMARY)
        main_frame.pack(fill=tk.BOTH, expand=True, padx=20, pady=20)
        
        # Header
        header_frame = tk.Frame(main_frame, bg=Colors.BG_PRIMARY)
        header_frame.pack(fill=tk.X, pady=(0, 20))
        
        title_label = tk.Label(
            header_frame,
            text="✅ SSH Key Generated Successfully",
            font=(FONT_FAMILY_PRIMARY, 18, "bold"),
            bg=Colors.BG_PRIMARY,
            fg=Colors.SUCCESS
        )
        title_label.pack()
        
        # Content card (simple frame instead of PremiumCard)
        content_card = tk.Frame(main_frame, bg=Colors.BG_CARD, relief=tk.SOLID, borderwidth=1)
        content_card.pack(fill=tk.BOTH, expand=True, padx=5, pady=5)
        
        # Inner padding frame
        inner_frame = tk.Frame(content_card, bg=Colors.BG_CARD)
        inner_frame.pack(fill=tk.BOTH, expand=True, padx=20, pady=20)
        
        # Success message
        success_message = (
            "Your SSH key has been generated and copied to the clipboard!\n\n"
            "Next steps:\n"
            "1. The key is already copied to your clipboard\n"
            "2. Click 'Add SSH Key to GitHub' below\n"
            "3. Paste the key into GitHub's SSH key field\n"
            "4. Give it a descriptive title (e.g., 'Ogresync Key')\n"
            "5. Click 'Add SSH key'\n"
            "6. Return to Ogresync and click 'Execute: Test SSH' to continue\n\n"
            "After adding the key to GitHub, the SSH test should pass."
        )
        
        message_label = tk.Label(
            inner_frame,
            text=success_message,
            font=(FONT_FAMILY_PRIMARY, 11),
            bg=Colors.BG_CARD,
            fg=Colors.TEXT_SECONDARY,
            justify=tk.LEFT,
            wraplength=640
        )
        message_label.pack(pady=(0, 20), anchor=tk.W)
        
        # Key preview section with copy button
        key_container = tk.Frame(inner_frame, bg=Colors.BG_CARD)
        key_container.pack(fill=tk.BOTH, expand=True, pady=(0, 20))
        
        # Key preview header with copy button
        key_header = tk.Frame(key_container, bg=Colors.BG_CARD)
        key_header.pack(fill=tk.X, pady=(0, 5))
        
        key_title = tk.Label(
            key_header,
            text="SSH Public Key Preview",
            font=(FONT_FAMILY_PRIMARY, 12, "bold"),
            bg=Colors.BG_CARD,
            fg=Colors.TEXT_PRIMARY
        )
        key_title.pack(side=tk.LEFT)
        
        # Copy button in the header
        def copy_key_from_preview():
            try:
                import pyperclip
                pyperclip.copy(public_key_preview)
                # Show temporary success message
                copy_status.config(text="✅ Copied!", fg=Colors.SUCCESS)
                dialog.after(2000, lambda: copy_status.config(text="", fg=Colors.TEXT_SECONDARY))
            except ImportError:
                copy_status.config(text="❌ Copy failed - select text manually", fg=Colors.ERROR)
                dialog.after(3000, lambda: copy_status.config(text="", fg=Colors.TEXT_SECONDARY))
        
        copy_preview_btn = tk.Button(
            key_header,
            text="📋 Copy",
            command=copy_key_from_preview,
            font=(FONT_FAMILY_PRIMARY, 9, "normal"),
            bg=Colors.PRIMARY,
            fg=Colors.TEXT_INVERSE,
            relief=tk.FLAT,
            cursor="hand2",
            padx=12,
            pady=4
        )
        copy_preview_btn.pack(side=tk.RIGHT)
        
        # Copy status label
        copy_status = tk.Label(
            key_header,
            text="",
            font=(FONT_FAMILY_PRIMARY, 9),
            bg=Colors.BG_CARD,
            fg=Colors.TEXT_SECONDARY
        )
        copy_status.pack(side=tk.RIGHT, padx=(0, 10))
        
        # SSH Key text area with scrollbar
        key_text_frame = tk.Frame(key_container, bg=Colors.BG_SECONDARY)
        key_text_frame.pack(fill=tk.BOTH, expand=True)
        
        # Create text widget with scrollbar
        key_text = tk.Text(
            key_text_frame,
            height=8,
            wrap=tk.WORD,
            font=(FONT_FAMILY_MONO, 9),
            bg=Colors.BG_SECONDARY,
            fg=Colors.TEXT_PRIMARY,
            relief=tk.FLAT,
            borderwidth=1,
            selectbackground=Colors.PRIMARY_LIGHT,
            selectforeground=Colors.TEXT_PRIMARY
        )
        
        # Add scrollbar
        scrollbar = tk.Scrollbar(key_text_frame, orient=tk.VERTICAL, command=key_text.yview)
        key_text.configure(yscrollcommand=scrollbar.set)
        
        # Pack text widget and scrollbar
        key_text.pack(side=tk.LEFT, fill=tk.BOTH, expand=True, padx=5, pady=5)
        scrollbar.pack(side=tk.RIGHT, fill=tk.Y, pady=5)
        
        # Insert the FULL SSH key content (no truncation)
        key_text.insert(tk.END, public_key_preview)
        key_text.config(state=tk.DISABLED)
        
        # Bottom buttons
        button_frame = tk.Frame(inner_frame, bg=Colors.BG_CARD)
        button_frame.pack(fill=tk.X, pady=(20, 0))
        
        def open_github_settings():
            import webbrowser
            webbrowser.open("https://github.com/settings/ssh/new")  # Direct link to add SSH key
        
        def copy_key_again():
            try:
                import pyperclip
                pyperclip.copy(public_key_preview)
                copy_status.config(text="✅ Copied to clipboard!", fg=Colors.SUCCESS)
                dialog.after(2000, lambda: copy_status.config(text="", fg=Colors.TEXT_SECONDARY))
            except ImportError:
                copy_status.config(text="❌ Copy failed", fg=Colors.ERROR)
                dialog.after(3000, lambda: copy_status.config(text="", fg=Colors.TEXT_SECONDARY))
        
        # GitHub button (primary action)
        github_btn = tk.Button(
            button_frame,
            text="🌐 Add SSH Key to GitHub",
            command=open_github_settings,
            font=(FONT_FAMILY_PRIMARY, 11, "bold"),
            bg=Colors.SUCCESS,
            fg=Colors.TEXT_INVERSE,
            relief=tk.FLAT,
            cursor="hand2",
            padx=20,
            pady=10
        )
        github_btn.pack(side=tk.LEFT, padx=(0, 10))
        
        # Copy again button
        copy_btn = tk.Button(
            button_frame,
            text="📋 Copy Key Again",
            command=copy_key_again,
            font=(FONT_FAMILY_PRIMARY, 11, "normal"),
            bg=Colors.PRIMARY,
            fg=Colors.TEXT_INVERSE,
            relief=tk.FLAT,
            cursor="hand2",
            padx=20,
            pady=10
        )
        copy_btn.pack(side=tk.LEFT, padx=(0, 10))
        
        # Close button
        close_btn = tk.Button(
            button_frame,
            text="Close",
            command=dialog.destroy,
            font=(FONT_FAMILY_PRIMARY, 11, "normal"),
            bg=Colors.BG_SECONDARY,
            fg=Colors.TEXT_PRIMARY,
            relief=tk.SOLID,
            borderwidth=1,
            cursor="hand2",
            padx=20,
            pady=10
        )
        close_btn.pack(side=tk.RIGHT)
        
        # Make sure dialog is modal
        dialog.focus_set()
        dialog.grab_set()
        
        # Wait for user to close
        if parent:
            parent.wait_window(dialog)
        else:
            dialog.mainloop()
            
    except Exception as e:
        # Fallback to simple message dialog
        print(f"Error in SSH dialog: {e}")
        try:
            show_info_message(
                "SSH Key Generated",
                f"SSH key generated successfully!\n\n"
                f"Key preview: {public_key_preview[:100]}...\n\n"
                f"The key has been copied to your clipboard. "
                f"Please add it to your GitHub account at:\n"
                f"https://github.com/settings/ssh/new"
            )
        except:
            import tkinter.messagebox as messagebox
            messagebox.showinfo(
                "SSH Key Generated",
                f"SSH key generated successfully!\n\n"
                f"Please copy your SSH key from ~/.ssh/id_rsa.pub\n"
                f"and add it to GitHub at:\n"
                f"https://github.com/settings/ssh/new"
            )

# =============================================================================
# ENHANCED CONFLICT RESOLUTION DIALOGS
# =============================================================================

def create_two_stage_conflict_dialog(parent_window, analysis, vault_path):
    """
    Creates an enhanced two-stage conflict resolution dialog.
    Stage 1: Strategy selection
    Stage 2: File-by-file resolution (if needed)
    """
    try:
        # Import conflict resolution module
        import conflict_resolution
        
        # Create resolver and show dialog
        resolver = conflict_resolution.ConflictResolver(vault_path, parent_window)
        return resolver.resolve_conflicts()
        
    except ImportError:
        # Fallback to original dialog if conflict_resolution module not available
        return create_repository_conflict_dialog(parent_window, 
                                                "Repository conflicts detected", 
                                                analysis)
    except Exception as e:
        print(f"Error in two-stage conflict dialog: {e}")
        # Fallback to original dialog
        return create_repository_conflict_dialog(parent_window,
                                                "Repository conflicts detected",
                                                analysis)