pygpt-net 2.6.64__py3-none-any.whl → 2.6.66__py3-none-any.whl

pygpt_net/provider/vector_stores/qdrant.py

@@ -0,0 +1,117 @@
+#!/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: 2025.09.30 13:00:00                  #
+# ================================================== #
+
+import datetime
+import os.path
+from typing import Optional
+
+from llama_index.core.indices.base import BaseIndex
+from llama_index.core import StorageContext
+
+from pygpt_net.utils import parse_args
+from .base import BaseStore
+
+
+class QdrantProvider(BaseStore):
+    def __init__(self, *args, **kwargs):
+        super(QdrantProvider, self).__init__(*args, **kwargs)
+        """
+        Qdrant vector store provider
+
+        :param args: args
+        :param kwargs: kwargs
+        """
+        self.window = kwargs.get('window', None)
+        self.id = "QdrantVectorStore"
+        self.prefix = "qdrant_"  # prefix for index directory
+        self.indexes = {}
+
+    def create(self, id: str):
+        """
+        Create empty index
+
+        :param id: index name
+        """
+        path = self.get_path(id)
+        if not os.path.exists(path):
+            os.makedirs(path, exist_ok=True)
+            self.store(id)
+
+    def get_qdrant_store(self, id: str):
+        """
+        Get Qdrant vector store
+
+        :param id: index name
+        :return: QdrantVectorStore instance
+        """
+        from llama_index.vector_stores.qdrant import QdrantVectorStore
+        
+        additional_args = parse_args(
+            self.window.core.config.get('llama.idx.storage.args', []),
+        )
+        
+        url = additional_args.get('url', 'http://localhost:6333')
+        api_key = additional_args.get('api_key', '')
+        
+        store_args = {k: v for k, v in additional_args.items() if k not in ['url', 'api_key', 'collection_name']}
+        
+        return QdrantVectorStore(
+            url=url,
+            api_key=api_key,
+            collection_name=id,
+            **store_args
+        )
+
+    def get(
+            self,
+            id: str,
+            llm: Optional = None,
+            embed_model: Optional = None,
+    ) -> BaseIndex:
+        """
+        Get index
+
+        :param id: index name
+        :param llm: LLM instance
+        :param embed_model: Embedding model instance
+        :return: index instance
+        """
+        if not self.exists(id):
+            self.create(id)
+        vector_store = self.get_qdrant_store(id)
+        storage_context = StorageContext.from_defaults(
+            vector_store=vector_store,
+        )
+        self.indexes[id] = self.index_from_store(
+            vector_store=vector_store,
+            storage_context=storage_context,
+            llm=llm,
+            embed_model=embed_model,
+        )
+        return self.indexes[id]
+
+    def store(
+            self,
+            id: str,
+            index: Optional[BaseIndex] = None
+    ):
+        """
+        Store index
+
+        :param id: index name
+        :param index: index instance
+        """
+        path = self.get_path(id)
+        os.makedirs(path, exist_ok=True)
+        lock_file = os.path.join(path, 'store.lock')
+        with open(lock_file, 'w') as f:
+            f.write(id + ': ' + str(datetime.datetime.now()))
+        self.indexes[id] = index

pygpt_net/ui/__init__.py

@@ -113,6 +113,8 @@ class UI:
         """Set default sizes"""
         def set_initial_splitter_height():
             """Set initial splitter height"""
+            if 'main.output' not in self.window.ui.splitters:
+                return
             total_height = self.window.ui.splitters['main.output'].size().height()
             if total_height > 0:
                 size_output = int(total_height * 0.9)
@@ -124,6 +126,8 @@ class UI:
 
         def set_initial_splitter_width():
             """Set initial splitter width"""
+            if 'main' not in self.window.ui.splitters:
+                return
             total_width = self.window.ui.splitters['main'].size().width()
             if total_width > 0:
                 size_output = int(total_width * 0.75)
@@ -139,7 +143,8 @@ class UI:
         suffix = self.window.core.platforms.get_env_suffix()
         profile_name = self.window.core.config.profile.get_current_name()
         self.window.setWindowTitle(
-            f"PyGPT - Desktop AI Assistant {self.window.meta['version']} | build {self.window.meta['build'].replace('.', '-')}{suffix} ({profile_name})"
+            f"PyGPT - Desktop AI Assistant {self.window.meta['version']} | "
+            f"build {self.window.meta['build'].replace('.', '-')}{suffix} ({profile_name})"
         )
 
     def post_setup(self):

