pygpt-net 2.6.33__py3-none-any.whl → 2.6.34__py3-none-any.whl

pygpt_net/ui/dialog/assistant_store.py

@@ -6,15 +6,15 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.06.29 18:00:00                  #
+# Updated Date: 2025.09.02 22:00:00                  #
 # ================================================== #
 
 import copy
 
-from PySide6.QtCore import Qt
+from PySide6.QtCore import Qt, QPoint
 from PySide6.QtGui import QStandardItemModel, QAction, QIcon
 from PySide6.QtWidgets import QPushButton, QHBoxLayout, QLabel, QVBoxLayout, QScrollArea, QWidget, QTabWidget, QFrame, \
-    QSplitter, QSizePolicy, QMenuBar, QCheckBox
+    QSplitter, QSizePolicy, QMenuBar, QCheckBox, QMenu, QListView
 
 from pygpt_net.ui.widget.dialog.assistant_store import AssistantVectorStoreDialog
 from pygpt_net.ui.widget.element.group import CollapsedGroup
@@ -46,122 +46,83 @@ class AssistantVectorStore:
 
         :param idx: current model tab index
         """
-        self.window.ui.nodes['assistant.store.btn.new'] = \
-            QPushButton(trans("dialog.assistant.store.btn.new"))
-        self.window.ui.nodes['assistant.store.btn.save'] = \
-            QPushButton(trans("dialog.assistant.store.btn.save"))
-        self.window.ui.nodes['assistant.store.btn.refresh_status'] = \
-            QPushButton(QIcon(":/icons/reload.svg"), "")
-        self.window.ui.nodes['assistant.store.btn.close'] = \
-            QPushButton(trans("dialog.assistant.store.btn.close"))
-
-        self.window.ui.nodes['assistant.store.btn.refresh_status'].setToolTip(
+        ui = self.window.ui
+        nodes = ui.nodes
+        models = ui.models
+        splitters = ui.splitters
+        controller = self.window.controller
+        core = self.window.core
+
+        nodes['assistant.store.btn.new'] = QPushButton(trans("dialog.assistant.store.btn.new"))
+        nodes['assistant.store.btn.save'] = QPushButton(trans("dialog.assistant.store.btn.save"))
+        nodes['assistant.store.btn.refresh_status'] = QPushButton(QIcon(":/icons/reload.svg"), "")
+        nodes['assistant.store.btn.close'] = QPushButton(trans("dialog.assistant.store.btn.close"))
+
+        nodes['assistant.store.btn.refresh_status'].setToolTip(
             trans("dialog.assistant.store.btn.refresh_status")
         )
-        self.window.ui.nodes['assistant.store.btn.refresh_status'].setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
-
-        self.window.ui.nodes['assistant.store.btn.upload.files'] = \
-            QPushButton(trans("dialog.assistant.store.btn.upload.files"))
-        self.window.ui.nodes['assistant.store.btn.upload.dir'] = \
-            QPushButton(trans("dialog.assistant.store.btn.upload.dir"))
-
-        self.window.ui.nodes['assistant.store.hide_thread'] = QCheckBox(trans("assistant.store.hide_threads"))
-        self.window.ui.nodes['assistant.store.hide_thread'].setChecked(True)
-        self.window.ui.nodes['assistant.store.hide_thread'].stateChanged.connect(
-            lambda: self.window.controller.assistant.store.set_hide_thread(
-                self.window.ui.nodes['assistant.store.hide_thread'].isChecked()
-            )
-        )
-        if not self.window.core.config.get("assistant.store.hide_threads"):
-            self.window.ui.nodes['assistant.store.hide_thread'].setChecked(False)
+        nodes['assistant.store.btn.refresh_status'].setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
 
-        self.window.ui.nodes['assistant.store.btn.new'].clicked.connect(
-            lambda: self.window.controller.assistant.store.new()
-        )
-        self.window.ui.nodes['assistant.store.btn.save'].clicked.connect(
-            lambda: self.window.controller.assistant.store.save()
-        )
-        self.window.ui.nodes['assistant.store.btn.close'].clicked.connect(
-            lambda: self.window.controller.assistant.store.close()
-        )
-        self.window.ui.nodes['assistant.store.btn.refresh_status'].clicked.connect(
-            lambda: self.window.controller.assistant.store.refresh_status()
-        )
-        self.window.ui.nodes['assistant.store.btn.upload.files'].clicked.connect(
-            lambda: self.window.controller.assistant.batch.open_upload_files()
-        )
-        self.window.ui.nodes['assistant.store.btn.upload.dir'].clicked.connect(
-            lambda: self.window.controller.assistant.batch.open_upload_dir()
+        nodes['assistant.store.btn.upload.files'] = QPushButton(trans("dialog.assistant.store.btn.upload.files"))
+        nodes['assistant.store.btn.upload.dir'] = QPushButton(trans("dialog.assistant.store.btn.upload.dir"))
+
+        nodes['assistant.store.hide_thread'] = QCheckBox(trans("assistant.store.hide_threads"))
+        nodes['assistant.store.hide_thread'].setChecked(bool(core.config.get("assistant.store.hide_threads")))
+        nodes['assistant.store.hide_thread'].toggled.connect(
+            controller.assistant.store.set_hide_thread
         )
 
-        # set enter key to save button
-        self.window.ui.nodes['assistant.store.btn.new'].setAutoDefault(False)
-        self.window.ui.nodes['assistant.store.btn.refresh_status'].setAutoDefault(False)
-        self.window.ui.nodes['assistant.store.btn.save'].setAutoDefault(True)
+        nodes['assistant.store.btn.new'].clicked.connect(controller.assistant.store.new)
+        nodes['assistant.store.btn.save'].clicked.connect(controller.assistant.store.save_btn)
+        nodes['assistant.store.btn.close'].clicked.connect(controller.assistant.store.close)
+        nodes['assistant.store.btn.refresh_status'].clicked.connect(controller.assistant.store.refresh_status)
+        nodes['assistant.store.btn.upload.files'].clicked.connect(controller.assistant.batch.open_upload_files)
+        nodes['assistant.store.btn.upload.dir'].clicked.connect(controller.assistant.batch.open_upload_dir)
 
-        # footer buttons
-        footer = QHBoxLayout()
-        footer.addWidget(self.window.ui.nodes['assistant.store.btn.close'])
-        footer.addWidget(self.window.ui.nodes['assistant.store.btn.save'])
+        nodes['assistant.store.btn.new'].setAutoDefault(False)
+        nodes['assistant.store.btn.refresh_status'].setAutoDefault(False)
+        nodes['assistant.store.btn.save'].setAutoDefault(True)
 
-        # upload buttons
-        upload_layout = QHBoxLayout()
-        upload_layout.addWidget(self.window.ui.nodes['assistant.store.btn.refresh_status'])  # reload status btn
-        upload_layout.addWidget(self.window.ui.nodes['assistant.store.btn.upload.files'])
-        upload_layout.addWidget(self.window.ui.nodes['assistant.store.btn.upload.dir'])
-        upload_layout.setContentsMargins(0, 0, 0, 0)
+        footer = QHBoxLayout()
+        footer.addWidget(nodes['assistant.store.btn.close'])
+        footer.addWidget(nodes['assistant.store.btn.save'])
 
-        # editor tabs
-        self.window.ui.tabs['assistant.store'] = QTabWidget()
+        ui.tabs['assistant.store'] = QTabWidget()
 
-        # build settings tabs
         parent_id = "assistant.store"
 
         scroll = QScrollArea()
         scroll.setWidgetResizable(True)
         content = QVBoxLayout()
 
-        # create config options entry if not exists
-        if parent_id not in self.window.ui.config:
-            self.window.ui.config[parent_id] = {}
+        if parent_id not in ui.config:
+            ui.config[parent_id] = {}
 
-        # get options
-        options = copy.deepcopy(self.window.controller.assistant.store.get_options())
+        options = copy.deepcopy(controller.assistant.store.get_options())
         widgets = self.build_widgets(options)
-        advanced_keys = []
-        for key in options:
-            if 'advanced' in options[key] and options[key]['advanced']:
-                advanced_keys.append(key)
+        advanced_keys = [k for k, v in options.items() if v.get('advanced')]
 
-        # apply settings widgets
-        for key in widgets:
-            self.window.ui.config[parent_id][key] = widgets[key]
+        for key, w in widgets.items():
+            ui.config[parent_id][key] = w
 
         for key in widgets:
-            if key in advanced_keys:  # hide advanced options
+            if key in advanced_keys:
                 continue
-            content.addLayout(self.add_option(widgets[key], options[key]))  # add to scroll
+            content.addLayout(self.add_option(widgets[key], options[key]))
 
-        # append advanced options at the end
-        if len(advanced_keys) > 0:
+        if advanced_keys:
             group_id = 'assistant.store.advanced'
-            self.window.ui.groups[group_id] = CollapsedGroup(self.window, group_id, None, False, None)
-            self.window.ui.groups[group_id].box.setText(trans('settings.advanced.collapse'))
+            ui.groups[group_id] = CollapsedGroup(self.window, group_id, None, False, None)
+            ui.groups[group_id].box.setText(trans('settings.advanced.collapse'))
             for key in widgets:
-                if key not in advanced_keys:  # ignore non-advanced options
+                if key not in advanced_keys:
                     continue
+                option = self.add_option(widgets[key], options[key])
+                ui.groups[group_id].add_layout(option)
+            content.addWidget(ui.groups[group_id])
 
-                option = self.add_option(widgets[key], options[key])  # build option
-                self.window.ui.groups[group_id].add_layout(option)  # add option to group
-
-            # add advanced options group to scroll
-            content.addWidget(self.window.ui.groups[group_id])
-
-
-        content.addLayout(upload_layout)  # upload buttons
         content.setContentsMargins(0, 0, 0, 0)
 
-        # scroll widget
         scroll_widget = QWidget()
         scroll_widget.setLayout(content)
         scroll.setWidget(scroll_widget)
@@ -173,65 +134,93 @@ class AssistantVectorStore:
         area_widget = QWidget()
         area_widget.setLayout(area)
 
-        data = {}
-        for store_id in self.window.core.assistants.store.items.keys():
-            store = self.window.core.assistants.store.items[store_id]
-            data[store_id] = store
-
-        # storage list
-        id = 'assistant.store.list'
-        self.window.ui.nodes[id] = AssistantVectorStoreEditorList(self.window, id)
-        self.window.ui.models[id] = self.create_model(self.window)
-        self.window.ui.nodes[id].setModel(self.window.ui.models[id])
+        list_id = 'assistant.store.list'
+        nodes[list_id] = AssistantVectorStoreEditorList(self.window, list_id)
+        models[list_id] = self.create_model(self.window)
+        nodes[list_id].setModel(models[list_id])
+        nodes[list_id].setMinimumWidth(250)
 
-        # left widget
         left_layout = QVBoxLayout()
-        left_layout.addWidget(self.window.ui.nodes['assistant.store.btn.new'])
-        left_layout.addWidget(self.window.ui.nodes[id])
-        left_layout.addWidget(self.window.ui.nodes['assistant.store.hide_thread'])
-
+        left_layout.addWidget(nodes['assistant.store.btn.new'])
+        left_layout.addWidget(nodes[list_id])
+        left_layout.addWidget(nodes['assistant.store.hide_thread'])
         left_layout.setContentsMargins(0, 0, 0, 0)
         left_widget = QWidget()
         left_widget.setLayout(left_layout)
 
-        # update models list
-        self.update_list(id, data)
-
-        # set max width to list
-        self.window.ui.nodes[id].setMinimumWidth(250)
-
-        # splitter
-        self.window.ui.splitters['dialog.assistant.store'] = QSplitter(Qt.Horizontal)
-        self.window.ui.splitters['dialog.assistant.store'].addWidget(left_widget)  # list
-        self.window.ui.splitters['dialog.assistant.store'].addWidget(area_widget)  # tabs
-        self.window.ui.splitters['dialog.assistant.store'].setStretchFactor(0, 2)
-        self.window.ui.splitters['dialog.assistant.store'].setStretchFactor(1, 5)
-        self.window.ui.splitters['dialog.assistant.store'].setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
+        self.update_list(list_id, core.assistants.store.items)
+
+        # ==================== Files panel ====================
+        files_panel = QVBoxLayout()
+        files_label = QLabel("Files")
+        files_label.setStyleSheet("font-weight: bold;")
+        files_panel.addWidget(files_label)
+
+        files_list_id = 'assistant.store.files.list'
+        nodes[files_list_id] = QListView()
+        nodes[files_list_id].setEditTriggers(QListView.NoEditTriggers)
+        nodes[files_list_id].setSelectionMode(QListView.SingleSelection)
+        nodes[files_list_id].setContextMenuPolicy(Qt.CustomContextMenu)
+        nodes[files_list_id].customContextMenuRequested.connect(self.on_files_context_menu)
+
+        models[files_list_id] = self.create_model(self.window)
+        nodes[files_list_id].setModel(models[files_list_id])
+        nodes[files_list_id].setMinimumWidth(280)
+
+        files_bottom = QHBoxLayout()
+        files_bottom.setContentsMargins(0, 0, 0, 0)
+        files_bottom.addWidget(nodes['assistant.store.btn.upload.files'])
+        files_bottom.addWidget(nodes['assistant.store.btn.upload.dir'])
+        files_bottom.addWidget(nodes['assistant.store.btn.refresh_status'])
+
+        files_panel.addWidget(nodes[files_list_id])
+        files_panel.setContentsMargins(0, 0, 0, 5)
+        files_panel.addLayout(files_bottom)
+        files_widget = QWidget()
+        files_widget.setLayout(files_panel)
+        # ========================================================================
+
+        splitters['dialog.assistant.store'] = QSplitter(Qt.Horizontal)
+        splitters['dialog.assistant.store'].addWidget(left_widget)
+
+        splitters['dialog.assistant.store.right'] = QSplitter(Qt.Horizontal)
+        splitters['dialog.assistant.store.right'].addWidget(files_widget)
+        splitters['dialog.assistant.store.right'].addWidget(area_widget)
+        splitters['dialog.assistant.store.right'].setStretchFactor(1, 7)
+        splitters['dialog.assistant.store.right'].setStretchFactor(0, 3)
+
+        splitters['dialog.assistant.store'].addWidget(splitters['dialog.assistant.store.right'])
+        splitters['dialog.assistant.store'].setStretchFactor(0, 2)
+        splitters['dialog.assistant.store'].setStretchFactor(1, 8)
+        splitters['dialog.assistant.store'].setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
 
         main_layout = QHBoxLayout()
-        main_layout.addWidget(self.window.ui.splitters['dialog.assistant.store'])
+        main_layout.addWidget(splitters['dialog.assistant.store'])
         main_layout.setContentsMargins(0, 0, 0, 0)
 
         layout = QVBoxLayout()
-        layout.addLayout(main_layout)  # list + model tabs
-        layout.addLayout(footer)  # bottom buttons (save, defaults)
+        layout.addLayout(main_layout)
+        layout.addLayout(footer)
 
-        # add menu
         layout.setMenuBar(self.setup_menu())
 
-        self.window.ui.dialog[self.dialog_id] = AssistantVectorStoreDialog(self.window, self.dialog_id)
-        self.window.ui.dialog[self.dialog_id].setLayout(layout)
-        self.window.ui.dialog[self.dialog_id].setWindowTitle(trans('dialog.assistant.store'))
+        ui.dialog[self.dialog_id] = AssistantVectorStoreDialog(self.window, self.dialog_id)
+        ui.dialog[self.dialog_id].setLayout(layout)
+        ui.dialog[self.dialog_id].setWindowTitle(trans('dialog.assistant.store'))
 
-        # restore current opened tab if idx is set
         if idx is not None:
             try:
-                self.window.controller.assistant.store.set_by_tab(idx)
+                controller.assistant.store.set_by_tab(idx)
             except Exception as e:
                 print('Failed restore store editor tab: {}'.format(idx))
         else:
-            if self.window.controller.assistant.store.current is None:
-                self.window.controller.assistant.store.set_by_tab(0)
+            if controller.assistant.store.current is None:
+                controller.assistant.store.set_by_tab(0)
+
+        try:
+            controller.assistant.store.update_files_list()
+        except Exception:
+            pass
 
     def setup_menu(self) -> QMenuBar:
         """
