agentcrew-ai 0.8.13__py3-none-any.whl → 0.9.1__py3-none-any.whl

AgentCrew/modules/gui/themes/style_provider.py

@@ -1,422 +1,236 @@
-from .catppuccin import CatppuccinTheme
-from .atom_light import AtomLightTheme
-from .nord import NordTheme
-from .dracula import DraculaTheme
-from .unicorn import UnicornTheme
-from .saigontech import SaigonTechTheme
+from .theme_loader import ThemeLoader, ThemeData
 from AgentCrew.modules.config import ConfigManagement
 from PySide6.QtCore import Signal, QObject
 
 
 class StyleProvider(QObject):
-    """Provides styling for the chat window and components."""
-
-    # Signal emitted when theme changes
     theme_changed = Signal(str)
-
     _instance = None
 
     def __new__(cls):
-        """Singleton pattern to ensure only one instance exists."""
         if cls._instance is None:
             cls._instance = super(StyleProvider, cls).__new__(cls)
             cls._instance._initialized = False
         return cls._instance
 
     def __init__(self):
-        """Initialize the style provider by reading theme from global config."""
         if self._initialized:
             return
 
         super().__init__()
         self._initialized = True
 
-        # Read theme from global config
         self.config_manager = ConfigManagement()
         global_config = self.config_manager.read_global_config_data()
-        self.theme = global_config.get("global_settings", {}).get("theme", "saigontech")
+        self.theme = global_config.get("global_settings", {}).get("theme", "catppuccin")
 
         self._set_theme_class()
 
     def _set_theme_class(self):
-        """Set the theme class based on the current theme setting."""
-        if self.theme == "light":
-            self.theme_class = AtomLightTheme
-        elif self.theme == "nord":
-            self.theme_class = NordTheme
-        elif self.theme == "dracula":
-            self.theme_class = DraculaTheme
-        elif self.theme == "unicorn":
-            self.theme_class = UnicornTheme
-        elif self.theme == "dark":
-            self.theme_class = CatppuccinTheme
-        else:
-            self.theme_class = SaigonTechTheme  # Default to Catppuccin for "dark"
+        self.theme_class: ThemeData = ThemeLoader.load_theme(self.theme)
 
     def update_theme(self, reload=True):
-        """
-        Update the theme based on the current configuration.
-
-        Args:
-            reload (bool): If True, reload the theme from the configuration.
-                          If False, use the currently set theme.
-
-        Returns:
-            bool: True if the theme changed, False otherwise.
-        """
         if reload:
-            # Re-read from config
             global_config = self.config_manager.read_global_config_data()
-            new_theme = global_config.get("global_settings", {}).get("theme", "dark")
+            new_theme = global_config.get("global_settings", {}).get(
+                "theme", "catppuccin"
+            )
 
-            # Check if theme changed
             if new_theme != self.theme:
                 self.theme = new_theme
+                ThemeLoader.clear_cache()
                 self._set_theme_class()
                 self.theme_changed.emit(self.theme)
                 return True
         return False
 
+    def _get_attr(self, name: str, default: str = "") -> str:
+        return getattr(self.theme_class, name, default)
+
     def get_main_style(self):
-        """Get the main style for the chat window."""
-        return self.theme_class.MAIN_STYLE
+        return self._get_attr("MAIN_STYLE")
 
     def get_config_window_style(self):
-        return self.theme_class.CONFIG_DIALOG
+        return self._get_attr("CONFIG_DIALOG")
 
     def get_button_style(self, button_type="primary"):