pygpt_net/ui/dialog/preset.py

@@ -6,13 +6,13 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.09.25 13:04:51                  #
+# Updated Date: 2025.09.28 08:00:00                  #
 # ================================================== #
 
 from PySide6.QtCore import Qt
 from PySide6.QtGui import QIcon
 from PySide6.QtWidgets import QPushButton, QHBoxLayout, QLabel, QVBoxLayout, QSplitter, QWidget, QSizePolicy, \
-    QTabWidget, QFileDialog
+    QTabWidget, QFileDialog, QScrollArea, QFrame
 
 from pygpt_net.core.types import (
     MODE_AGENT,
@@ -169,7 +169,6 @@ class Preset(BaseConfigDialog):
         desc.setContentsMargins(0, 5, 0, 5)
         self.window.ui.nodes['preset.editor.description'] = desc
 
-
         # prompt + extra options
         prompt_layout = QVBoxLayout()
         prompt_layout.addWidget(widgets['prompt'])
@@ -334,8 +333,14 @@ class Preset(BaseConfigDialog):
         self.window.ui.nodes['preset.editor.extra'] = {}
 
         tabs = QTabWidget()
+
+        # Make the prompt tab scrollable to avoid vertical overlap in narrow layouts.
+        scroll_prompt = QScrollArea()
+        scroll_prompt.setWidget(prompt_widget)
+        scroll_prompt.setWidgetResizable(True)
+        scroll_prompt.setFrameShape(QFrame.NoFrame)
         tabs.addTab(
-            prompt_widget,
+            scroll_prompt,
             trans("preset.prompt"),
         )
         tabs.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)

pygpt_net/ui/layout/chat/attachments.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.24 23:00:00                  #
+# Updated Date: 2025.09.28 08:00:00                  #
 # ================================================== #
 
 import os
@@ -19,8 +19,10 @@ from PySide6.QtWidgets import QVBoxLayout, QPushButton, QHBoxLayout, QCheckBox,
 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.core.attachments.clipboard import AttachmentDropHandler
 from pygpt_net.utils import trans
 
+
 class Attachments:
     def __init__(self, window=None):
         """
@@ -30,6 +32,8 @@ class Attachments:
         """
         self.window = window
         self.id = 'attachments'
+        # Keep a strong reference to DnD handler(s)
+        self._dnd_handlers = {}
 
     def setup(self) -> QVBoxLayout:
         """
@@ -132,6 +136,19 @@ class Attachments:
         self.window.ui.models[self.id] = self.create_model(self.window)
         self.window.ui.nodes[self.id].setModel(self.window.ui.models[self.id])
 
+        # Drag & Drop: allow dropping files/images/urls/text directly onto the list
+        try:
+            self._dnd_handlers[self.id] = AttachmentDropHandler(
+                self.window,
+                self.window.ui.nodes[self.id],
+                policy=AttachmentDropHandler.SWALLOW_ALL,
+            )
+        except Exception as e:
+            try:
+                self.window.core.debug.log(e)
+            except Exception:
+                pass
+
     def create_model(self, parent) -> QStandardItemModel:
         """
         Create list model

pygpt_net/ui/layout/status.py

@@ -14,6 +14,7 @@ from PySide6.QtWidgets import QLabel, QHBoxLayout, QSizePolicy, QPushButton
 
 from pygpt_net.ui.widget.element.labels import HelpLabel
 from pygpt_net.ui.widget.anims.loader import Loader
+from pygpt_net.ui.widget.element.status import BottomStatus
 from pygpt_net.utils import trans
 
 