@@ -246,107 +235,104 @@ class AssistantVectorStore:
         self.menu["current"] = self.menu_bar.addMenu(trans("dialog.assistant.store.menu.current"))
         self.menu["all"] = self.menu_bar.addMenu(trans("dialog.assistant.store.menu.all"))
 
-        # --------------------------------------------
-
-        # import files (current)
         self.actions["current.import_files"] = QAction(QIcon(":/icons/download.svg"),
-                                                   trans("dialog.assistant.store.menu.current.import_files"))
+                                                   trans("dialog.assistant.store.menu.current.import_files"),
+                                                   self.menu_bar)
         self.actions["current.import_files"].triggered.connect(
             lambda: self.window.controller.assistant.batch.import_store_files(
-                self.window.controller.assistant.store.current  # current selected store
+                self.window.controller.assistant.store.current
             )
         )
 
-        # refresh (current)
         self.actions["current.refresh_store"] = QAction(QIcon(":/icons/reload.svg"),
-                                                     trans("dialog.assistant.store.menu.current.refresh_store"))
+                                                     trans("dialog.assistant.store.menu.current.refresh_store"),
+                                                     self.menu_bar)
         self.actions["current.refresh_store"].triggered.connect(
-            lambda: self.window.controller.assistant.store.refresh_status()
+            self.window.controller.assistant.store.refresh_status
         )
 