-        """Get style for buttons based on type."""
-        if button_type == "primary":
-            return self.theme_class.PRIMARY_BUTTON
-        elif button_type == "secondary":
-            return self.theme_class.SECONDARY_BUTTON
-        elif button_type == "stop":
-            return self.theme_class.STOP_BUTTON
-        elif button_type == "disabled":
-            return self.theme_class.DISABLED_BUTTON
-        elif button_type == "stop_stopping":
-            return self.theme_class.STOP_BUTTON_STOPPING
-        elif button_type == "red":
-            return self.theme_class.RED_BUTTON
-        elif button_type == "green":
-            return self.theme_class.GREEN_BUTTON
-        elif button_type == "agent_menu":
-            return self.theme_class.AGENT_MENU_BUTTON
-        else:
-            return ""
+        button_map = {
+            "primary": "PRIMARY_BUTTON",
+            "secondary": "SECONDARY_BUTTON",
+            "stop": "STOP_BUTTON",
+            "disabled": "DISABLED_BUTTON",
+            "stop_stopping": "STOP_BUTTON_STOPPING",
+            "red": "RED_BUTTON",
+            "green": "GREEN_BUTTON",
+            "agent_menu": "AGENT_MENU_BUTTON",
+        }
+        attr_name = button_map.get(button_type, "")
+        return self._get_attr(attr_name) if attr_name else ""
 
     def get_input_style(self):
-        """Get style for text input."""
-        return self.theme_class.TEXT_INPUT
+        return self._get_attr("TEXT_INPUT")
 
     def get_menu_style(self):
-        """Get style for menus."""
-        return self.theme_class.MENU_BAR
+        return self._get_attr("MENU_BAR")
 
     def get_status_indicator_style(self):
-        """Get style for status indicator."""
-        return self.theme_class.STATUS_INDICATOR
+        return self._get_attr("STATUS_INDICATOR")
 
     def get_version_label_style(self):
-        """Get style for version label."""
-        return self.theme_class.VERSION_LABEL
+        return self._get_attr("VERSION_LABEL")
 
     def get_tool_dialog_text_edit_style(self):
-        """Get style for tool dialog text edit."""
-        return self.theme_class.TOOL_DIALOG_TEXT_EDIT
+        return self._get_attr("TOOL_DIALOG_TEXT_EDIT")
 
     def get_tool_dialog_yes_button_style(self):
-        """Get style for tool dialog yes button."""
-        return self.theme_class.TOOL_DIALOG_YES_BUTTON
+        return self._get_attr("TOOL_DIALOG_YES_BUTTON")
 
     def get_tool_dialog_all_button_style(self):
-        """Get style for tool dialog all button."""
-        return self.theme_class.TOOL_DIALOG_ALL_BUTTON
+        return self._get_attr("TOOL_DIALOG_ALL_BUTTON")
 
     def get_tool_dialog_no_button_style(self):
-        """Get style for tool dialog no button."""
-        return self.theme_class.TOOL_DIALOG_NO_BUTTON
+        return self._get_attr("TOOL_DIALOG_NO_BUTTON")
 
     def get_system_message_label_style(self):
-        """Get style for system message labels."""
-        return self.theme_class.SYSTEM_MESSAGE_LABEL
+        return self._get_attr("SYSTEM_MESSAGE_LABEL")
 
     def get_system_message_toggle_style(self):
-        """Get style for system message toggle buttons."""
-        return self.theme_class.SYSTEM_MESSAGE_TOGGLE
+        return self._get_attr("SYSTEM_MESSAGE_TOGGLE")
 
     def get_sidebar_style(self):
-        """Get style for sidebar widgets."""
-        return self.theme_class.SIDEBAR
+        return self._get_attr("SIDEBAR")
 
     def get_conversation_list_style(self):
-        """Get style for conversation list."""
-        return self.theme_class.CONVERSATION_LIST
+        return self._get_attr("CONVERSATION_LIST")
 
     def get_search_box_style(self):
-        """Get style for search boxes."""
-        return self.theme_class.SEARCH_BOX
+        return self._get_attr("SEARCH_BOX")
 
     def get_token_usage_style(self):
-        """Get style for token usage widgets."""
-        return self.theme_class.TOKEN_USAGE
+        return self._get_attr("TOKEN_USAGE")
 
     def get_token_usage_widget_style(self):