@@ -34,8 +35,7 @@ class Status:
         """
         nodes = self.window.ui.nodes
 
-        nodes['status'] = QLabel(trans('status.started'))
-        nodes['status'].setParent(self.window)
+        nodes['status'] = BottomStatus(window=self.window)
 
         nodes['status.agent'] = HelpLabel("")
         nodes['status.agent'].setParent(self.window)
@@ -53,7 +53,7 @@ class Status:
         layout = QHBoxLayout()
         layout.addWidget(nodes['anim.loading.status'])
         layout.addWidget(nodes['status.agent'])
-        layout.addWidget(nodes['status'])
+        layout.addWidget(nodes['status'].setup())
         layout.addWidget(nodes['global.stop'])
         layout.setAlignment(Qt.AlignLeft)
 

pygpt_net/ui/layout/toolbox/raw.py

@@ -6,11 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.09.01 23:00:00                  #
+# Updated Date: 2025.12.25 20:00:00                  #
 # ================================================== #
 
 from PySide6.QtWidgets import QVBoxLayout, QHBoxLayout, QWidget, QCheckBox
 
+from pygpt_net.ui.widget.option.combo import OptionCombo
 from pygpt_net.utils import trans
 
 
@@ -39,7 +40,12 @@ class Raw:
         conf_global['img_raw'] = QCheckBox(trans("img.raw"), parent=container)
         conf_global['img_raw'].toggled.connect(self.window.controller.media.toggle_raw)
 
+        conf_global = ui.config['global']
+        option_modes = self.window.core.image.get_available_modes()
+        conf_global['img_mode'] = OptionCombo(self.window, 'global', 'img_mode', option_modes)
+
         cols = QHBoxLayout()
+        cols.addWidget(conf_global['img_mode'])
         cols.addWidget(conf_global['img_raw'])
 
         rows = QVBoxLayout()

pygpt_net/ui/widget/element/status.py

@@ -0,0 +1,55 @@
+#!/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: 2025.09.28 00:00:00                  #
+# ================================================== #
+
+from PySide6.QtWidgets import QLabel, QHBoxLayout, QWidget
+
+from datetime import datetime
+from pygpt_net.utils import trans
+
+
+class BottomStatus:
+    def __init__(self, window=None):
+        self.window = window
+        self.timer = QLabel(parent=self.window)
+        self.timer.setObjectName("StatusBarTimer")
+        self.msg = QLabel(parent=self.window)
+        self.msg.setObjectName("StatusBarMessage")
+        self.set_text(trans('status.started'))
+
+    def set_text(self, text):
+        """Set status text"""
+        self.msg.setText(text)
+        if text:
+            now = datetime.now()
+            self.timer.setText(now.strftime("%H:%M"))
+        else:
+            self.timer.setText("")
+
+    def setText(self, text):
+        """Fallback for set_text method"""
+        self.set_text(text)
+
+    def text(self) -> str:
+        """Get status text"""
+        return self.msg.text()
+
+    def setup(self):
+        """Setup status bar widget"""
+        self.timer.setText("00:00")
+        layout = QHBoxLayout()
+        layout.setContentsMargins(0, 0, 0, 0)
+        layout.setSpacing(5)
+        layout.addWidget(self.timer)
+        layout.addWidget(self.msg)
+        layout.addStretch()
+        widget = QWidget(self.window)
+        widget.setLayout(layout)
+        return widget

pygpt_net/ui/widget/filesystem/explorer.py

@@ -6,13 +6,13 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.24 23:00:00                  #
+# Updated Date: 2025.09.28 08:00:00                  #
 # ================================================== #
 
 import datetime
 import os
 
-from PySide6.QtCore import Qt, QModelIndex, QDir
+from PySide6.QtCore import Qt, QModelIndex, QDir, QObject, QEvent
 from PySide6.QtGui import QAction, QIcon, QCursor, QResizeEvent
 from PySide6.QtWidgets import QTreeView, QMenu, QWidget, QVBoxLayout, QFileSystemModel, QLabel, QHBoxLayout, \
     QPushButton, QSizePolicy
@@ -23,6 +23,117 @@ from pygpt_net.ui.widget.element.labels import HelpLabel
 from pygpt_net.utils import trans
 
 
+class ExplorerDropHandler(QObject):
+    """
+    Drag & drop handler for FileExplorer (uploads into target directory).
+    - Accepts local file and directory URLs.
+    - Determines target directory from drop position:
+      * directory item -> that directory
+      * file item      -> parent directory
+      * empty area     -> explorer root directory
+    - Uses Files controller to perform the actual copy and refresh view.
+    """
+    def __init__(self, explorer):
+        super().__init__(explorer)
+        self.explorer = explorer
+        self.view = explorer.treeView
+
+        # Enable drops on both the view and its viewport
+        try:
+            self.view.setAcceptDrops(True)
+        except Exception:
+            pass
+        vp = self.view.viewport()
+        if vp is not None:
+            try:
+                vp.setAcceptDrops(True)
+            except Exception:
+                pass
+            vp.installEventFilter(self)
+        self.view.installEventFilter(self)
+
+    def _mime_has_local_urls(self, md) -> bool:
+        try:
+            if md and md.hasUrls():
+                for url in md.urls():
+                    if url.isLocalFile():
+                        return True
+        except Exception:
+            pass
+        return False
+
+    def _local_paths_from_mime(self, md) -> list:
+        out = []
+        try:
+            if not (md and md.hasUrls()):
+                return out
+            for url in md.urls():
+                try:
+                    if url.isLocalFile():
+                        p = url.toLocalFile()
+                        if p:
+                            out.append(p)
+                except Exception:
+                    continue
+        except Exception:
+            pass
+        return out
+
+    def _target_dir_from_pos(self, event) -> str:
+        # QDropEvent in Qt6: use position() -> QPointF
+        try:
+            pos = event.position().toPoint()
+        except Exception:
+            pos = event.pos()
+        idx = self.view.indexAt(pos)
+        if idx.isValid():
+            path = self.explorer.model.filePath(idx)
+            if os.path.isdir(path):
+                return path
+            return os.path.dirname(path)
+        # Fallback: explorer root directory
+        return self.explorer.directory
+
+    def eventFilter(self, obj, event):
+        et = event.type()
+
+        if et in (QEvent.DragEnter, QEvent.DragMove):
+            md = getattr(event, 'mimeData', lambda: None)()
+            if self._mime_has_local_urls(md):
+                try:
+                    event.setDropAction(Qt.CopyAction)
+                    event.acceptProposedAction()
+                except Exception:
+                    event.accept()
+                return True
+            return False
+
+        if et == QEvent.Drop:
+            md = getattr(event, 'mimeData', lambda: None)()
+            if not self._mime_has_local_urls(md):
+                return False
+
+            paths = self._local_paths_from_mime(md)
+            target_dir = self._target_dir_from_pos(event)
+            try:
+                self.explorer.window.controller.files.upload_paths(paths, target_dir)
+            except Exception as e:
+                try:
+                    self.explorer.window.core.debug.log(e)
+                except Exception:
+                    pass
+
+            try:
+                event.setDropAction(Qt.CopyAction)
+                event.acceptProposedAction()
+            except Exception:
+                event.accept()
+            # Swallow so the view does not try to handle the drop itself
+            return True
+
+        return False
+
+
 class FileExplorer(QWidget):
     def __init__(self, window, directory, index_data):
         """
