pygpt-net 2.4.30__py3-none-any.whl → 2.4.35__py3-none-any.whl

pygpt_net/ui/layout/chat/attachments.py

@@ -6,18 +6,20 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.23 00:00:00                  #
+# Updated Date: 2024.11.26 02:00:00                  #
 # ================================================== #
 
 import os
 
-from PySide6.QtGui import QStandardItemModel, Qt
+from PySide6.QtGui import QStandardItemModel, Qt, QIcon
 from PySide6.QtWidgets import QVBoxLayout, QPushButton, QHBoxLayout, QCheckBox, QWidget
 
+from pygpt_net.item.attachment import AttachmentItem
 from pygpt_net.ui.widget.element.labels import HelpLabel
 from pygpt_net.ui.widget.lists.attachment import AttachmentList
 from pygpt_net.utils import trans
 
+import pygpt_net.icons_rc
 
 class Attachments:
     def __init__(self, window=None):
@@ -38,12 +40,21 @@ class Attachments:
         self.setup_attachments()
         self.setup_buttons()
 
+        empty_widget = QWidget()
+        self.window.ui.nodes['input.attachments.options.label'] = HelpLabel(trans("attachments.options.label"))
+
         # buttons layout
         buttons = QHBoxLayout()
         buttons.addWidget(self.window.ui.nodes['attachments.btn.add'])
+        buttons.addWidget(self.window.ui.nodes['attachments.btn.add_url'])
         buttons.addWidget(self.window.ui.nodes['attachments.btn.clear'])
+        buttons.addWidget(empty_widget)
+        buttons.addWidget(self.window.ui.nodes['input.attachments.options.label'])
+
+        buttons.addWidget(self.setup_auto_index())
         buttons.addWidget(self.setup_send_clear())
         buttons.addWidget(self.setup_capture_clear())
+        buttons.addStretch()
 
         self.window.ui.nodes['tip.input.attachments'] = HelpLabel(trans('tip.input.attachments'), self.window)
 
@@ -88,15 +99,34 @@ class Attachments:
 
         return widget
 
+    def setup_auto_index(self) -> QWidget:
+        """
+        Setup auto index checkbox
+
+        :return: QWidget
+        """
+        layout = QHBoxLayout()
+        layout.setContentsMargins(0, 0, 0, 0)
+        layout.setAlignment(Qt.AlignCenter)
+        layout.addWidget(self.window.ui.nodes['attachments.auto_index'])
+
+        widget = QWidget()
+        widget.setLayout(layout)
+
+        return widget
+
     def setup_buttons(self):
         """
         Setup buttons
         """
-        self.window.ui.nodes['attachments.btn.add'] = QPushButton(trans('attachments.btn.add'))
-        self.window.ui.nodes['attachments.btn.clear'] = QPushButton(trans('attachments.btn.clear'))
+        self.window.ui.nodes['attachments.btn.add'] = QPushButton(QIcon(":/icons/add.svg"), trans('attachments.btn.add'))
+        self.window.ui.nodes['attachments.btn.add_url'] = QPushButton(QIcon(":/icons/add.svg"), trans('attachments.btn.add_url'))
+        self.window.ui.nodes['attachments.btn.clear'] = QPushButton(QIcon(":/icons/close.svg"), trans('attachments.btn.clear'))
 
         self.window.ui.nodes['attachments.btn.add'].clicked.connect(
             lambda: self.window.controller.attachment.open_add())
+        self.window.ui.nodes['attachments.btn.add_url'].clicked.connect(
+            lambda: self.window.controller.attachment.open_add_url())
         self.window.ui.nodes['attachments.btn.clear'].clicked.connect(
             lambda: self.window.controller.attachment.clear(remove_local=True))
 
@@ -110,6 +140,11 @@ class Attachments:
             lambda: self.window.controller.attachment.toggle_capture_clear(
                 self.window.ui.nodes['attachments.capture_clear'].isChecked()))
 