-        # clear files (current, local)
         self.actions["current.clear_files"] = QAction(QIcon(":/icons/close.svg"),
-                                                  trans("dialog.assistant.store.menu.current.clear_files"))
+                                                  trans("dialog.assistant.store.menu.current.clear_files"),
+                                                  self.menu_bar)
         self.actions["current.clear_files"].triggered.connect(
             lambda: self.window.controller.assistant.batch.clear_store_files(
-                self.window.controller.assistant.store.current  # current selected store
+                self.window.controller.assistant.store.current
             )
         )
 
-        # truncate files (current, local + remote)
         self.actions["current.truncate_files"] = QAction(QIcon(":/icons/delete.svg"),
-                                                     trans("dialog.assistant.store.menu.current.truncate_files"))
+                                                     trans("dialog.assistant.store.menu.current.truncate_files"),
+                                                     self.menu_bar)
         self.actions["current.truncate_files"].triggered.connect(
             lambda: self.window.controller.assistant.batch.truncate_store_files(
-                self.window.controller.assistant.store.current  # current selected store
+                self.window.controller.assistant.store.current
             )
         )
 
-        # delete (current, local + remote)
         self.actions["current.delete"] = QAction(QIcon(":/icons/delete.svg"),
-                                                         trans("dialog.assistant.store.menu.current.delete"))
+                                                         trans("dialog.assistant.store.menu.current.delete"),
+                                                         self.menu_bar)
         self.actions["current.delete"].triggered.connect(
             lambda: self.window.controller.assistant.store.delete(
-                self.window.controller.assistant.store.current  # current selected store
+                self.window.controller.assistant.store.current
             )
         )
 