-        """Get style for token usage widget background."""
-        return self.theme_class.TOKEN_USAGE_WIDGET
+        return self._get_attr("TOKEN_USAGE_WIDGET")
 
     def get_context_menu_style(self):
-        """Get style for context menus."""
-        return self.theme_class.CONTEXT_MENU
+        return self._get_attr("CONTEXT_MENU")
 
     def get_agent_menu_style(self):
-        """Get style for agent menus."""
-        return self.theme_class.AGENT_MENU
+        return self._get_attr("AGENT_MENU")
 
     def get_user_bubble_style(self):
-        """Get style for user message bubbles."""
-        return self.theme_class.USER_BUBBLE
+        return self._get_attr("USER_BUBBLE")
 
     def get_assistant_bubble_style(self):
-        """Get style for assistant message bubbles."""
-        return self.theme_class.ASSISTANT_BUBBLE
+        return self._get_attr("ASSISTANT_BUBBLE")
 
     def get_thinking_bubble_style(self):
-        """Get style for thinking message bubbles."""
-        return self.theme_class.THINKING_BUBBLE
+        return self._get_attr("THINKING_BUBBLE")
 
     def get_consolidated_bubble_style(self):
-        """Get style for consolidated message bubbles."""
-        return self.theme_class.CONSOLIDATED_BUBBLE
+        return self._get_attr("CONSOLIDATED_BUBBLE")
 
     def get_splitter_style(self):
-        return self.theme_class.SPLITTER_COLOR
+        return self._get_attr("SPLITTER_COLOR")
 
     def get_code_color_style(self):
-        return self.theme_class.CODE_CSS
+        return self._get_attr("CODE_CSS")
 
     def get_rollback_button_style(self):
-        """Get style for rollback buttons."""
-        return self.theme_class.ROLLBACK_BUTTON
+        return self._get_attr("ROLLBACK_BUTTON")
 
     def get_consolidated_button_style(self):
-        """Get style for consolidated buttons."""
-        return self.theme_class.CONSOLIDATED_BUTTON
+        return self._get_attr("CONSOLIDATED_BUTTON")
 
     def get_unconsolidate_button_style(self):
-        """Get style for unconsolidate buttons."""
-        return self.theme_class.UNCONSOLIDATE_BUTTON
+        return self._get_attr("UNCONSOLIDATE_BUTTON")
 
     def get_user_message_label_style(self):
-        """Get style for user message labels."""
-        return self.theme_class.USER_MESSAGE_LABEL
+        return self._get_attr("USER_MESSAGE_LABEL")
 
     def get_assistant_message_label_style(self):
-        """Get style for assistant message labels."""
-        return self.theme_class.ASSISTANT_MESSAGE_LABEL
+        return self._get_attr("ASSISTANT_MESSAGE_LABEL")
 
     def get_thinking_message_label_style(self):
-        """Get style for thinking message labels."""
-        return self.theme_class.THINKING_MESSAGE_LABEL
+        return self._get_attr("THINKING_MESSAGE_LABEL")
 
     def get_user_sender_label_style(self):
-        """Get style for user sender labels."""
-        return self.theme_class.USER_SENDER_LABEL
+        return self._get_attr("USER_SENDER_LABEL")
 
     def get_assistant_sender_label_style(self):
-        """Get style for assistant sender labels."""
-        return self.theme_class.ASSISTANT_SENDER_LABEL
+        return self._get_attr("ASSISTANT_SENDER_LABEL")
 
     def get_thinking_sender_label_style(self):
-        """Get style for thinking sender labels."""
-        return self.theme_class.THINKING_SENDER_LABEL
+        return self._get_attr("THINKING_SENDER_LABEL")
 
     def get_metadata_header_label_style(self):