+        self.window.ui.nodes['attachments.auto_index'] = QCheckBox(trans('attachments.auto_index'))
+        self.window.ui.nodes['attachments.auto_index'].stateChanged.connect(
+            lambda: self.window.controller.attachment.toggle_auto_index(
+                self.window.ui.nodes['attachments.auto_index'].isChecked()))
+
     def setup_attachments(self):
         """
         Setup attachments list
@@ -144,8 +179,9 @@ class Attachments:
         for id in data:
             path = data[id].path
             size = ""
-            if path and os.path.exists(path):
-                size = self.window.core.filesystem.sizeof_fmt(os.path.getsize(path))
+            if data[id].type == AttachmentItem.TYPE_FILE:
+                if path and os.path.exists(path):
+                    size = self.window.core.filesystem.sizeof_fmt(os.path.getsize(path))
             ctx_str = ""
             if data[id].ctx:
                 ctx_str = "YES"

pygpt_net/ui/layout/chat/attachments_ctx.py

@@ -6,19 +6,20 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.23 00:00:00                  #
+# Updated Date: 2024.11.26 02:00:00                  #
 # ================================================== #
 
 import os
 
 from PySide6 import QtCore
-from PySide6.QtGui import QStandardItemModel, Qt
+from PySide6.QtGui import QStandardItemModel, Qt, QIcon
 from PySide6.QtWidgets import QVBoxLayout, QPushButton, QHBoxLayout, QWidget, QRadioButton
 
 from pygpt_net.ui.widget.element.labels import HelpLabel
 from pygpt_net.ui.widget.lists.attachment_ctx import AttachmentCtxList
 from pygpt_net.utils import trans
 
+import pygpt_net.icons_rc
 
 class AttachmentsCtx:
     def __init__(self, window=None):
@@ -90,7 +91,7 @@ class AttachmentsCtx:
         self.window.ui.nodes[self.id] = AttachmentCtxList(self.window)
 
         # buttons
-        self.window.ui.nodes['attachments_ctx.btn.clear'] = QPushButton(trans('attachments_uploaded.btn.clear'))
+        self.window.ui.nodes['attachments_ctx.btn.clear'] = QPushButton(QIcon(":/icons/close.svg"), trans('attachments_uploaded.btn.clear'))
         self.window.ui.nodes['attachments_ctx.btn.clear'].clicked.connect(
             lambda: self.window.controller.chat.attachment.clear()
         )
@@ -105,11 +106,12 @@ class AttachmentsCtx:
         :param parent: parent widget
         :return: QStandardItemModel
         """
-        model = QStandardItemModel(0, 4, parent)
+        model = QStandardItemModel(0, 5, parent)
         model.setHeaderData(0, Qt.Horizontal, trans('attachments.header.name'))
         model.setHeaderData(1, Qt.Horizontal, trans('attachments.header.path'))
         model.setHeaderData(2, Qt.Horizontal, trans('attachments.header.size'))
         model.setHeaderData(3, Qt.Horizontal, trans('attachments.header.length'))
+        model.setHeaderData(4, Qt.Horizontal, trans('attachments.header.idx'))
         return model
 
     def update(self, data):
@@ -121,6 +123,7 @@ class AttachmentsCtx:
         self.window.ui.models[self.id].removeRows(0, self.window.ui.models[self.id].rowCount())
         i = 0
         for item in data:
+            indexed = False
             name = "No name"
             if 'name' in item:
                 name = item['name']
@@ -136,6 +139,12 @@ class AttachmentsCtx:
                 length = str(item['length'])
             if 'tokens' in item:
                 length += ' / ~' + str(item['tokens'])
+            if 'indexed' in item and item['indexed']:
+                indexed = True
+
+            idx_str = ""
+            if indexed:
+                idx_str = trans("attachments.ctx.indexed")
 
             if os.path.exists(path):
                 size = self.window.core.filesystem.sizeof_fmt(os.path.getsize(path))
@@ -149,4 +158,5 @@ class AttachmentsCtx:
             self.window.ui.models[self.id].setData(self.window.ui.models[self.id].index(i, 1), path)
             self.window.ui.models[self.id].setData(self.window.ui.models[self.id].index(i, 2), size)
             self.window.ui.models[self.id].setData(self.window.ui.models[self.id].index(i, 3), length)
+            self.window.ui.models[self.id].setData(self.window.ui.models[self.id].index(i, 4), idx_str)
             i += 1

pygpt_net/ui/layout/chat/attachments_uploaded.py

@@ -12,7 +12,7 @@
 import os
 
 from PySide6 import QtCore
-from PySide6.QtGui import QStandardItemModel, Qt
+from PySide6.QtGui import QStandardItemModel, Qt, QIcon
 from PySide6.QtWidgets import QVBoxLayout, QPushButton, QHBoxLayout, QCheckBox, QLabel, QWidget
 
 from pygpt_net.ui.widget.element.button import SyncButton