-        # --------------------------------------------
-
-        # import all (stores + files)
-        self.actions["all.import_all"] = QAction(QIcon(":/icons/download.svg"), trans("dialog.assistant.store.menu.all.import_all"))
+        self.actions["all.import_all"] = QAction(QIcon(":/icons/download.svg"),
+                                                 trans("dialog.assistant.store.menu.all.import_all"),
+                                                 self.menu_bar)
         self.actions["all.import_all"].triggered.connect(
-            lambda: self.window.controller.assistant.batch.import_stores()
+            self.window.controller.assistant.batch.import_stores
         )
 
-        # import files (all)
         self.actions["all.import_files"] = QAction(QIcon(":/icons/download.svg"),
-                                             trans("dialog.assistant.store.menu.all.import_files"))
+                                             trans("dialog.assistant.store.menu.all.import_files"),
+                                             self.menu_bar)
         self.actions["all.import_files"].triggered.connect(
-            lambda: self.window.controller.assistant.batch.import_files()
+            self.window.controller.assistant.batch.import_files
         )
 
-        # refresh (local)
         self.actions["all.refresh_stores"] = QAction(QIcon(":/icons/reload.svg"),
-                                               trans("dialog.assistant.store.menu.all.refresh_store"))
+                                               trans("dialog.assistant.store.menu.all.refresh_store"),
+                                               self.menu_bar)
         self.actions["all.refresh_stores"].triggered.connect(
-            lambda: self.window.controller.assistant.batch.refresh_stores()
+            self.window.controller.assistant.batch.refresh_stores
         )
 
