pygpt-net 2.6.21__py3-none-any.whl → 2.6.23__py3-none-any.whl

pygpt_net/ui/layout/chat/attachments_ctx.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.26 02:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
 import os
@@ -19,8 +19,6 @@ 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):
         """
@@ -65,11 +63,9 @@ class AttachmentsCtx:
                 self.window.controller.chat.attachment.MODE_DISABLED
             ))
 
-        # buttons layout
         buttons_layout = QHBoxLayout()
         buttons_layout.addWidget(self.window.ui.nodes['attachments_ctx.btn.clear'])
         buttons_layout.addWidget(empty_widget)
-        # buttons_layout.addStretch()
         buttons_layout.addWidget(self.window.ui.nodes['input.attachments.ctx.mode.label'])
         buttons_layout.addWidget(self.window.ui.nodes['input.attachments.ctx.mode.full'])
         buttons_layout.addWidget(self.window.ui.nodes['input.attachments.ctx.mode.query'])
@@ -77,7 +73,6 @@ class AttachmentsCtx:
         buttons_layout.addWidget(self.window.ui.nodes['input.attachments.ctx.mode.off'])
         buttons_layout.addStretch()
 
-        # layout
         layout = QVBoxLayout()
         layout.addWidget(self.window.ui.nodes['tip.input.attachments.ctx'])
         layout.addWidget(self.window.ui.nodes['attachments_ctx'])
@@ -87,16 +82,14 @@ class AttachmentsCtx:
 
     def setup_attachments(self):
         """Setup attachments uploaded list"""
-        # attachments
         self.window.ui.nodes[self.id] = AttachmentCtxList(self.window)
 
-        # buttons
         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()
         )
 
-        self.window.ui.models[self.id] = self.create_model(self.window)
+        self.window.ui.models[self.id] = self.create_model(self.window.ui.nodes[self.id])
         self.window.ui.nodes[self.id].setModel(self.window.ui.models[self.id])
 
     def create_model(self, parent) -> QStandardItemModel:
@@ -120,43 +113,46 @@ class AttachmentsCtx:
 
         :param data: Data to update
         """
-        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']
-            size = "-"
-            path = "No path"
-            if 'path' in item:
-                path = item['path']
-            uuid = ""
-            if 'uuid' in item:
-                uuid = item['uuid']
+        model = self.window.ui.models[self.id]
+        row_count = len(data)
+        model.beginResetModel()
+        model.setRowCount(row_count)
+
+        m_index = model.index
+        m_setData = model.setData
+        tooltip_role = QtCore.Qt.ToolTipRole
+        trans_indexed = trans("attachments.ctx.indexed")
+        sizeof_fmt = self.window.core.filesystem.sizeof_fmt
+        stat = os.stat
+
+        for i, item in enumerate(data):
+            name = item.get('name', "No name")
+            path = item.get('path', "No path")
+            uuid = item.get('uuid', "")
             length = "-"
             if 'length' in item:
                 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))
-            elif 'size' in item:
-                size = self.window.core.filesystem.sizeof_fmt(item['size'])
-
-            self.window.ui.models[self.id].insertRow(i)
-            index = self.window.ui.models[self.id].index(i, 0)
-            self.window.ui.models[self.id].setData(index, "uuid: " + str(uuid), QtCore.Qt.ToolTipRole)
-            self.window.ui.models[self.id].setData(self.window.ui.models[self.id].index(i, 0), name)
-            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
+            idx_str = trans_indexed if item.get('indexed') else ""
+
+            size = "-"
+            if isinstance(path, str):
+                try:
+                    st = stat(path)
+                except (OSError, ValueError, TypeError):
+                    pass
+                else:
+                    size = sizeof_fmt(st.st_size)
+            if size == "-" and 'size' in item:
+                size = sizeof_fmt(item['size'])
+
+            idx0 = m_index(i, 0)
+            m_setData(idx0, "uuid: " + str(uuid), tooltip_role)
+            m_setData(idx0, name)
+            m_setData(m_index(i, 1), path)
+            m_setData(m_index(i, 2), size)
+            m_setData(m_index(i, 3), length)
+            m_setData(m_index(i, 4), idx_str)
+
+        model.endResetModel()

pygpt_net/ui/layout/chat/attachments_uploaded.py

@@ -6,22 +6,20 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.29 16:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
 import os
 
 from PySide6 import QtCore
 from PySide6.QtGui import QStandardItemModel, Qt, QIcon