@@ -20,6 +20,7 @@ from pygpt_net.ui.widget.element.labels import HelpLabel
 from pygpt_net.ui.widget.lists.uploaded import UploadedFileList
 from pygpt_net.utils import trans
 
+import pygpt_net.icons_rc
 
 class AttachmentsUploaded:
     def __init__(self, window=None):
@@ -38,10 +39,11 @@ class AttachmentsUploaded:
         :return: QVBoxLayout
         """
         self.setup_attachments()
+        empty_widget = QWidget()
 
-        self.window.ui.nodes['attachments_uploaded.sync.tip'] = QLabel(trans('attachments_uploaded.sync.tip'))
+        self.window.ui.nodes['attachments_uploaded.sync.tip'] = HelpLabel(trans('attachments_uploaded.sync.tip'))
+        self.window.ui.nodes['attachments_uploaded.sync.tip'].setWordWrap(False)
         self.window.ui.nodes['attachments_uploaded.sync.tip'].setAlignment(Qt.AlignCenter)
-        empty_widget = QWidget()
 
         self.window.ui.nodes['tip.input.attachments.uploaded'] = HelpLabel(trans('tip.input.attachments.uploaded'),
                                                                            self.window)
@@ -52,6 +54,8 @@ class AttachmentsUploaded:
         buttons_layout.addWidget(self.window.ui.nodes['attachments_uploaded.btn.clear'])
         buttons_layout.addWidget(empty_widget)
         buttons_layout.addWidget(self.window.ui.nodes['attachments_uploaded.sync.tip'])
+        buttons_layout.addStretch()
+
 
         # layout
         layout = QVBoxLayout()
@@ -70,7 +74,7 @@ class AttachmentsUploaded:
 
         # buttons
         self.window.ui.nodes['attachments_uploaded.btn.sync'] = SyncButton(trans('attachments_uploaded.btn.sync'), self.window)
-        self.window.ui.nodes['attachments_uploaded.btn.clear'] = QPushButton(trans('attachments_uploaded.btn.clear'))
+        self.window.ui.nodes['attachments_uploaded.btn.clear'] = QPushButton(QIcon(":/icons/close.svg"), trans('attachments_uploaded.btn.clear'))
         self.window.ui.nodes['attachments_uploaded.btn.clear'].clicked.connect(
             lambda: self.window.controller.assistant.files.clear()
         )

pygpt_net/ui/widget/anims/toggles.py

@@ -26,8 +26,9 @@ class AnimToggle(QCheckBox):
     _light_grey_pen = QPen(Qt.lightGray)
 
     def __init__(self, title="", parent=None):
-        super().__init__(parent)
+        super().__init__()
         self.setText(title)
+        self.option = None
 
         # Initialize the colors based on the current palette
         self.updateColors()
@@ -103,7 +104,6 @@ class AnimToggle(QCheckBox):
         self.animations_group.start()
 
     def paintEvent(self, e: QPaintEvent):
-
         contRect = self.contentsRect()
         handleRadius = round(0.24 * contRect.height())
 

pygpt_net/ui/widget/dialog/url.py

@@ -0,0 +1,59 @@
+#!/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.26 02:00:00                  #
+# ================================================== #
+
+from PySide6.QtWidgets import QDialog, QLabel, QHBoxLayout, QVBoxLayout, QPushButton
+
+from pygpt_net.ui.widget.element.labels import HelpLabel
+from pygpt_net.utils import trans
+from pygpt_net.ui.widget.textarea.url import UrlInput
+
+
+class UrlDialog(QDialog):
+    def __init__(self, window=None, id=None):
+        """
+        Url dialog
+
+        :param window: main window
+        :param id: info window id
+        """
+        super(UrlDialog, self).__init__(window)
+        self.window = window
+        self.id = id
+        self.current = None
+        self.input = UrlInput(window, id)
+        self.input.setMinimumWidth(400)
+
+        self.window.ui.nodes['dialog.url.btn.update'] = QPushButton(trans('dialog.url.update'))
+        self.window.ui.nodes['dialog.url.btn.update'].clicked.connect(
+            lambda: self.window.controller.dialogs.confirm.accept_url(
+                self.id,
+                self.window.ui.dialog['url'].current,
+                self.input.text()),
+        )
+
+        self.window.ui.nodes['dialog.url.btn.dismiss'] = QPushButton(trans('dialog.url.dismiss'))
+        self.window.ui.nodes['dialog.url.btn.dismiss'].clicked.connect(
+            lambda: self.window.controller.dialogs.confirm.dismiss_url())
+
+        bottom = QHBoxLayout()
+        bottom.addWidget(self.window.ui.nodes['dialog.url.btn.dismiss'])
+        bottom.addWidget(self.window.ui.nodes['dialog.url.btn.update'])
+
+        self.window.ui.nodes['dialog.url.label'] = QLabel(trans("dialog.url.title"))
+        self.window.ui.nodes['dialog.url.tip'] = HelpLabel(trans("dialog.url.tip"))
+
+        layout = QVBoxLayout()
+        layout.addWidget(self.window.ui.nodes['dialog.url.label'])
+        layout.addWidget(self.input)
+        layout.addWidget(self.window.ui.nodes['dialog.url.tip'])
+        layout.addLayout(bottom)
+
+        self.setLayout(layout)

pygpt_net/ui/widget/lists/attachment.py

@@ -6,13 +6,14 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.23 00:00:00                  #
+# Updated Date: 2024.11.26 02:00:00                  #
 # ================================================== #
 
 from PySide6.QtCore import Qt
 from PySide6.QtGui import QAction, QIcon, QResizeEvent, QImage
 from PySide6.QtWidgets import QMenu, QApplication, QHeaderView
 
+from pygpt_net.item.attachment import AttachmentItem
 from pygpt_net.ui.widget.lists.base import BaseList
 from pygpt_net.utils import trans
 import pygpt_net.icons_rc
@@ -91,9 +92,12 @@ class AttachmentList(BaseList):
         idx = item.row()
         preview_actions = []
         path = None
+        attachment = None
 
         if idx >= 0:
-            path = self.window.controller.attachment.get_path_by_idx(mode, idx)
+            attachment = self.window.controller.attachment.get_by_idx(mode, idx)
+            if attachment:
+                path = attachment.path
             preview_actions = []
             if self.window.core.filesystem.actions.has_preview(path):
                 preview_actions = self.window.core.filesystem.actions.get_preview(self, path)
@@ -116,21 +120,22 @@ class AttachmentList(BaseList):
             lambda: self.action_delete(event))
 
         menu = QMenu(self)
-        if idx >= 0 and preview_actions:
-            for action in preview_actions:
-                menu.addAction(action)
-        menu.addAction(actions['open'])
-        menu.addAction(actions['open_dir'])
-
-        if idx >= 0:
-            if self.window.core.filesystem.actions.has_use(path):
-                use_actions = self.window.core.filesystem.actions.get_use(self, path)
-                use_menu = QMenu(trans('action.use'), self)
-                for action in use_actions:
-                    use_menu.addAction(action)
-                menu.addMenu(use_menu)
-
-        menu.addAction(actions['rename'])
+        if attachment and attachment.type == AttachmentItem.TYPE_FILE:
+            if idx >= 0 and preview_actions:
+                for action in preview_actions:
+                    menu.addAction(action)
+
+            menu.addAction(actions['open'])
+            menu.addAction(actions['open_dir'])
+            if idx >= 0:
+                if self.window.core.filesystem.actions.has_use(path):
+                    use_actions = self.window.core.filesystem.actions.get_use(self, path)
+                    use_menu = QMenu(trans('action.use'), self)
+                    for action in use_actions:
+                        use_menu.addAction(action)
+                    menu.addMenu(use_menu)
+
+            menu.addAction(actions['rename'])
         menu.addAction(actions['delete'])
 
         if idx >= 0:

pygpt_net/ui/widget/lists/attachment_ctx.py

@@ -85,18 +85,47 @@ class AttachmentCtxList(BaseList):
         """
         actions = {}
 