-        # clear stores (all, local)
         self.actions["all.clear_stores"] = QAction(QIcon(":/icons/close.svg"),
-                                                  trans("dialog.assistant.store.menu.all.clear_store"))
+                                                  trans("dialog.assistant.store.menu.all.clear_store"),
+                                                  self.menu_bar)
         self.actions["all.clear_stores"].triggered.connect(
-            lambda: self.window.controller.assistant.batch.clear_stores()
+            self.window.controller.assistant.batch.clear_stores
         )
 
-        # clear files (all, local)
         self.actions["all.clear_files"] = QAction(QIcon(":/icons/close.svg"),
-                                               trans("dialog.assistant.store.menu.all.clear_files"))
+                                               trans("dialog.assistant.store.menu.all.clear_files"),
+                                               self.menu_bar)
         self.actions["all.clear_files"].triggered.connect(
-            lambda: self.window.controller.assistant.batch.clear_files()
+            self.window.controller.assistant.batch.clear_files
         )
 
-        # truncate stores (all, local + remote)
-        self.actions["all.truncate_stores"] = QAction(QIcon(":/icons/delete.svg"), trans("dialog.assistant.store.menu.all.truncate_store"))
+        self.actions["all.truncate_stores"] = QAction(QIcon(":/icons/delete.svg"),
+                                                      trans("dialog.assistant.store.menu.all.truncate_store"),
+                                                      self.menu_bar)
         self.actions["all.truncate_stores"].triggered.connect(
-            lambda: self.window.controller.assistant.batch.truncate_stores()
+            self.window.controller.assistant.batch.truncate_stores
         )
 
