pygpt-net 2.1.37__py3-none-any.whl → 2.4.48__py3-none-any.whl

pygpt_net/ui/widget/lists/uploaded.py

@@ -6,10 +6,10 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.01.29 14:00:00                  #
+# Updated Date: 2024.04.26 23:00:00                  #
 # ================================================== #
 
-from PySide6.QtGui import QAction, QIcon, QResizeEvent
+from PySide6.QtGui import QAction, QIcon, QResizeEvent, Qt
 from PySide6.QtWidgets import QMenu
 
 from pygpt_net.ui.widget.lists.base import BaseList
@@ -30,31 +30,44 @@ class UploadedFileList(BaseList):
         self.id = id
         self.doubleClicked.connect(self.dblclick)
         self.setHeaderHidden(False)
+        self.clicked.disconnect(self.click)
 
         self.header = self.header()
-        self.header.setStretchLastSection(True)
+        self.header.setStretchLastSection(False)
 
-        self.column_proportion = 0.5
+        self.column_proportion = 0.3
         self.adjustColumnWidths()
 
     def adjustColumnWidths(self):
         total_width = self.width()
         first_column_width = int(total_width * self.column_proportion)
         self.setColumnWidth(0, first_column_width)
-        for column in range(1, 2):
-            self.setColumnWidth(column, (total_width - first_column_width) // (2 - 1))
+        for column in range(1, 4):
+            self.setColumnWidth(column, (total_width - first_column_width) // (4 - 1))
 
     def resizeEvent(self, event: QResizeEvent):
         super().resizeEvent(event)
         self.adjustColumnWidths()
 
+    def mousePressEvent(self, event):
+        """
+        Mouse press event
+
+        :param event: mouse event
+        """
+        if event.buttons() == Qt.LeftButton:
+            index = self.indexAt(event.pos())
+            if index.isValid():
+                self.window.controller.assistant.files.select(index.row())
+        super(UploadedFileList, self).mousePressEvent(event)
+
     def click(self, val):
         """
         Click event
 
         :param val: click event
         """
-        self.window.controller.assistant.files.select(val.row())
+        pass
 
     def dblclick(self, val):
         """

pygpt_net/ui/widget/option/checkbox.py

@@ -6,16 +6,26 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.01.19 18:00:00                  #
+# Updated Date: 2024.11.24 22:00:00                  #
 # ================================================== #
 
-from PySide6.QtWidgets import QCheckBox, QHBoxLayout, QWidget
+from PySide6.QtGui import QIcon
+from PySide6.QtWidgets import QCheckBox, QHBoxLayout, QWidget, QLabel
 
+from pygpt_net.ui.widget.anims.toggles import AnimToggle
 from pygpt_net.utils import trans
 
+import pygpt_net.icons_rc
 
 class OptionCheckbox(QWidget):
-    def __init__(self, window=None, parent_id: str = None, id: str = None, option: dict = None):
+    def __init__(
+            self,
+            window=None,
+            parent_id: str = None,
+            id: str = None,
+            option: dict = None,
+            icon = None
+    ):
         """
         Settings checkbox
 
@@ -23,7 +33,10 @@ class OptionCheckbox(QWidget):
         :param id: option id
         :param parent_id: parent option id
         :param option: option data
+        :param icon: icon
         """
+        # TODO: https://pypi.org/project/QtAwesome/
+
         super(OptionCheckbox, self).__init__(window)
         self.window = window
         self.id = id
@@ -43,8 +56,10 @@ class OptionCheckbox(QWidget):
             if "real_time" in self.option:
                 self.real_time = self.option["real_time"]
 
-        self.box = QCheckBox(self.title, self.window)
-        self.box.setChecked(self.value)
+        # self.box = QCheckBox(self.title, self.window)
+        self.box = AnimToggle()
+        if self.value is not None:
+            self.box.setChecked(self.value)
         self.box.stateChanged.connect(
             lambda: self.window.controller.config.checkbox.on_update(
                 self.parent_id,
@@ -53,8 +68,56 @@ class OptionCheckbox(QWidget):
                 self.box.isChecked()
             )
         )
-
+        self.label = QLabel(self.title)
         self.layout = QHBoxLayout()
         self.layout.addWidget(self.box)
 
+        # add icon if defined
+        if icon is not None:
+            ico = QLabel()
+            pixmap = QIcon(icon).pixmap(24, 24)
+            ico.setPixmap(pixmap)
+            self.layout.addWidget(ico)
+
+
+        self.layout.addWidget(self.label)
+        self.layout.addStretch()
+        self.layout.setContentsMargins(0, 0, 0, 0)
         self.setLayout(self.layout)
+
+        # self.layout = QHBoxLayout()
+        #self.layout.addWidget(self.box)
+
+        #self.setLayout(self.layout)
+
+    def setIcon(self, icon: str):
+        """
+        Set icon
+
+        :param icon: icon
+        """
+        self.box.setIcon(icon)
+
+    def setText(self, text: str):
+        """
+        Set label text
+
+        :param text: text
+        """
+        self.label.setText(text)
+
+    def setChecked(self, state: bool):
+        """
+        Set checkbox state
+
+        :param state: state
+        """
+        self.box.setChecked(state)
+
+    def isChecked(self) -> bool:
+        """
+        Get checkbox state
+
+        :return: state
+        """
+        return self.box.isChecked()

pygpt_net/ui/widget/option/cmd.py

@@ -6,17 +6,19 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.03.12 06:00:00                  #
+# Updated Date: 2024.11.24 22:00:00                  #
 # ================================================== #
 
+from PySide6.QtGui import QIcon
 from PySide6.QtWidgets import QWidget, QVBoxLayout, QLabel
 
-from pygpt_net.plugin.base import BasePlugin
+from pygpt_net.plugin.base.plugin import BasePlugin
 from pygpt_net.ui.widget.element.group import CollapsedGroup
 from pygpt_net.ui.widget.option.checkbox import OptionCheckbox
 from pygpt_net.ui.widget.option.dictionary import OptionDict
 from pygpt_net.ui.widget.option.textarea import OptionTextarea
 from pygpt_net.utils import trans
+
 import pygpt_net.icons_rc
 
 
@@ -98,6 +100,8 @@ class OptionCmd(QWidget):
         self.window.ui.nodes[desc_key].setWordWrap(True)
         self.window.ui.nodes[desc_key].setMaximumHeight(40)
         self.window.ui.nodes[desc_key].setStyleSheet("font-size: 10px;")
+        self.window.ui.nodes[desc_key].setProperty('class', 'label-help')
+        self.window.ui.nodes[desc_key].setContentsMargins(35, 0, 0, 0)
 
         instr_key = "settings.cmd.field.instruction"
         params_key = "settings.cmd.field.params"
@@ -105,7 +109,7 @@ class OptionCmd(QWidget):
         params_label = QLabel(trans(params_key))
 
         # widgets
-        self.enabled = OptionCheckbox(self.window, parent_id, key_enabled, option_enabled)  # enable checkbox
+        self.enabled = OptionCheckbox(self.window, parent_id, key_enabled, option_enabled, icon = ":/icons/build.svg")  # enable checkbox
         self.params = OptionDict(self.window, parent_id, key_params, option_params)  # command params
         self.instruction = OptionTextarea(self.window, parent_id, key_instruction, option_instruction)  # command instruction
 
@@ -124,7 +128,9 @@ class OptionCmd(QWidget):
         group_id = self.parent_id + '.' + id + '.config'
         group = CollapsedGroup(self.window, group_id, None, False, None)
         group.box.setText(trans('settings.cmd.config.collapse'))
+        group.box.setIcon(QIcon(":/icons/expand.svg"))
         group.add_layout(params_layout)
+        group.layout.setContentsMargins(25, 0, 0, 0)
 
         # add to groups
         self.window.ui.groups[group_id] = group
@@ -133,8 +139,8 @@ class OptionCmd(QWidget):
         self.setLayout(self.layout)
 
         # show tooltip only if different from description
-        if txt_tooltip != txt_desc:
-            self.setToolTip(txt_tooltip)
+        # if txt_tooltip != txt_desc:
+            # self.setToolTip(txt_tooltip)
 
         # update
         self.update()

pygpt_net/ui/widget/option/combo.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2023.12.25 21:00:00                  #
+# Updated Date: 2024.11.17 17:00:00                  #
 # ================================================== #
 
 from PySide6.QtWidgets import QHBoxLayout, QWidget, QComboBox
@@ -43,13 +43,16 @@ class OptionCombo(QWidget):
         self.real_time = False
         self.combo = NoScrollCombo()
         self.combo.currentIndexChanged.connect(self.on_combo_change)
+        self.current_id = None
 
         # add items
         self.update()
 
         self.layout = QHBoxLayout()
         self.layout.addWidget(self.combo)
+        self.layout.setContentsMargins(0, 0, 0, 0)
         self.setLayout(self.layout)
+        self.fit_to_content()
 
     def update(self):
         """Prepare items"""
@@ -61,6 +64,7 @@ class OptionCombo(QWidget):
                 self.keys = self.option["keys"]
             if "value" in self.option:
                 self.value = self.option["value"]
+                self.current_id = self.value
             if "real_time" in self.option:
                 self.real_time = self.option["real_time"]
 
@@ -71,7 +75,36 @@ class OptionCombo(QWidget):
                     for key, value in item.items():
                         self.combo.addItem(value, key)
                 else:
-                    self.combo.addItem(item)
+                    self.combo.addItem(item, item)
+
+    def set_value(self, value):
+        """
+        Set value
+
+        :param value: value
+        """
+        index = self.combo.findData(value)
+        if index != -1:
+            self.combo.setCurrentIndex(index)
+
+    def get_value(self):
+        """
+        Get value
+
+        :return: value
+        """
+        return self.current_id
+
+    def set_keys(self, keys):
+        """
+        Set keys
+
+        :param keys: keys
+        """
+        self.keys = keys
+        self.option["keys"] = keys
+        self.combo.clear()
+        self.update()
 
     def on_combo_change(self, index):
         """
@@ -80,6 +113,10 @@ class OptionCombo(QWidget):
         :param index: combo index
         :return:
         """
-        current_id = self.combo.itemData(index)
-        self.window.controller.config.combo.on_update(self.parent_id, self.id, self.option, current_id)
+        self.current_id = self.combo.itemData(index)
+        self.window.controller.config.combo.on_update(self.parent_id, self.id, self.option, self.current_id)
+
+    def fit_to_content(self):
+        """Fit to content"""
+        self.combo.setSizeAdjustPolicy(QComboBox.AdjustToContents)
 

pygpt_net/ui/widget/option/dictionary.py

@@ -6,13 +6,13 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.03.11 01:00:00                  #
+# Updated Date: 2024.08.20 00:00:00                  #
 # ================================================== #
 
 from PySide6.QtCore import Qt, QAbstractItemModel, QModelIndex, QSize
 from PySide6.QtGui import QAction, QIcon
 from PySide6.QtWidgets import QWidget, QVBoxLayout, QPushButton, QTreeView, QMenu, QStyledItemDelegate, QComboBox, \
-    QCheckBox, QSizePolicy
+    QCheckBox, QHeaderView, QHBoxLayout
 
 from pygpt_net.utils import trans
 import pygpt_net.icons_rc
@@ -63,22 +63,30 @@ class OptionDict(QWidget):
         # append dict model
         self.list.setModel(self.model)
 
+        # init layout
+        self.init_layout()
+
+        # update model list
+        self.update()
+
+    def init_layout(self):
         # add button
         self.add_btn = QPushButton(trans('action.add'), self)
         self.add_btn.setCursor(Qt.PointingHandCursor)
         self.add_btn.clicked.connect(self.add)
         self.add_btn.setAutoDefault(False)
 
+        self.buttons = QHBoxLayout()
+        self.buttons.addWidget(self.add_btn)
+        self.buttons.setContentsMargins(0, 0, 0, 0)
+
         # layout
         self.layout = QVBoxLayout()
         self.layout.setContentsMargins(0, 0, 0, 0)
-        self.layout.addWidget(self.add_btn)
+        self.layout.addLayout(self.buttons)
         self.layout.addWidget(self.list)
         self.setLayout(self.layout)
 
-        # update model list
-        self.update()
-
     def add(self):
         """Add new empty item to settings dict"""
         # create empty dict
@@ -182,6 +190,10 @@ class OptionDictItems(QTreeView):
         self.setIndentation(0)
         self.setHeaderHidden(False)
 
+        header = self.header()
+        header.setStretchLastSection(True)
+        header.setSectionResizeMode(QHeaderView.Stretch)
+
         # setup fields
         keys = self.parent.keys
         idx = 0
@@ -374,7 +386,7 @@ class ComboBoxDelegate(QStyledItemDelegate):
     def createEditor(self, parent, option, index):
         editor = QComboBox(parent)
         for item in self.keys:
-            if type(item) is dict:
+            if isinstance(item, dict):
                 for key, value in item.items():
                     editor.addItem(value, key)
             else:
@@ -382,11 +394,22 @@ class ComboBoxDelegate(QStyledItemDelegate):
         return editor
 
     def setEditorData(self, editor, index):
-        value = index.model().data(index, Qt.EditRole)
-        editor.setCurrentText(value)
+        key = index.model().data(index, Qt.EditRole)
+        if key is not None:
+            index = editor.findData(key)
+            if index >= 0:
+                editor.setCurrentIndex(index)
 
     def setModelData(self, editor, model, index):
-        model.setData(index, editor.currentText(), Qt.EditRole)
+        key = editor.currentData()
+        model.setData(index, key, Qt.EditRole)
+
+    def displayText(self, value, locale):
+        for item in self.keys:
+            if isinstance(item, dict):
+                if value in item:
+                    return item[value]
+        return value
 
 
 class MaxHeightDelegate(QStyledItemDelegate):

pygpt_net/ui/widget/option/input.py

@@ -6,11 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.02.16 16:00:00                  #
+# Updated Date: 2024.04.26 23:00:00                  #
 # ================================================== #
 
+from PySide6.QtCore import Qt
 from PySide6.QtGui import QAction, QIcon
-from PySide6.QtWidgets import QLineEdit
+from PySide6.QtWidgets import QLineEdit, QFileDialog
 import pygpt_net.icons_rc
 
 
@@ -44,6 +45,8 @@ class OptionInputInline(QLineEdit):
                 self.value = self.option["value"]
             if "real_time" in self.option:
                 self.real_time = self.option["real_time"]
+            if "read_only" in self.option and self.option["read_only"]:
+                self.setReadOnly(True)
 
     def focusOutEvent(self, event):
         """On focus out event"""
@@ -101,6 +104,8 @@ class OptionInput(QLineEdit):
                 self.value = self.option["value"]
             if "real_time" in self.option:
                 self.real_time = self.option["real_time"]
+            if "read_only" in self.option and self.option["read_only"]:
+                self.setReadOnly(True)
 
     def keyPressEvent(self, event):
         """
@@ -183,3 +188,102 @@ class PasswordInput(QLineEdit):
             self.option,
             self.text()
         )
+
+
+class DirectoryInput(QLineEdit):
+    def __init__(self, window=None, parent_id: str = None, id: str = None, option: dict = None):
+        """
+        Directory select input
+
+        :param window: main window
+        :param id: option id
+        :param parent_id: parent option id
+        :param option: option data
+        """
+        super(DirectoryInput, self).__init__(window)
+        self.window = window
+        self.id = id
+        self.parent_id = parent_id
+        self.option = option
+        self.value = ""
+        self.title = ""
+        self.setReadOnly(True)
+        self.allow_file = False
+        self.allow_multiple = False
+
+        # from option data
+        if self.option is not None:
+            if "label" in self.option:
+                self.title = self.option["label"]
+            if "value" in self.option:
+                self.value = self.option["value"]
+                self.setText(self.value)
+            if "extra" in self.option:
+                if "allow_file" in self.option["extra"]:
+                    self.allow_file = self.option["extra"]["allow_file"]
+                if "allow_multiple" in self.option["extra"]:
+                    self.allow_multiple = self.option["extra"]["allow_multiple"]
+
+        self.select = QAction(self)
+        self.select.setIcon(QIcon(":/icons/more_horizontal.svg"))
+        if self.allow_file:
+            self.select.triggered.connect(self.open_select_files)
+        else:
+            self.select.triggered.connect(self.open_select_dir)
+        self.addAction(self.select, QLineEdit.TrailingPosition)
+
+    def open_select_dir(self):
+        """Open directory dialog"""
+        value = None
+        options = QFileDialog.Options()
+        directory = QFileDialog.getExistingDirectory(
+            self.window,
+            "Select directory...",
+            options=options
+        )
+        if directory:
+            value = directory
+        if value:
+            self.value = value
+            self.setText(value)
+
+    def open_select_files(self):
+        """Open file(s) dialog"""
+        value = None
+        options = QFileDialog.Options()
+        if self.allow_multiple:
+            files, _ = QFileDialog.getOpenFileNames(
+                self.window,
+                "Select file(s)...",
+                "",
+                "All Files (*)",
+                options=options
+            )
+        else:
+            files, _ = QFileDialog.getOpenFileName(
+                self.window,
+                "Select file...",
+                "",
+                "All Files (*)",
+                options=options
+            )
+        if files:
+            value = files
+        if value:
+            self.value = value
+            if self.allow_multiple:
+                value = ", ".join(value)
+            self.setText(value)
+
+    def keyPressEvent(self, event):
+        """
+        Key press event
+
+        :param event: key event
+        """
+        super(DirectoryInput, self).keyPressEvent(event)
+
+    def clear(self):
+        """Clear input"""
+        self.setText("")
+        self.value = ""

pygpt_net/ui/widget/option/prompt.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.03.15 15:00:00                  #
+# Updated Date: 2024.11.08 22:00:00                  #
 # ================================================== #
 
 from PySide6.QtWidgets import QTextEdit
@@ -57,6 +57,19 @@ class PromptTextarea(QTextEdit):
         """
         super(PromptTextarea, self).keyPressEvent(event)
 
+    def contextMenuEvent(self, event):
+        """
+        Context menu event
+
+        :param event: Event
+        """
+        menu = self.createStandardContextMenu()
+        try:
+            self.window.core.prompt.template.to_menu_options(menu, "global")
+            self.window.core.prompt.custom.to_menu_options(menu, "global")
+        except Exception as e:
+            self.window.core.debug.log(e)
+        menu.exec_(event.globalPos())
 
     def on_prompt_changed(self):
         """On prompt changed"""

pygpt_net/ui/widget/option/textarea.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.02.18 18:00:00                  #
+# Updated Date: 2024.11.08 22:00:00                  #
 # ================================================== #
 
 from PySide6.QtWidgets import QTextEdit
@@ -38,6 +38,7 @@ class OptionTextarea(QTextEdit):
         self.real_time = False
         self.update_ui = True
         self.setAcceptRichText(False)
+        self.context_options = []
 
         # init from option data
         if self.option is not None:
@@ -47,6 +48,22 @@ class OptionTextarea(QTextEdit):
                 self.value = self.option["value"]
             if "real_time" in self.option:
                 self.real_time = self.option["real_time"]
+            if "context_options" in self.option:
+                self.context_options = self.option["context_options"]
+            if "read_only" in self.option and self.option["read_only"]:
+                self.setReadOnly(True)
+
+    def contextMenuEvent(self, event):
+        """
+        Context menu event
+
+        :param event: Event
+        """
+        menu = self.createStandardContextMenu()
+        if "prompt.template.paste" in self.context_options:  # tpl options
+            self.window.core.prompt.template.to_menu_options(menu, "editor")
+            self.window.core.prompt.custom.to_menu_options(menu, "editor")
+        menu.exec_(event.globalPos())
 
     def keyPressEvent(self, event):
         """

pygpt_net/ui/widget/option/toggle.py

@@ -0,0 +1,63 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# ================================================== #
+# This file is a part of PYGPT package               #
+# Website: https://pygpt.net                         #
+# GitHub:  https://github.com/szczyglis-dev/py-gpt   #
+# MIT License                                        #
+# Created By  : Marcin Szczygliński                  #
+# Updated Date: 2024.11.24 22:00:00                  #
+# ================================================== #
+
+from PySide6.QtWidgets import QCheckBox, QHBoxLayout, QWidget
+
+from pygpt_net.ui.widget.anims.toggles import AnimToggle
+from pygpt_net.utils import trans
+
+
+class OptionCheckbox(QWidget):
+    def __init__(self, window=None, parent_id: str = None, id: str = None, option: dict = None):
+        """
+        Settings checkbox
+
+        :param window: main window
+        :param id: option id
+        :param parent_id: parent option id
+        :param option: option data
+        """
+        super(OptionCheckbox, self).__init__(window)
+        self.window = window
+        self.id = id
+        self.parent_id = parent_id
+        self.option = option
+        self.value = False
+        self.title = ""
+        self.real_time = False
+
+        # init from option data
+        if self.option is not None:
+            if "label" in self.option and self.option["label"] is not None \
+                    and self.option["label"] != "":
+                self.title = trans(self.option["label"])
+            if "value" in self.option:
+                self.value = self.option["value"]
+            if "real_time" in self.option:
+                self.real_time = self.option["real_time"]
+
+        self.box = AnimToggle(self.title, self.window)
+        if self.value is not None:
+            self.box.setChecked(self.value)
+
+        self.box.stateChanged.connect(
+            lambda: self.window.controller.config.checkbox.on_update(
+                self.parent_id,
+                self.id,
+                self.option,
+                self.box.isChecked()
+            )
+        )
+
+        self.layout = QHBoxLayout()
+        self.layout.addWidget(self.box)
+
+        self.setLayout(self.layout)