mcp-freecad

"""Centralized Theme System for FreeCAD AI GUI Components""" from enum import Enum from PySide2 import QtCore, QtWidgets class Theme(Enum): """Available themes.""" LIGHT = "light" DARK = "dark" AUTO = "auto" # Follow system theme class ColorScheme: """Color scheme definitions.""" def __init__(self, theme: Theme = Theme.LIGHT): self.theme = theme self._load_colors() def _load_colors(self): """Load colors based on theme.""" if self.theme == Theme.LIGHT: self.colors = { # Background colors "background_primary": "#ffffff", "background_secondary": "#f8f9fa", "background_tertiary": "#e9ecef", "background_accent": "#e3f2fd", # Text colors "text_primary": "#212529", "text_secondary": "#6c757d", "text_muted": "#adb5bd", "text_inverse": "#ffffff", # Brand colors "primary": "#2196F3", "secondary": "#6c757d", "success": "#4CAF50", "warning": "#FF9800", "error": "#f44336", "info": "#17a2b8", # Chat-specific colors "user_message_bg": "#e3f2fd", "ai_message_bg": "#e8f5e9", "system_message_bg": "#fff3e0", "user_text": "#2196F3", "ai_text": "#4CAF50", "system_text": "#FF9800", # UI element colors "border": "#dee2e6", "border_focus": "#2196F3", "button_primary": "#2196F3", "button_success": "#4CAF50", "button_warning": "#FF9800", "button_danger": "#f44336", # Status colors "status_connected": "#4CAF50", "status_warning": "#FF9800", "status_error": "#f44336", "status_unknown": "#9E9E9E", } else: # DARK theme self.colors = { # Background colors "background_primary": "#1a1a1a", "background_secondary": "#2d2d2d", "background_tertiary": "#3a3a3a", "background_accent": "#1e3a5f", # Text colors "text_primary": "#ffffff", "text_secondary": "#b3b3b3", "text_muted": "#808080", "text_inverse": "#000000", # Brand colors (slightly adjusted for dark theme) "primary": "#42A5F5", "secondary": "#9e9e9e", "success": "#66BB6A", "warning": "#FFA726", "error": "#EF5350", "info": "#29B6F6", # Chat-specific colors "user_message_bg": "#1e3a5f", "ai_message_bg": "#2e5d33", "system_message_bg": "#4a3326", "user_text": "#42A5F5", "ai_text": "#66BB6A", "system_text": "#FFA726", # UI element colors "border": "#4a4a4a", "border_focus": "#42A5F5", "button_primary": "#42A5F5", "button_success": "#66BB6A", "button_warning": "#FFA726", "button_danger": "#EF5350", # Status colors "status_connected": "#66BB6A", "status_warning": "#FFA726", "status_error": "#EF5350", "status_unknown": "#9E9E9E", } def get_color(self, color_name: str) -> str: """Get color by name.""" return self.colors.get(color_name, "#000000") def get_rgba(self, color_name: str, alpha: float = 1.0) -> str: """Get color as rgba string.""" hex_color = self.get_color(color_name) if hex_color.startswith("#"): hex_color = hex_color[1:] r = int(hex_color[0:2], 16) g = int(hex_color[2:4], 16) b = int(hex_color[4:6], 16) return f"rgba({r}, {g}, {b}, {alpha})" class StyleSheet: """Generate consistent stylesheets.""" def __init__(self, color_scheme: ColorScheme): self.colors = color_scheme def get_button_style(self, button_type: str = "primary") -> str: """Get button stylesheet.""" color_map = { "primary": self.colors.get_color("button_primary"), "success": self.colors.get_color("button_success"), "warning": self.colors.get_color("button_warning"), "danger": self.colors.get_color("button_danger"), } base_color = color_map.get(button_type, self.colors.get_color("primary")) hover_color = self._darken_color(base_color, 0.1) return f""" QPushButton {{ background-color: {base_color}; color: {self.colors.get_color("text_inverse")}; border: none; padding: 8px 16px; border-radius: 4px; font-weight: bold; }} QPushButton:hover {{ background-color: {hover_color}; }} QPushButton:pressed {{ background-color: {self._darken_color(base_color, 0.2)}; }} QPushButton:disabled {{ background-color: {self.colors.get_color("text_muted")}; }} """ def get_input_style(self) -> str: """Get input field stylesheet.""" return f""" QLineEdit, QTextEdit {{ background-color: {self.colors.get_color("background_primary")}; border: 2px solid {self.colors.get_color("border")}; border-radius: 4px; padding: 8px; color: {self.colors.get_color("text_primary")}; }} QLineEdit:focus, QTextEdit:focus {{ border-color: {self.colors.get_color("border_focus")}; }} """ def get_groupbox_style(self) -> str: """Get group box stylesheet.""" return f""" QGroupBox {{ background-color: {self.colors.get_color("background_secondary")}; border: 1px solid {self.colors.get_color("border")}; border-radius: 6px; margin-top: 10px; padding-top: 10px; color: {self.colors.get_color("text_primary")}; font-weight: bold; }} QGroupBox::title {{ subcontrol-origin: margin; left: 10px; padding: 0 5px 0 5px; }} """ def get_conversation_style(self) -> str: """Get conversation display stylesheet.""" return f""" QTextBrowser {{ background-color: {self.colors.get_color("background_primary")}; border: 1px solid {self.colors.get_color("border")}; border-radius: 4px; padding: 10px; color: {self.colors.get_color("text_primary")}; font-family: Arial, sans-serif; }} """ def get_status_indicator_style(self, status: str) -> str: """Get status indicator stylesheet.""" status_colors = { "connected": self.colors.get_color("status_connected"), "warning": self.colors.get_color("status_warning"), "error": self.colors.get_color("status_error"), "unknown": self.colors.get_color("status_unknown"), } color = status_colors.get(status, self.colors.get_color("status_unknown")) return f""" QLabel {{ color: {color}; font-weight: bold; font-size: 16px; }} """ def get_tab_style(self) -> str: """Get tab widget stylesheet.""" return f""" QTabWidget::pane {{ border: 1px solid {self.colors.get_color("border")}; background-color: {self.colors.get_color("background_primary")}; }} QTabBar::tab {{ background-color: {self.colors.get_color("background_secondary")}; color: {self.colors.get_color("text_primary")}; padding: 8px 16px; margin-right: 2px; border-top-left-radius: 4px; border-top-right-radius: 4px; }} QTabBar::tab:selected {{ background-color: {self.colors.get_color("primary")}; color: {self.colors.get_color("text_inverse")}; }} QTabBar::tab:hover {{ background-color: {self.colors.get_rgba("primary", 0.2)}; }} """ def _darken_color(self, hex_color: str, factor: float) -> str: """Darken a hex color by factor (0.0 to 1.0).""" if hex_color.startswith("#"): hex_color = hex_color[1:] r = int(hex_color[0:2], 16) g = int(hex_color[2:4], 16) b = int(hex_color[4:6], 16) r = max(0, int(r * (1 - factor))) g = max(0, int(g * (1 - factor))) b = max(0, int(b * (1 - factor))) return f"#{r:02x}{g:02x}{b:02x}" class ThemeManager: """Manages themes across the application.""" def __init__(self): self.current_theme = Theme.LIGHT self.color_scheme = ColorScheme(self.current_theme) self.stylesheet = StyleSheet(self.color_scheme) self._theme_changed_callbacks = [] def set_theme(self, theme: Theme): """Set application theme.""" if theme == self.current_theme: return self.current_theme = theme self.color_scheme = ColorScheme(theme) self.stylesheet = StyleSheet(self.color_scheme) # Notify callbacks for callback in self._theme_changed_callbacks: try: callback(theme) except Exception as e: print(f"ThemeManager: Error in theme change callback: {e}") def register_theme_change_callback(self, callback): """Register callback for theme changes.""" self._theme_changed_callbacks.append(callback) def apply_theme_to_widget(self, widget: QtWidgets.QWidget): """Apply current theme to a widget.""" try: # Apply base styling widget.setStyleSheet(f""" QWidget {{ background-color: {self.color_scheme.get_color("background_primary")}; color: {self.color_scheme.get_color("text_primary")}; }} """) # Apply specific styling based on widget type self._apply_specific_styling(widget) except Exception as e: print(f"ThemeManager: Error applying theme to widget: {e}") def _apply_specific_styling(self, widget: QtWidgets.QWidget): """Apply theme-specific styling to widget types.""" # Find and style specific widget types buttons = widget.findChildren(QtWidgets.QPushButton) for button in buttons: # Determine button type from style or text if "danger" in button.objectName().lower() or "delete" in button.text().lower(): button.setStyleSheet(self.stylesheet.get_button_style("danger")) elif "success" in button.objectName().lower() or "save" in button.text().lower(): button.setStyleSheet(self.stylesheet.get_button_style("success")) elif "warning" in button.objectName().lower(): button.setStyleSheet(self.stylesheet.get_button_style("warning")) else: button.setStyleSheet(self.stylesheet.get_button_style("primary")) # Style input fields inputs = widget.findChildren(QtWidgets.QLineEdit) + widget.findChildren(QtWidgets.QTextEdit) for input_widget in inputs: input_widget.setStyleSheet(self.stylesheet.get_input_style()) # Style group boxes groupboxes = widget.findChildren(QtWidgets.QGroupBox) for groupbox in groupboxes: groupbox.setStyleSheet(self.stylesheet.get_groupbox_style()) # Style tab widgets tabwidgets = widget.findChildren(QtWidgets.QTabWidget) for tabwidget in tabwidgets: tabwidget.setStyleSheet(self.stylesheet.get_tab_style()) # Global theme manager instance _theme_manager = None def get_theme_manager() -> ThemeManager: """Get global theme manager instance.""" global _theme_manager if _theme_manager is None: _theme_manager = ThemeManager() return _theme_manager def apply_theme_to_widget(widget: QtWidgets.QWidget, theme: Theme = None): """Convenience function to apply theme to widget.""" manager = get_theme_manager() if theme and theme != manager.current_theme: manager.set_theme(theme) manager.apply_theme_to_widget(widget) def get_current_color_scheme() -> ColorScheme: """Get current color scheme.""" return get_theme_manager().color_scheme def get_current_stylesheet() -> StyleSheet: """Get current stylesheet generator.""" return get_theme_manager().stylesheet