+        item = self.indexAt(event.pos())
+        idx = item.row()
+
+        has_file = False
+        has_src = False
+        has_dest = False
+
+        if idx >= 0:
+            has_file = self.window.controller.chat.attachment.has_file_by_idx(idx)
+            has_src = self.window.controller.chat.attachment.has_src_by_idx(idx)
+            has_dest = self.window.controller.chat.attachment.has_dest_by_idx(idx)
+
+        actions['open'] = QAction(QIcon(":/icons/view.svg"), trans('action.open'), self)
+        actions['open'].triggered.connect(
+            lambda: self.action_open(event)
+        )
+        actions['open_dir_src'] = QAction(QIcon(":/icons/folder.svg"), trans('action.open_dir_src'), self)
+        actions['open_dir_src'].triggered.connect(
+            lambda: self.action_open_dir_src(event)
+        )
+        actions['open_dir_dest'] = QAction(QIcon(":/icons/folder.svg"), trans('action.open_dir_storage'), self)
+        actions['open_dir_dest'].triggered.connect(
+            lambda: self.action_open_dir_dest(event)
+        )
+
         actions['delete'] = QAction(QIcon(":/icons/delete.svg"), trans('action.delete'), self)
         actions['delete'].triggered.connect(
             lambda: self.action_delete(event)
         )
 
         menu = QMenu(self)