-from PySide6.QtWidgets import QVBoxLayout, QPushButton, QHBoxLayout, QCheckBox, QLabel, QWidget
+from PySide6.QtWidgets import QVBoxLayout, QPushButton, QHBoxLayout, QWidget
 
 from pygpt_net.ui.widget.element.button import SyncButton
 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):
         """
@@ -39,16 +37,15 @@ class AttachmentsUploaded:
         :return: QVBoxLayout
         """
         self.setup_attachments()
-        empty_widget = QWidget()
+        empty_widget = QWidget(self.window)
 
-        self.window.ui.nodes['attachments_uploaded.sync.tip'] = HelpLabel(trans('attachments_uploaded.sync.tip'))
+        self.window.ui.nodes['attachments_uploaded.sync.tip'] = HelpLabel(trans('attachments_uploaded.sync.tip'), self.window)
         self.window.ui.nodes['attachments_uploaded.sync.tip'].setWordWrap(False)
         self.window.ui.nodes['attachments_uploaded.sync.tip'].setAlignment(Qt.AlignCenter)
 
         self.window.ui.nodes['tip.input.attachments.uploaded'] = HelpLabel(trans('tip.input.attachments.uploaded'),
                                                                            self.window)
 
-        # buttons layout
         buttons_layout = QHBoxLayout()
         buttons_layout.addWidget(self.window.ui.nodes['attachments_uploaded.btn.sync'])
         buttons_layout.addWidget(self.window.ui.nodes['attachments_uploaded.btn.clear'])
@@ -56,8 +53,6 @@ class AttachmentsUploaded:
         buttons_layout.addWidget(self.window.ui.nodes['attachments_uploaded.sync.tip'])
         buttons_layout.addStretch()
 
-
-        # layout
         layout = QVBoxLayout()
         layout.addWidget(self.window.ui.nodes['tip.input.attachments.uploaded'])
         layout.addWidget(self.window.ui.nodes['attachments_uploaded'])
@@ -69,10 +64,8 @@ class AttachmentsUploaded:
         """
         Setup attachments uploaded list
         """
-        # attachments
         self.window.ui.nodes[self.id] = UploadedFileList(self.window)
 