-        # truncate files (all, local + remote)
-        self.actions["all.truncate_files"] = QAction(QIcon(":/icons/delete.svg"), trans("dialog.assistant.store.menu.all.truncate_files"))
+        self.actions["all.truncate_files"] = QAction(QIcon(":/icons/delete.svg"),
+                                                     trans("dialog.assistant.store.menu.all.truncate_files"),
+                                                     self.menu_bar)
         self.actions["all.truncate_files"].triggered.connect(
-            lambda: self.window.controller.assistant.batch.truncate_files()
+            self.window.controller.assistant.batch.truncate_files
         )
 
-        # current
         self.menu["current"].addAction(self.actions["current.import_files"])
         self.menu["current"].addAction(self.actions["current.refresh_store"])
         self.menu["current"].addAction(self.actions["current.clear_files"])
         self.menu["current"].addAction(self.actions["current.truncate_files"])
         self.menu["current"].addAction(self.actions["current.delete"])
 
-        # all
         self.menu["all"].addAction(self.actions["all.import_all"])
         self.menu["all"].addAction(self.actions["all.import_files"])
         self.menu["all"].addAction(self.actions["all.refresh_stores"])
@@ -370,30 +356,29 @@ class AssistantVectorStore:
         for key in options:
             option = options[key]
             option['id'] = key