+        if has_file:
+            menu.addAction(actions['open'])
+        if has_src:
+            menu.addAction(actions['open_dir_src'])
+        if has_dest:
+            menu.addAction(actions['open_dir_dest'])
         menu.addAction(actions['delete'])
 
-        item = self.indexAt(event.pos())
-        idx = item.row()
         if idx >= 0:
-            self.window.controller.assistant.files.select(item.row())
+            self.window.controller.chat.attachment.select(item.row())
             menu.exec_(event.globalPos())
 
     def action_delete(self, event):
@@ -109,3 +138,36 @@ class AttachmentCtxList(BaseList):
         idx = item.row()
         if idx >= 0:
             self.window.controller.chat.attachment.delete_by_idx(idx)
+
+    def action_open(self, event):
+        """
+        Open action handler
+
+        :param event: mouse event
+        """
+        item = self.indexAt(event.pos())
+        idx = item.row()
+        if idx >= 0:
+            self.window.controller.chat.attachment.open_by_idx(idx)
+
+    def action_open_dir_src(self, event):
+        """
+        Open source directory action handler
+
+        :param event: mouse event
+        """
+        item = self.indexAt(event.pos())
+        idx = item.row()
+        if idx >= 0:
+            self.window.controller.chat.attachment.open_dir_src_by_idx(idx)
+
+    def action_open_dir_dest(self, event):
+        """
+        Open destination directory action handler
+
+        :param event: mouse event
+        """
+        item = self.indexAt(event.pos())
+        idx = item.row()
+        if idx >= 0:
+            self.window.controller.chat.attachment.open_dir_dest_by_idx(idx)

pygpt_net/ui/widget/option/checkbox.py

@@ -8,6 +8,7 @@
 # Created By  : Marcin Szczygliński                  #
 # Updated Date: 2024.11.24 22:00:00                  #
 # ================================================== #
+
 from PySide6.QtGui import QIcon
 from PySide6.QtWidgets import QCheckBox, QHBoxLayout, QWidget, QLabel
 
@@ -67,10 +68,7 @@ class OptionCheckbox(QWidget):
                 self.box.isChecked()
             )
         )
-
         self.label = QLabel(self.title)
-        self.box = AnimToggle()
-
         self.layout = QHBoxLayout()
         self.layout.addWidget(self.box)
 

pygpt_net/ui/widget/option/toggle.py

@@ -47,6 +47,7 @@ class OptionCheckbox(QWidget):
         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,

pygpt_net/ui/widget/textarea/url.py

@@ -0,0 +1,43 @@
+#!/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.26 02:00:00                  #
+# ================================================== #
+
+from PySide6 import QtCore
+from PySide6.QtWidgets import QLineEdit
+
+
+class UrlInput(QLineEdit):
+    def __init__(self, window=None, id=None):
+        """
+        Url dialog input
+
+        :param window: main window
+        :param id: info window id
+        """
+        super(UrlInput, self).__init__(window)
+
+        self.window = window
+        self.id = id
+
+    def keyPressEvent(self, event):
+        """
+        Key press event
+
+        :param event: key event
+        """
+        super(UrlInput, self).keyPressEvent(event)
+
+        # save on Enter
+        if event.key() == QtCore.Qt.Key_Return or event.key() == QtCore.Qt.Key_Enter:
+            self.window.controller.dialogs.confirm.accept_url(
+                self.window.ui.dialog['url'].id,
+                self.window.ui.dialog['url'].current,
+                self.text(),
+            )