@@ -138,6 +249,9 @@ class FileExplorer(QWidget):
             'db': QIcon(":/icons/db.svg"),
         }
 
+        # Drag & Drop upload support
+        self._dnd_handler = ExplorerDropHandler(self)
+
     def eventFilter(self, source, event):
         """
         Focus event filter

pygpt_net/ui/widget/lists/context.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.09.15 22:00:00                  #
+# Updated Date: 2025.09.28 00:00:00                  #
 # ================================================== #
 
 import datetime
@@ -40,7 +40,7 @@ class ContextList(BaseList):
             'chat': QIcon(":/icons/chat.svg"),
             'copy': QIcon(":/icons/copy.svg"),
             'close': QIcon(":/icons/close.svg"),
-            'pin': QIcon(":/icons/pin.svg"),
+            'pin': QIcon(":/icons/pin3.svg"),
             'clock': QIcon(":/icons/clock.svg"),
             'db': QIcon(":/icons/db.svg"),
             'folder': QIcon(":/icons/folder_filled.svg"),
@@ -49,7 +49,8 @@ class ContextList(BaseList):
         self._color_icon_cache = {}
 
         # Use a custom delegate for labels/pinned/attachment indicators and group border indicator
-        self.setItemDelegate(ImportantItemDelegate(self, self._icons['attachment']))
+        # Pass both: attachment icon and pin icon (pin2.svg) for pinned indicator rendering
+        self.setItemDelegate(ImportantItemDelegate(self, self._icons['attachment'], self._icons['pin']))
 
         # Ensure context menu works as before
         self.setContextMenuPolicy(Qt.CustomContextMenu)
@@ -467,15 +468,17 @@ class ImportantItemDelegate(QtWidgets.QStyledItemDelegate):
     """
     Item delegate that paints:
     - Attachment icon on the right side (centered vertically),