-            # create widget by option type
             if option['type'] == 'text' or option['type'] == 'int' or option['type'] == 'float':
                 if 'slider' in option and option['slider'] \
                         and (option['type'] == 'int' or option['type'] == 'float'):
-                    widgets[key] = OptionSlider(self.window, parent, key, option)  # slider + text input
+                    widgets[key] = OptionSlider(self.window, parent, key, option)
                 else:
                     if 'secret' in option and option['secret']:
-                        widgets[key] = PasswordInput(self.window, parent, key, option)  # password input
+                        widgets[key] = PasswordInput(self.window, parent, key, option)
                     else:
-                        widgets[key] = OptionInput(self.window, parent, key, option)  # text input
+                        widgets[key] = OptionInput(self.window, parent, key, option)
             elif option['type'] == 'textarea':
-                widgets[key] = OptionTextarea(self.window, parent, key, option)  # textarea
+                widgets[key] = OptionTextarea(self.window, parent, key, option)
             elif option['type'] == 'bool':
-                widgets[key] = OptionCheckbox(self.window, parent, key, option)  # checkbox
+                widgets[key] = OptionCheckbox(self.window, parent, key, option)
             elif option['type'] == 'bool_list':
                 self.window.controller.config.placeholder.apply(option)
-                widgets[key] = OptionCheckboxList(self.window, parent, key, option)  # checkbox list
+                widgets[key] = OptionCheckboxList(self.window, parent, key, option)
             elif option['type'] == 'dict':
                 self.window.controller.config.placeholder.apply(option)
-                widgets[key] = OptionDict(self.window, parent, key, option)  # dictionary
+                widgets[key] = OptionDict(self.window, parent, key, option)
                 widgets[key].setMinimumHeight(200)
             elif option['type'] == 'combo':
                 self.window.controller.config.placeholder.apply(option)
-                widgets[key] = OptionCombo(self.window, parent, key, option)  # combobox
+                widgets[key] = OptionCombo(self.window, parent, key, option)
 
         return widgets
 
@@ -448,10 +433,9 @@ class AssistantVectorStore:
             desc = trans(option['description'])
             desc_key = 'model.' + key + '.desc'
 
-        # 2-columns layout
         if option['type'] not in one_column_types:
             cols = QHBoxLayout()
-            cols.addWidget(self.window.ui.nodes[label_key])  # disable label in bool type
+            cols.addWidget(self.window.ui.nodes[label_key])
             cols.addWidget(widget)
 
             cols_widget = QWidget()
@@ -461,12 +445,11 @@ class AssistantVectorStore:
             layout = QVBoxLayout()
             layout.addWidget(cols_widget)
         else:
-            # 1-column layout: textarea and dict fields
             layout = QVBoxLayout()
             if option['type'] != 'bool':
                 layout.addWidget(self.window.ui.nodes[label_key])
             else:
-                widget.box.setText(label)  # set checkbox label
+                widget.box.setText(label)
             layout.addWidget(widget)
 
         if desc:
@@ -507,23 +490,50 @@ class AssistantVectorStore:
         :param id: ID of the list
         :param data: Data to update
         """
-        if id not in self.window.ui.models:
+        models = self.window.ui.models
+        if id not in models:
             return
-        self.window.ui.models[id].removeRows(0, self.window.ui.models[id].rowCount())
-        i = 0
-        for n in data:
-            name = data[n].name
-            extra = ""
-            if name == "" or name is None:
-                if self.window.core.config.get("assistant.store.hide_threads"):
+        model = models[id]
+        hide_threads = bool(self.window.core.config.get("assistant.store.hide_threads"))
+        suffix = trans("assistant.store.files.suffix")
+
+        names = []
+        for n, store in data.items():
+            num_files = store.get_file_count()
+            name = store.name
+            extras = []
+            if not name:
+                if hide_threads:
                     continue
-                name = data[n].id
-            if data[n].num_files > 0:
-                extra += "{} {}".format(data[n].num_files, trans("assistant.store.files.suffix"))
-            if data[n].usage_bytes > 0:
-                extra += " - {}".format(self.window.core.filesystem.sizeof_fmt(data[n].usage_bytes))
-            if extra != "":
-                name += " ({})".format(extra)
-            self.window.ui.models[id].insertRow(i)
-            self.window.ui.models[id].setData(self.window.ui.models[id].index(i, 0), name)
-            i += 1
+                name = store.id
+            if num_files > 0:
+                extras.append(f"{num_files} {suffix}")
+            if store.usage_bytes > 0:
+                extras.append(self.window.core.filesystem.sizeof_fmt(store.usage_bytes))
+            if extras:
+                name = f"{name} ({' - '.join(extras)})"
+            names.append(name)
+
+        model.setRowCount(len(names))
+        for i, name in enumerate(names):
+            model.setData(model.index(i, 0), name)
+
+    # ==================== files context menu handler ====================
+    def on_files_context_menu(self, pos: QPoint):
+        """
+        Context menu for the files list: provides Delete action.
+        """
+        view_id = 'assistant.store.files.list'
+        if view_id not in self.window.ui.nodes:
+            return
+        view = self.window.ui.nodes[view_id]
+        index = view.indexAt(pos)
+        if not index.isValid():
+            return
+
+        row = index.row()
+        menu = QMenu(view)
+        act_delete = QAction(QIcon(":/icons/delete.svg"), trans("assistant.store.menu.file.delete") if hasattr(self.window, 'tr') else "Delete", view)
+        act_delete.triggered.connect(lambda r=row: self.window.controller.assistant.store.delete_file_by_idx(r))
+        menu.addAction(act_delete)
+        menu.exec(view.mapToGlobal(pos))

pygpt_net/ui/layout/chat/input.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.27 07:00:00                  #
+# Updated Date: 2025.09.03 00:00:00                  #
 # ================================================== #
 
 from functools import partial
@@ -246,7 +246,7 @@ class Input:
 
         idx = tabs.currentIndex()
         if idx == 0:
-            nodes['input'].setMinimumHeight(self.min_height_input)
+            # nodes['input'].setMinimumHeight(self.min_height_input)
             tabs.setMinimumHeight(self.min_height_input_tab)
             sizes = controller_ui.splitter_output_size_input
             if sizes and sizes != [0, 0]:
@@ -255,5 +255,5 @@ class Input:
             sizes = controller_ui.splitter_output_size_files
             if sizes and controller_ui.splitter_output_size_input != [0, 0]:
                 splitters['main.output'].setSizes(sizes)
-            nodes['input'].setMinimumHeight(self.min_height_files_tab)
+            # nodes['input'].setMinimumHeight(self.min_height_files_tab)
             tabs.setMinimumHeight(self.min_height_files_tab + 90)