-        """Get style for metadata header labels."""
-        return self.theme_class.METADATA_HEADER_LABEL
+        return self._get_attr("METADATA_HEADER_LABEL")
 
     def get_user_file_name_label_style(self):
-        """Get style for user file name labels."""
-        return self.theme_class.USER_FILE_NAME_LABEL
+        return self._get_attr("USER_FILE_NAME_LABEL")
 
     def get_assistant_file_name_label_style(self):
-        """Get style for assistant file name labels."""
-        return self.theme_class.ASSISTANT_FILE_NAME_LABEL
+        return self._get_attr("ASSISTANT_FILE_NAME_LABEL")
 
     def get_user_file_info_label_style(self):
-        """Get style for user file info labels."""
-        return self.theme_class.USER_FILE_INFO_LABEL
+        return self._get_attr("USER_FILE_INFO_LABEL")
 
     def get_assistant_file_info_label_style(self):
-        """Get style for assistant file info labels."""
-        return self.theme_class.ASSISTANT_FILE_INFO_LABEL
+        return self._get_attr("ASSISTANT_FILE_INFO_LABEL")
 
     def get_api_keys_group_style(self):
-        """Get style for API keys group boxes."""
-        return self.theme_class.API_KEYS_GROUP
+        return self._get_attr("API_KEYS_GROUP")
 
     def get_editor_container_widget_style(self):
-        """Get style for editor container widgets."""
-        return self.theme_class.EDITOR_CONTAINER_WIDGET
+        return self._get_attr("EDITOR_CONTAINER_WIDGET")
 
     def get_combo_box_style(self):
-        """Get style for combo boxes."""
-        return self.theme_class.COMBO_BOX
+        return self._get_attr("COMBO_BOX")
 
     def get_checkbox_style(self):
-        """Get style for checkboxes with enhanced tristate support."""
-        return getattr(self.theme_class, "CHECKBOX_STYLE", "")
+        return self._get_attr("CHECKBOX_STYLE", "")
 
     def get_tool_widget_style(self):
-        """Get style for tool widgets."""
-        return (
-            self.theme_class.TOOL_WIDGET
-            if hasattr(self.theme_class, "TOOL_WIDGET")
-            else ""
-        )
+        return self._get_attr("TOOL_WIDGET", "")
 
     def get_tool_card_style(self):
-        """Get style for tool widget cards."""
-        return (
-            self.theme_class.TOOL_CARD if hasattr(self.theme_class, "TOOL_CARD") else ""
-        )
+        return self._get_attr("TOOL_CARD", "")
 
     def get_tool_card_error_style(self):
-        """Get style for tool widget cards in error state."""
-        return (
-            self.theme_class.TOOL_CARD_ERROR
-            if hasattr(self.theme_class, "TOOL_CARD_ERROR")
-            else ""
-        )
+        return self._get_attr("TOOL_CARD_ERROR", "")
 
     def get_tool_header_style(self):
-        """Get style for tool widget headers."""
-        return (
-            self.theme_class.TOOL_HEADER
-            if hasattr(self.theme_class, "TOOL_HEADER")
-            else ""
-        )
+        return self._get_attr("TOOL_HEADER", "")
 
     def get_tool_toggle_button_style(self):
-        """Get style for tool widget toggle buttons."""
-        return (
-            self.theme_class.TOOL_TOGGLE_BUTTON
-            if hasattr(self.theme_class, "TOOL_TOGGLE_BUTTON")
-            else ""
-        )
+        return self._get_attr("TOOL_TOGGLE_BUTTON", "")
 
     def get_tool_status_style(self):
-        """Get style for tool widget status indicators."""
-        return (
-            self.theme_class.TOOL_STATUS
-            if hasattr(self.theme_class, "TOOL_STATUS")
-            else ""
-        )
+        return self._get_attr("TOOL_STATUS", "")
 
     def get_tool_content_style(self):