-        # 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(QIcon(":/icons/close.svg"), trans('attachments_uploaded.btn.clear'))
         self.window.ui.nodes['attachments_uploaded.btn.clear'].clicked.connect(
@@ -102,32 +95,36 @@ class AttachmentsUploaded:
 
         :param data: Data to update
         """
+        model = self.window.ui.models[self.id]
         store_names = self.window.core.assistants.store.get_names()
-        self.window.ui.models[self.id].removeRows(0, self.window.ui.models[self.id].rowCount())
-        i = 0
-        for id in data:
-            item = data[id]
-            size = "-"
-            path = item.path
-
-            # size
-            if item.size is not None:
-                size = self.window.core.filesystem.sizeof_fmt(item.size)
-            else:
-                if path and os.path.exists(path):
-                    size = self.window.core.filesystem.sizeof_fmt(os.path.getsize(path))
-
-            # vector stores
-            if item.store_id is not None and item.store_id in store_names:
-                vector_store = store_names[item.store_id]
-            else:
-                vector_store = trans("assistant.store.thread_only")
-
-            self.window.ui.models[self.id].insertRow(i)
-            index = self.window.ui.models[self.id].index(i, 0)
-            self.window.ui.models[self.id].setData(index, "file_id: " + str(item.file_id), QtCore.Qt.ToolTipRole)
-            self.window.ui.models[self.id].setData(self.window.ui.models[self.id].index(i, 0), item.name)
-            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), vector_store)
-            i += 1
+        thread_only_label = trans("assistant.store.thread_only")
+        fs = self.window.core.filesystem
+
+        model.beginResetModel()
+        model.setRowCount(0)
+        count = len(data)
+        if count:
+            model.setRowCount(count)
+            for i, item in enumerate(data.values()):
+                path = item.path
+                if item.size is not None:
+                    size_str = fs.sizeof_fmt(item.size)
+                else:
+                    size_str = "-"
+                    if path:
+                        try:
+                            st = os.stat(path)
+                        except OSError:
+                            pass
+                        else:
+                            size_str = fs.sizeof_fmt(st.st_size)
+
+                store_name = store_names.get(item.store_id, thread_only_label)
+
+                idx0 = model.index(i, 0)
+                model.setData(idx0, item.name)
+                model.setData(idx0, f"file_id: {item.file_id}", QtCore.Qt.ToolTipRole)
+                model.setData(model.index(i, 1), path)
+                model.setData(model.index(i, 2), size_str)
+                model.setData(model.index(i, 3), store_name)
+        model.endResetModel()

pygpt_net/ui/layout/chat/calendar.py

@@ -6,11 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.01.19 02:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
 from PySide6.QtCore import Qt
-from PySide6.QtWidgets import QVBoxLayout, QLabel, QHBoxLayout, QWidget, QSplitter, QSizePolicy, QRadioButton, QCheckBox
+from PySide6.QtWidgets import QVBoxLayout, QLabel, QHBoxLayout, QWidget, QSplitter, QSizePolicy, QRadioButton, QCheckBox, QButtonGroup
 
 from pygpt_net.ui.widget.calendar.select import CalendarSelect
 from pygpt_net.ui.widget.element.checkbox import ColorCheckbox
@@ -20,6 +20,8 @@ from pygpt_net.utils import trans
 
 
 class Calendar:
+    __slots__ = ('window',)
+
     def __init__(self, window=None):
         """
         Calendar UI
@@ -34,8 +36,11 @@ class Calendar:
 
         :return: QWidget
         """
-        self.window.ui.calendar['select'] = CalendarSelect(self.window)
-        self.window.ui.calendar['note'] = CalendarNote(self.window)
+        ui = self.window.ui
+        ui.calendar['select'] = CalendarSelect(self.window)
+        ui.calendar['select'].setMinimumSize(200, 200)
+        ui.calendar['select'].setGridVisible(True)
+        ui.calendar['note'] = CalendarNote(self.window)
 
     def setup(self) -> QWidget:
         """
@@ -49,58 +54,76 @@ class Calendar:
         body.append(self.window.ui.calendar['select'])
         return body
 
+    def _on_filter_id_clicked(self, id_: int) -> None:
+        key = "all" if id_ == 0 else ("pinned" if id_ == 1 else "indexed")
+        self.window.controller.ctx.common.toggle_display_filter(key)
+
+    def _on_counters_all_toggled(self, checked: bool) -> None:
+        self.window.controller.calendar.note.toggle_counters_all(checked)
+
     def setup_filters(self) -> QWidget:
         """
         Setup calendar filters
 
         :return: QWidget
         """
-        self.window.ui.nodes['filter.ctx.label'] = QLabel(trans("filter.ctx.label"))
-
-        # display: all
-        self.window.ui.nodes['filter.ctx.radio.all'] = QRadioButton(trans("filter.ctx.radio.all"))
-        self.window.ui.nodes['filter.ctx.radio.all'].clicked.connect(
-            lambda: self.window.controller.ctx.common.toggle_display_filter("all"))
-
-        # display: only pinned
-        self.window.ui.nodes['filter.ctx.radio.pinned'] = QRadioButton(trans("filter.ctx.radio.pinned"))
-        self.window.ui.nodes['filter.ctx.radio.pinned'].clicked.connect(
-            lambda: self.window.controller.ctx.common.toggle_display_filter("pinned"))
-
-        # display: only indexed
-        self.window.ui.nodes['filter.ctx.radio.indexed'] = QRadioButton(trans("filter.ctx.radio.indexed"))
-        self.window.ui.nodes['filter.ctx.radio.indexed'].clicked.connect(
-            lambda: self.window.controller.ctx.common.toggle_display_filter("indexed"))
-
-        # all counters
-        self.window.ui.nodes['filter.ctx.counters.all'] = QCheckBox(trans("filter.ctx.counters.all"))
-        self.window.ui.nodes['filter.ctx.counters.all'].stateChanged.connect(
-            lambda: self.window.controller.calendar.note.toggle_counters_all(
-                self.window.ui.nodes['filter.ctx.counters.all'].isChecked())
-        )
-        if self.window.core.config.get("ctx.counters.all"):
-            self.window.ui.nodes['filter.ctx.counters.all'].setChecked(True)
+        ui = self.window.ui
+        nodes = ui.nodes
+
+        label_existing = nodes.get('filter.ctx.label')
+        if label_existing is None:
+            layout = QHBoxLayout()
+            widget = QWidget()
+            rows = QVBoxLayout()
+            widget.setLayout(rows)
+
+            label = QLabel(trans("filter.ctx.label"), widget)
+
+            radio_all = QRadioButton(trans("filter.ctx.radio.all"), widget)
+            radio_pinned = QRadioButton(trans("filter.ctx.radio.pinned"), widget)
+            radio_indexed = QRadioButton(trans("filter.ctx.radio.indexed"), widget)
+
+            counters_all = QCheckBox(trans("filter.ctx.counters.all"), widget)
+
+            layout.addWidget(label)
+            layout.addWidget(radio_all)
+            layout.addWidget(radio_pinned)
+            layout.addWidget(radio_indexed)
+            layout.addWidget(counters_all)
+            layout.addStretch()
+
+            nodes['filter.ctx.labels'] = ColorCheckbox(self.window)
+
+            rows.addLayout(layout)
+            rows.addWidget(nodes['filter.ctx.labels'])
+
+            group = QButtonGroup(widget)
+            group.setExclusive(True)
+            group.addButton(radio_all, 0)
+            group.addButton(radio_pinned, 1)
+            group.addButton(radio_indexed, 2)
+            group.idClicked.connect(self._on_filter_id_clicked)
+
+            counters_all.toggled.connect(self._on_counters_all_toggled)
+
+            nodes['filter.ctx.label'] = label
+            nodes['filter.ctx.radio.all'] = radio_all
+            nodes['filter.ctx.radio.pinned'] = radio_pinned
+            nodes['filter.ctx.radio.indexed'] = radio_indexed
+            nodes['filter.ctx.counters.all'] = counters_all
         else:
-            self.window.ui.nodes['filter.ctx.counters.all'].setChecked(False)
-
-        # layout
-        layout = QHBoxLayout()
-        layout.addWidget(self.window.ui.nodes['filter.ctx.label'])
-        layout.addWidget(self.window.ui.nodes['filter.ctx.radio.all'])
-        layout.addWidget(self.window.ui.nodes['filter.ctx.radio.pinned'])
-        layout.addWidget(self.window.ui.nodes['filter.ctx.radio.indexed'])
-        layout.addWidget(self.window.ui.nodes['filter.ctx.counters.all'])
-        layout.addStretch()
-
-        self.window.ui.nodes['filter.ctx.labels'] = ColorCheckbox(self.window)
-
-        rows = QVBoxLayout()
-        rows.addLayout(layout)
-        rows.addWidget(self.window.ui.nodes['filter.ctx.labels'])
-
-        widget = QWidget()
-        widget.setLayout(rows)
-
+            widget = nodes['filter.ctx.label'].parentWidget()
+            nodes['filter.ctx.label'].setText(trans("filter.ctx.label"))
+            nodes['filter.ctx.radio.all'].setText(trans("filter.ctx.radio.all"))
+            nodes['filter.ctx.radio.pinned'].setText(trans("filter.ctx.radio.pinned"))
+            nodes['filter.ctx.radio.indexed'].setText(trans("filter.ctx.radio.indexed"))
+            nodes['filter.ctx.counters.all'].setText(trans("filter.ctx.counters.all"))
+
+        desired = bool(self.window.core.config.get("ctx.counters.all"))
+        if nodes['filter.ctx.counters.all'].isChecked() != desired:
+            nodes['filter.ctx.counters.all'].setChecked(desired)
+
+        widget.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
         return widget
 
     def setup_calendar(self) -> QWidget:
@@ -109,22 +132,32 @@ class Calendar:
 
         :return: QSplitter
         """
-        # calendar
+        ui = self.window.ui
+        nodes = ui.nodes
+        calendar = ui.calendar
+
         select_layout = QVBoxLayout()
-        self.window.ui.calendar['select'].setMinimumHeight(200)
-        self.window.ui.calendar['select'].setMinimumWidth(200)
-        self.window.ui.calendar['select'].setGridVisible(True)
-        select_layout.addWidget(self.window.ui.calendar['select'])
+        select_layout.addWidget(calendar['select'])
         select_layout.setContentsMargins(5, 0, 5, 0)
 
-        self.window.ui.nodes['tip.output.tab.calendar'] = HelpLabel(trans('tip.output.tab.calendar'), self.window)
+        tip = nodes.get('tip.output.tab.calendar')
+        if tip is None:
+            tip = HelpLabel(trans('tip.output.tab.calendar'), self.window)
+            nodes['tip.output.tab.calendar'] = tip
+        else:
+            tip.setText(trans('tip.output.tab.calendar'))
+
+        note_label = calendar.get('note.label')
+        if note_label is None:
+            note_label = QLabel(trans('calendar.note.label'))
+            calendar['note.label'] = note_label
+        else:
+            note_label.setText(trans('calendar.note.label'))
 
-        # note
-        self.window.ui.calendar['note.label'] = QLabel(trans('calendar.note.label'))
         layout = QVBoxLayout()
-        layout.addWidget(self.window.ui.calendar['note.label'])
-        layout.addWidget(self.window.ui.calendar['note'])
-        layout.addWidget(self.window.ui.nodes['tip.output.tab.calendar'])
+        layout.addWidget(note_label)
+        layout.addWidget(calendar['note'])
+        layout.addWidget(tip)
         layout.setContentsMargins(0, 0, 0, 0)
 
         widget = QWidget()
@@ -136,16 +169,13 @@ class Calendar:
         select_widget = QWidget()
         select_widget.setLayout(select_layout)
 
-        # layout / splitter
-        self.window.ui.splitters['calendar'] = QSplitter(Qt.Horizontal)
-        self.window.ui.splitters['calendar'].addWidget(select_widget)
-        self.window.ui.splitters['calendar'].addWidget(widget)
-        self.window.ui.splitters['calendar'].setStretchFactor(0, 6)  # 60%
-        self.window.ui.splitters['calendar'].setStretchFactor(1, 4)  # 40%
+        ui.splitters['calendar'] = QSplitter(Qt.Horizontal)
+        ui.splitters['calendar'].addWidget(select_widget)
+        ui.splitters['calendar'].addWidget(widget)
+        ui.splitters['calendar'].setStretchFactor(0, 6)
+        ui.splitters['calendar'].setStretchFactor(1, 4)
 
-        self.window.ui.splitters['calendar'].setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
+        ui.splitters['calendar'].setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
         filters.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
 
-        # build tab body
-        return self.window.core.tabs.from_widget(self.window.ui.splitters['calendar'])
-
+        return self.window.core.tabs.from_widget(ui.splitters['calendar'])

pygpt_net/ui/layout/chat/chat.py

@@ -6,10 +6,10 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.17 03:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
-from PySide6.QtCore import Qt
+from PySide6.QtCore import Qt, Slot
 from PySide6.QtWidgets import QSplitter
 
 from pygpt_net.ui.layout.chat.input import Input
@@ -34,28 +34,34 @@ class ChatMain:
         :return: QSplitter
         :rtype: QSplitter
         """
-        input = self.input.setup()
-        output = self.output.setup()
+        input_widget = self.input.setup()
+        output_widget = self.output.setup()
 
-        # main vertical splitter
-        self.window.ui.splitters['main.output'] = QSplitter(Qt.Vertical)
-        self.window.ui.splitters['main.output'].addWidget(output)
-        self.window.ui.splitters['main.output'].addWidget(input)
-        self.window.ui.splitters['main.output'].setStretchFactor(0, 9)  # Output widget stretch factor
-        self.window.ui.splitters['main.output'].setStretchFactor(1, 1)  # Input widget stretch factor
-        self.window.ui.splitters['main.output'].splitterMoved.connect(self.on_splitter_moved)
-        self.window.controller.ui.splitter_output_size_input = self.window.ui.splitters['main.output'].sizes()
+        splitter = QSplitter(Qt.Vertical)
+        self.window.ui.splitters['main.output'] = splitter
+        splitter.addWidget(output_widget)
+        splitter.addWidget(input_widget)
+        splitter.setStretchFactor(0, 9)  # Output widget stretch factor
+        splitter.setStretchFactor(1, 1)  # Input widget stretch factor
+        splitter.splitterMoved.connect(self.on_splitter_moved)
+        self.window.controller.ui.splitter_output_size_input = splitter.sizes()
 
-        return self.window.ui.splitters['main.output']
+        return splitter
 
+    @Slot(int, int)
     def on_splitter_moved(self, pos, index):
         """
         Store the size of the output splitter when it is moved
         """
-        if "input" not in self.window.ui.tabs:
+        tabs = self.window.ui.tabs
+        if "input" not in tabs:
             return
-        idx = self.window.ui.tabs['input'].currentIndex()
+        splitter = self.window.ui.splitters.get('main.output')
+        if splitter is None:
+            return
+        idx = tabs['input'].currentIndex()
+        sizes = splitter.sizes()
         if idx != 0:
-            self.window.controller.ui.splitter_output_size_files = self.window.ui.splitters['main.output'].sizes()
+            self.window.controller.ui.splitter_output_size_files = sizes
         else:
-            self.window.controller.ui.splitter_output_size_input = self.window.ui.splitters['main.output'].sizes()
+            self.window.controller.ui.splitter_output_size_input = sizes