-    - Pinned indicator (small circle) in the top-right corner (overlays if needed),
+    - Pinned indicator (pin.svg icon) in the top-right corner (overlays if needed),
     - Label color as a full-height vertical bar on the left for labeled items,
     - Group enclosure indicator for expanded groups:
         - thin vertical bar (default 2 px) on the left side of child rows area,
         - thin horizontal bar (default 2 px) at the bottom of the last child row.
     """
-    def __init__(self, parent=None, attachment_icon: QIcon = None):
+    def __init__(self, parent=None, attachment_icon: QIcon = None, pin_icon: QIcon = None):
         super().__init__(parent)
         self._attachment_icon = attachment_icon or QIcon(":/icons/attachment.svg")
+        # Use provided pin icon (transparent background) as pinned indicator
+        self._pin_icon = pin_icon or QIcon(":/icons/pin.svg")
 
         # Predefined label colors (status -> QColor)
         self._status_colors = {
@@ -490,8 +493,8 @@ class ImportantItemDelegate(QtWidgets.QStyledItemDelegate):
         }
 
         # Visual tuning constants
-        self._pin_pen = QtGui.QPen(QtCore.Qt.black, 0.5, QtCore.Qt.SolidLine)
-        self._pin_diameter = 4            # Small pinned circle diameter
+        self._pin_pen = QtGui.QPen(QtCore.Qt.black, 0.5, QtCore.Qt.SolidLine)  # kept for compatibility
+        self._pin_diameter = 4            # legacy circle diameter (not used anymore)
         self._pin_margin = 3              # Margin from top and right edges
         self._attach_spacing = 4          # Kept for potential future layout tweaks
         self._label_bar_width = 4         # Full-height label bar width (left side)
@@ -507,6 +510,10 @@ class ImportantItemDelegate(QtWidgets.QStyledItemDelegate):
         self._group_indicator_gap = 6  # gap between child content left and the vertical bar
         self._group_indicator_bottom_offset = 6
 
+        # Pinned icon sizing (kept deliberately small, similar to previous yellow dot)
+        # The actual painted size is min(max_size, availableHeightWithMargins)
+        self._pin_icon_max_size = 12  # px
+
         # Try to load customization from application config (safe if missing)
         self._init_group_indicator_from_config()
 
@@ -665,20 +672,23 @@ class ImportantItemDelegate(QtWidgets.QStyledItemDelegate):
                     )
                     self._attachment_icon.paint(painter, icon_rect, QtCore.Qt.AlignCenter)
 
-                # Pinned indicator (small circle) kept at a fixed top-right position.
-                # It does not shift left when the attachment is present; it overlays above it.
+                # Pinned indicator: small pin.svg painted at fixed top-right position.
+                # It overlays above any other right-side icons.
                 if is_important:
                     painter.setRenderHint(QtGui.QPainter.Antialiasing, True)
                     painter.setCompositionMode(QtGui.QPainter.CompositionMode_SourceOver)
-                    color = self.get_color_for_status(3)
 
-                    x = option.rect.x() + option.rect.width() - self._pin_margin - self._pin_diameter
-                    y = option.rect.y() + self._pin_margin
-                    pin_rect = QtCore.QRect(x, y, self._pin_diameter, self._pin_diameter)
+                    # Compute a compact size similar in footprint to previous circle,
+                    # but readable for vector icon; clamp to available height.
+                    available = max(8, option.rect.height() - 2 * self._pin_margin)
+                    pin_size = min(self._pin_icon_max_size, available)
 
-                    painter.setBrush(color)
-                    painter.setPen(self._pin_pen)
-                    painter.drawEllipse(pin_rect)
+                    x = option.rect.right() - self._pin_margin - pin_size
+                    y = option.rect.top() + self._pin_margin
+                    pin_rect = QtCore.QRect(x, y, pin_size, pin_size)
+
+                    # Paint the pin icon (transparent background)
+                    self._pin_icon.paint(painter, pin_rect, QtCore.Qt.AlignCenter)
 
                 # Label bar on the left with 3px vertical margins
                 if label > 0:

pygpt_net/ui/widget/option/checkbox_list.py

@@ -6,10 +6,10 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.24 23:00:00                  #
+# Updated Date: 2025.12.25 20:00:00                  #
 # ================================================== #
 
-from PySide6.QtWidgets import QCheckBox, QWidget
+from PySide6.QtWidgets import QCheckBox, QWidget, QPushButton
 
 from pygpt_net.ui.base.flow_layout import FlowLayout
 from pygpt_net.utils import trans
@@ -84,6 +84,18 @@ class OptionCheckboxList(QWidget):
         for widget in widgets:
             self.layout.addWidget(widget)
 
+        # select/unselect all button
+        btn_select = QPushButton("X", self)
+        btn_select.setToolTip(trans("action.select_unselect_all"))
+        btn_select.clicked.connect(
+            lambda: self.window.controller.config.checkbox_list.on_select_all(
+                self.parent_id,
+                self.id,
+                self.option
+            )
+        )
+        self.layout.addWidget(btn_select)
+
         self.layout.setContentsMargins(0, 0, 0, 0)
         self.setLayout(self.layout)