-        """Get style for tool widget content."""
-        return (
-            self.theme_class.TOOL_CONTENT
-            if hasattr(self.theme_class, "TOOL_CONTENT")
-            else ""
-        )
+        return self._get_attr("TOOL_CONTENT", "")
 
     def get_tool_progress_style(self):
-        """Get style for tool widget progress bars."""
-        return (
-            self.theme_class.TOOL_PROGRESS
-            if hasattr(self.theme_class, "TOOL_PROGRESS")
-            else ""
-        )
+        return self._get_attr("TOOL_PROGRESS", "")
 
     def get_tool_separator_style(self):
-        """Get style for tool widget separators."""
-        return (
-            self.theme_class.TOOL_SEPARATOR
-            if hasattr(self.theme_class, "TOOL_SEPARATOR")
-            else ""
-        )
+        return self._get_attr("TOOL_SEPARATOR", "")
 
     def get_tool_icon(self, tool_name):
-        """Get icon for a specific tool."""
-        icons = getattr(self.theme_class, "TOOL_ICONS", {})
-        return icons.get(tool_name, icons.get("default", "🔧"))
+        return self.theme_class.get_icon(tool_name, "\U0001f527")
 
     def get_json_editor_colors(self):
-        """Get color scheme for JSON editor from current theme."""
         return self.theme_class.JSON_EDITOR_COLORS
 
     def get_json_editor_style(self):
-        """Get complete stylesheet for JSON editor."""
         return self.theme_class.JSON_EDITOR_STYLE
 
     def get_markdown_editor_colors(self):
-        """Get color scheme for Markdown editor from current theme."""
-        return getattr(
-            self.theme_class,
-            "MARKDOWN_EDITOR_COLORS",
-            {
-                "background": "#313244",
-                "text": "#cdd6f4",
-                "border": "#45475a",
-                "header": "#89b4fa",
-                "bold": "#fab387",
-                "italic": "#a6e3a1",
-                "code": "#f5c2e7",
-                "code_background": "#45475a",
-                "link": "#74c7ec",
-                "image": "#cba6f7",
-                "list": "#f9e2af",
-                "blockquote": "#94e2d5",
-                "hr": "#6c7086",
-                "strikethrough": "#eba0ac",
-                "error": "#f38ba8",
-            },
-        )
+        return self.theme_class.MARKDOWN_EDITOR_COLORS
 
     def get_markdown_editor_style(self):
-        """Get complete stylesheet for Markdown editor."""
-        return getattr(
-            self.theme_class,
-            "MARKDOWN_EDITOR_STYLE",
-            """
-QPlainTextEdit {
-    background-color: #313244;
-    color: #cdd6f4;
-    border: 1px solid #45475a;
-    border-radius: 4px;
-    padding: 8px;
-    font-family: 'Consolas', 'Monaco', 'Courier New', monospace;
-    font-size: 12px;
-    line-height: 1.4;
-}
-QPlainTextEdit:focus {
-    border: 1px solid #89b4fa;
-}
-""",
-        )
+        return self.theme_class.MARKDOWN_EDITOR_STYLE
 
     def get_diff_colors(self):
-        """Get color scheme for diff widget from current theme."""
-        return getattr(
-            self.theme_class,
-            "DIFF_COLORS",
-            {
-                "background": "#1e1e2e",
-                "panel_bg": "#313244",
-                "header_bg": "#45475a",
-                "header_text": "#cdd6f4",
-                "line_number_bg": "#181825",
-                "line_number_text": "#6c7086",
-                "removed_bg": "#3b2d33",
-                "removed_text": "#f38ba8",
-                "removed_highlight": "#f38ba8",
-                "added_bg": "#2d3b33",
-                "added_text": "#a6e3a1",
-                "added_highlight": "#a6e3a1",
-                "unchanged_text": "#6c7086",
-                "border": "#45475a",
-                "block_header_bg": "#585b70",
-                "block_header_text": "#b4befe",
-            },
-        )
+        return self.theme_class.DIFF_COLORS