pygpt-net 2.4.37__py3-none-any.whl → 2.4.41__py3-none-any.whl

pygpt_net/controller/theme/markdown.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.20 21:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
 import os
@@ -23,6 +23,7 @@ class Markdown:
         """
         self.window = window
         self.css = {}  # external styles
+        self.web_style = ""
 
     def update(self, force: bool = False):
         """
@@ -60,7 +61,8 @@ class Markdown:
 
         :return: stylesheet
         """
-        if "web" not in self.css:
+        web_style = self.window.core.config.get("theme.style", "blocks")
+        if "web" not in self.css or self.web_style != web_style:
             self.load()
         if "web" in self.css:
             return self.css["web"]
@@ -73,31 +75,41 @@ class Markdown:
         self.window.dispatch(event)  # per current engine
         self.window.controller.ctx.refresh()
         self.window.controller.ctx.refresh_output()
-        data = {
+        event = RenderEvent(RenderEvent.END, {
             "meta": meta,
-        }
-        event = RenderEvent(RenderEvent.END, data)
+        })
         self.window.dispatch(event)
 
     def load(self):
         """Load markdown styles"""
-        parents = ["markdown", "web"]
+        parents = [
+            "markdown",
+            "web",
+        ]
+        web_style = self.window.core.config.get("theme.style", "blocks")
         for base_name in parents:
+            suffix = ""
+            if base_name == 'web':
+                suffix = "-" + web_style
+                self.web_style = web_style
             theme = self.window.core.config.get('theme')
             name = str(base_name)
-            color_name = str(base_name)
             if theme.startswith('light'):
-                color_name += '.light'
+                color = '.light'
             else:
-                color_name += '.dark'
+                color = '.dark'
+
+            # load CSS, app + user
+            file_base = name + suffix + '.css'
+            file_color = name + suffix + color + '.css'
             paths = []
-            paths.append(os.path.join(self.window.core.config.get_app_path(), 'data', 'css', name + '.css'))
-            paths.append(os.path.join(self.window.core.config.get_app_path(), 'data', 'css', color_name + '.css'))
-            paths.append(os.path.join(self.window.core.config.get_user_path(), 'css', name + '.css'))
-            paths.append(os.path.join(self.window.core.config.get_user_path(), 'css', color_name + '.css'))
+            paths.append(os.path.join(self.window.core.config.get_app_path(), 'data', 'css', file_base))
+            paths.append(os.path.join(self.window.core.config.get_app_path(), 'data', 'css', file_color))
+            paths.append(os.path.join(self.window.core.config.get_user_path(), 'css', file_base))
+            paths.append(os.path.join(self.window.core.config.get_user_path(), 'css', file_color))
             content = ''
             for path in paths:
-                if os.path.exists(path):
+                if os.path.exists(path) and os.path.isfile(path):
                     with open(path, 'r') as file:
                         content += file.read()
 
@@ -107,8 +119,12 @@ class Markdown:
             except KeyError as e:  # ignore missing env vars
                 pass
 
-    def get_default(self):
-        """Set default markdown CSS"""
+    def get_default(self) -> str:
+        """
+        Set default markdown CSS
+
+        :return: default CSS
+        """
         colors = {
             "dark": {
                 "a": "#fff",

pygpt_net/controller/theme/menu.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.05 23:00:00                  #
+# Updated Date: 2024.12.07 21:00:00                  #
 # ================================================== #
 
 from PySide6.QtGui import QAction
@@ -30,6 +30,18 @@ class Menu:
         # setup themes list menu
         if self.loaded:
             return
+
+        # styles
+        styles = self.window.controller.theme.common.get_styles_list()
+        for style in styles:
+            style_id = style.lower()
+            title = style.replace('_', ' ').title()
+            self.window.ui.menu['theme_style'][style_id] = QAction(title, self.window, checkable=True)
+            self.window.ui.menu['theme_style'][style_id].triggered.connect(
+                lambda checked=None, style=style_id: self.window.controller.theme.toggle_style(style))
+            self.window.ui.menu['theme.style'].addAction(self.window.ui.menu['theme_style'][style_id])
+
+        # color themes
         themes = self.window.controller.theme.common.get_themes_list()
         for theme in themes:
             name = self.window.controller.theme.common.translate(theme)
@@ -42,6 +54,7 @@ class Menu:
                 self.window.ui.menu['theme.dark'].addAction(self.window.ui.menu['theme'][theme])
             elif theme.startswith('light'):
                 self.window.ui.menu['theme.light'].addAction(self.window.ui.menu['theme'][theme])
+
         self.loaded = True
 
     def setup_syntax(self):
@@ -86,14 +99,22 @@ class Menu:
 
     def update_list(self):
         """Update theme list menu"""
-        current = self.window.core.config.get('theme')
+        # styles
+        current_style = self.window.core.config.get('theme.style')
+        for style in self.window.ui.menu['theme_style']:
+            self.window.ui.menu['theme_style'][style].setChecked(False)
+        if current_style in self.window.ui.menu['theme_style']:
+            self.window.ui.menu['theme_style'][current_style].setChecked(True)
+
+        # color themes
+        current_theme = self.window.core.config.get('theme')
         for theme in self.window.ui.menu['theme']:
             self.window.ui.menu['theme'][theme].setChecked(False)
-        if current in self.window.ui.menu['theme']:
-            self.window.ui.menu['theme'][current].setChecked(True)
+        if current_theme in self.window.ui.menu['theme']:
+            self.window.ui.menu['theme'][current_theme].setChecked(True)
 
     def update_syntax(self):
-        """Update syntax menu"""
+        """Update code syntax highlight menu"""
         current = self.window.core.config.get('render.code_syntax')
         for style in self.window.ui.menu['theme_syntax']:
             self.window.ui.menu['theme_syntax'][style].setChecked(False)

pygpt_net/controller/theme/nodes.py

@@ -6,9 +6,9 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.20 21:00:00                  #
+# Updated Date: 2024.12.12 01:00:00                  #
 # ================================================== #
-# 
+
 from pygpt_net.core.events import RenderEvent
 
 
@@ -93,9 +93,6 @@ class Nodes:
                     continue
                 self.apply(key, type)
 
-        # self.window.interpreter.update_stylesheet(self.window.controller.theme.style('font.chat.output'))
-        # self.window.ui.nodes['interpreter.input'].update_stylesheet(self.window.controller.theme.style('font.chat.input'))
-
         # apply to notepads
         num_notepads = self.window.controller.notepad.get_num_notepads()
         if num_notepads > 0:

pygpt_net/controller/tools/__init__.py

@@ -6,10 +6,17 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.07 23:00:00                  #
+# Updated Date: 2024.12.14 08:00:00                  #
 # ================================================== #
 
+from typing import Dict, List
+
+from PySide6.QtGui import QAction, QIcon
+from PySide6.QtWidgets import QTabWidget, QMenu
+
 from pygpt_net.core.tabs.tab import Tab
+from pygpt_net.utils import trans
+
 
 class Tools:
     def __init__(self, window=None):
@@ -44,7 +51,38 @@ class Tools:
         if idx is not None:
             self.window.controller.ui.tabs.switch_tab_by_idx(idx)
 
-    def get_tab_tools(self) -> dict:
+    def append_tab_menu(
+            self,
+            parent: QTabWidget,
+            menu: QMenu,
+            idx: int,
+            column_idx: int
+    ) -> QMenu:
+        """
+        Append tab menu
+
+        :param parent: parent widget
+        :param menu: menu
+        :param idx: tab index
+        :param column_idx: column index
+        :return: tab add submenu
+        """
+        submenu = menu.addMenu(QIcon(":/icons/add.svg"), trans("action.tab.add.tool"))
+        tools = self.window.tools.get_all()
+        for id in tools:
+            tool = tools[id]
+            if not tool.has_tab:
+                continue
+            icon = tool.tab_icon
+            title = trans(tool.tab_title)
+            action = QAction(QIcon(icon), title, parent)
+            action.triggered.connect(
+                lambda idx=idx, column_idx=column_idx, id=id: parent.add_tab(idx, column_idx, Tab.TAB_TOOL, id)
+            )
+            submenu.addAction(action)
+        return submenu
+
+    def get_tab_tools(self) -> Dict[str, List[str]]:
         """
         Get tab tools
 

pygpt_net/controller/ui/__init__.py

@@ -6,9 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.21 20:00:00                  #
+# Updated Date: 2024.12.14 00:00:00                  #
 # ================================================== #
 
+from typing import Optional
+
 from PySide6.QtGui import QColor
 
 from pygpt_net.utils import trans
@@ -47,10 +49,6 @@ class UI:
         self.update()
         self.init_toolbox()
 
-    def pre_setup(self):
-        """Post setup UI"""
-        self.tabs.setup()
-
     def update(self):
         """Update all elements"""
 
@@ -149,7 +147,7 @@ class UI:
             model_str = str(model)
         self.window.ui.nodes['chat.model'].setText(model_str)
 
-    def update_ctx_label(self, label: str = None):
+    def update_ctx_label(self, label: Optional[str] = None):
         """
         Update ctx label