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

pygpt_net/ui/layout/chat/painter.py

@@ -6,11 +6,12 @@
 # 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.QtGui import QPixmap, QIcon
 from PySide6.QtWidgets import QVBoxLayout, QHBoxLayout, QWidget, QRadioButton, QPushButton, QComboBox, QScrollArea
+from PySide6.QtCore import QSize
 
 from pygpt_net.ui.widget.draw.painter import PainterWidget
 from pygpt_net.ui.widget.element.labels import HelpLabel
@@ -26,6 +27,7 @@ class Painter:
         :param window: Window instance
         """
         self.window = window
+        self._initialized = False
 
     def init(self):
         """
@@ -33,47 +35,62 @@ class Painter:
 
         :return: QWidget
         """
-        self.window.ui.painter = PainterWidget(self.window)
+        ui = self.window.ui
+        nodes = ui.nodes
+        common = self.window.controller.painter.common
+
+        if getattr(ui, 'painter', None) is None:
+            ui.painter = PainterWidget(self.window)
 
         key = 'painter.btn.brush'
-        self.window.ui.nodes[key] = QRadioButton(trans('painter.mode.paint'))
-        self.window.ui.nodes[key].setChecked(True)
-        self.window.ui.nodes[key].toggled.connect(self.window.controller.painter.common.set_brush_mode)
+        if nodes.get(key) is None:
+            rb = QRadioButton(trans('painter.mode.paint'))
+            rb.setChecked(True)
+            rb.toggled.connect(self.window.controller.painter.common.set_brush_mode)
+            nodes[key] = rb
 
         key = 'painter.btn.erase'
-        self.window.ui.nodes[key] = QRadioButton(trans('painter.mode.erase'))
-        self.window.ui.nodes[key].toggled.connect(self.window.controller.painter.common.set_erase_mode)
+        if nodes.get(key) is None:
+            rb = QRadioButton(trans('painter.mode.erase'))
+            rb.toggled.connect(self.window.controller.painter.common.set_erase_mode)
+            nodes[key] = rb
 
         key = 'painter.select.brush.size'
-        sizes = self.window.controller.painter.common.get_sizes()
-        self.window.ui.nodes[key] = QComboBox()
-        self.window.ui.nodes[key].addItems(sizes)
-        self.window.ui.nodes[key].currentTextChanged.connect(
-            self.window.controller.painter.common.change_brush_size)
-        self.window.ui.nodes[key].setMinimumContentsLength(10)
-        self.window.ui.nodes[key].setSizeAdjustPolicy(QComboBox.AdjustToContents)
+        if nodes.get(key) is None:
+            sizes = common.get_sizes()
+            cb = QComboBox()
+            cb.addItems(sizes)
+            cb.currentTextChanged.connect(common.change_brush_size)
+            cb.setMinimumContentsLength(10)
+            cb.setSizeAdjustPolicy(QComboBox.AdjustToContents)
+            nodes[key] = cb
 
         key = 'painter.select.canvas.size'
-        canvas_sizes = self.window.controller.painter.common.get_canvas_sizes()
-        self.window.ui.nodes[key] = QComboBox()
-        self.window.ui.nodes[key].addItems(canvas_sizes)
-        self.window.ui.nodes[key].setMinimumContentsLength(20)
-        self.window.ui.nodes[key].setSizeAdjustPolicy(QComboBox.AdjustToContents)
-        self.window.ui.nodes[key].currentTextChanged.connect(
-            self.window.controller.painter.common.change_canvas_size)
+        if nodes.get(key) is None:
+            canvas_sizes = common.get_canvas_sizes()
+            cb = QComboBox()
+            cb.addItems(canvas_sizes)
+            cb.setMinimumContentsLength(20)
+            cb.setSizeAdjustPolicy(QComboBox.AdjustToContents)
+            cb.currentTextChanged.connect(common.change_canvas_size)
+            nodes[key] = cb
 
         key = 'painter.select.brush.color'
-        self.window.ui.nodes[key] = QComboBox()
-        colors = self.window.controller.painter.common.get_colors()
-        for color_name, color_value in colors.items():
-            pixmap = QPixmap(100, 100)
-            pixmap.fill(color_value)
-            icon = QIcon(pixmap)
-            self.window.ui.nodes[key].addItem(icon, color_name, color_value)
-        self.window.ui.nodes[key].currentTextChanged.connect(
-            self.window.controller.painter.common.change_brush_color)
-        self.window.ui.nodes[key].setMinimumContentsLength(10)
-        self.window.ui.nodes[key].setSizeAdjustPolicy(QComboBox.AdjustToContents)
+        if nodes.get(key) is None:
+            cb = QComboBox()
+            cb.setIconSize(QSize(16, 16))
+            colors = common.get_colors()
+            for color_name, color_value in colors.items():
+                pixmap = QPixmap(16, 16)
+                pixmap.fill(color_value)
+                icon = QIcon(pixmap)
+                cb.addItem(icon, color_name, color_value)
+            cb.currentTextChanged.connect(common.change_brush_color)
+            cb.setMinimumContentsLength(10)
+            cb.setSizeAdjustPolicy(QComboBox.AdjustToContents)
+            nodes[key] = cb
+
+        self._initialized = True
 
     def setup(self) -> QWidget:
         """
@@ -81,7 +98,6 @@ class Painter:
 
         :return: QWidget
         """
-        # build tab body
         self.init()
         body = self.window.core.tabs.from_layout(self.setup_painter())
         body.append(self.window.ui.painter)
@@ -93,39 +109,51 @@ class Painter:
 
         :return: QVBoxLayout
         """
+        ui = self.window.ui
+        nodes = ui.nodes
+
         top = QHBoxLayout()
-        top.addWidget(self.window.ui.nodes['painter.btn.brush'])
-        top.addWidget(self.window.ui.nodes['painter.btn.erase'])
-        top.addWidget(self.window.ui.nodes['painter.select.brush.size'])
-        top.addWidget(self.window.ui.nodes['painter.select.brush.color'])
-        top.addWidget(self.window.ui.nodes['painter.select.canvas.size'])
+        top.addWidget(nodes['painter.btn.brush'])
+        top.addWidget(nodes['painter.btn.erase'])
+        top.addWidget(nodes['painter.select.brush.size'])
+        top.addWidget(nodes['painter.select.brush.color'])
+        top.addWidget(nodes['painter.select.canvas.size'])
         top.addStretch(1)
 
-        self.window.ui.nodes['painter.btn.capture'] = QPushButton(trans('painter.btn.capture'))
-        self.window.ui.nodes['painter.btn.capture'].clicked.connect(self.window.controller.painter.capture.use)
-        top.addWidget(self.window.ui.nodes['painter.btn.capture'])
-
-        self.window.ui.nodes['painter.btn.camera.capture'] = QPushButton(trans('painter.btn.camera.capture'))
-        self.window.ui.nodes['painter.btn.camera.capture'].clicked.connect(self.window.controller.painter.capture.camera)
-        top.addWidget(self.window.ui.nodes['painter.btn.camera.capture'])
-
-        self.window.ui.nodes['painter.btn.clear'] = QPushButton(trans('painter.btn.clear'))
-        self.window.ui.nodes['painter.btn.clear'].clicked.connect(self.window.ui.painter.clear_image)
-        top.addWidget(self.window.ui.nodes['painter.btn.clear'])
-
-        self.window.ui.painter_scroll = QScrollArea()
-        self.window.ui.painter_scroll.setWidget(self.window.ui.painter)
-        self.window.ui.painter_scroll.setWidgetResizable(True)
-
-        self.window.ui.nodes['tip.output.tab.draw'] = HelpLabel(trans('tip.output.tab.draw'), self.window)
+        if nodes.get('painter.btn.capture') is None:
+            btn = QPushButton(trans('painter.btn.capture'))
+            btn.clicked.connect(self.window.controller.painter.capture.use)
+            nodes['painter.btn.capture'] = btn
+        top.addWidget(nodes['painter.btn.capture'])
+
+        if nodes.get('painter.btn.camera.capture') is None:
+            btn = QPushButton(trans('painter.btn.camera.capture'))
+            btn.clicked.connect(self.window.controller.painter.capture.camera)
+            nodes['painter.btn.camera.capture'] = btn
+        top.addWidget(nodes['painter.btn.camera.capture'])
+
+        if nodes.get('painter.btn.clear') is None:
+            btn = QPushButton(trans('painter.btn.clear'))
+            btn.clicked.connect(ui.painter.clear_image)
+            nodes['painter.btn.clear'] = btn
+        top.addWidget(nodes['painter.btn.clear'])
+
+        if getattr(ui, 'painter_scroll', None) is None:
+            ui.painter_scroll = QScrollArea()
+            ui.painter_scroll.setWidget(ui.painter)
+            ui.painter_scroll.setWidgetResizable(True)
+        else:
+            if ui.painter_scroll.widget() is not ui.painter:
+                ui.painter_scroll.setWidget(ui.painter)
+            ui.painter_scroll.setWidgetResizable(True)
+
+        if nodes.get('tip.output.tab.draw') is None:
+            nodes['tip.output.tab.draw'] = HelpLabel(trans('tip.output.tab.draw'), self.window)
 
         layout = QVBoxLayout()
         layout.addLayout(top)
-        layout.addWidget( self.window.ui.painter_scroll)
-        layout.addWidget(self.window.ui.nodes['tip.output.tab.draw'])
+        layout.addWidget(ui.painter_scroll)
+        layout.addWidget(nodes['tip.output.tab.draw'])
         layout.setContentsMargins(0, 0, 0, 0)
 
-        return layout
-
-
-
+        return layout

pygpt_net/ui/layout/ctx/ctx_list.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.08.16 00:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
 from PySide6 import QtCore
@@ -20,7 +20,6 @@ from pygpt_net.ui.widget.element.button import NewCtxButton
 from pygpt_net.ui.widget.element.labels import TitleLabel
 from pygpt_net.ui.widget.lists.context import ContextList, Item, GroupItem, SectionItem
 from pygpt_net.utils import trans
-import pygpt_net.icons_rc
 
 
 class CtxList:
@@ -32,6 +31,8 @@ class CtxList:
         """
         self.window = window
         self.search_input = SearchInput(window)
+        self._group_separators = False
+        self._pinned_separators = False
 
     def setup(self) -> QWidget:
         """
@@ -39,28 +40,40 @@ class CtxList:
 
         :return: QWidget
         """
-        id = 'ctx.list'
-        self.window.ui.nodes['ctx.new'] = NewCtxButton(trans('ctx.new'), self.window)
-        self.window.ui.nodes[id] = ContextList(self.window, id)
-        self.window.ui.nodes[id].selection_locked = self.window.controller.ctx.context_change_locked
-        self.window.ui.nodes['ctx.label'] = TitleLabel(trans("ctx.list.label"))
-        self.window.ui.nodes['ctx.new'].setContentsMargins(0,0,0,0)
+        ctx_id = 'ctx.list'
+        ui = self.window.ui
+        nodes = ui.nodes
+        models = ui.models
+
+        widget = QWidget()
+        widget.setContentsMargins(0, 0, 0, 0)
+
+        new_btn = NewCtxButton(trans('ctx.new'), self.window)
+        new_btn.setContentsMargins(0, 0, 0, 0)
+        nodes['ctx.new'] = new_btn
+
+        ctx_list = ContextList(self.window, ctx_id)
+        ctx_list.selection_locked = self.window.controller.ctx.context_change_locked
+        nodes[ctx_id] = ctx_list
+
+        nodes['ctx.label'] = TitleLabel(trans("ctx.list.label"))
         search_input = self.search_input.setup()
 
-        layout = QVBoxLayout()
-        layout.addWidget(self.window.ui.nodes['ctx.new'])
-        layout.addWidget(search_input)
-        layout.addWidget(self.window.ui.nodes[id])
+        layout = QVBoxLayout(widget)
         layout.setContentsMargins(0, 0, 0, 0)
+        layout.addWidget(new_btn)
+        layout.addWidget(search_input)
+        layout.addWidget(ctx_list)
 
-        self.window.ui.models[id] = self.create_model(self.window)
-        self.window.ui.nodes[id].setModel(self.window.ui.models[id])
-        self.window.ui.nodes[id].selectionModel().selectionChanged.connect(
-            lambda: self.window.controller.ctx.selection_change()
-        )
-        widget = QWidget()
-        widget.setLayout(layout)
-        widget.setContentsMargins(0, 0, 0, 0)
+        model = self.create_model(self.window)
+        models[ctx_id] = model
+        ctx_list.setModel(model)
+
+        ctx = self.window.controller.ctx
+        ctx_list.selectionModel().selectionChanged.connect(lambda *_: ctx.selection_change())
+
+        self._group_separators = self.window.core.config.get("ctx.records.groups.separators")
+        self._pinned_separators = self.window.core.config.get("ctx.records.pinned.separators")
 
         return widget
 
@@ -84,6 +97,9 @@ class CtxList:
         node = self.window.ui.nodes[id]
         node.backup_selection()
 
+        self._group_separators = self.window.core.config.get("ctx.records.groups.separators")
+        self._pinned_separators = self.window.core.config.get("ctx.records.pinned.separators")
+
         model = self.window.ui.models.get(id)
         if model is not None:
             node.setUpdatesEnabled(False)
@@ -109,16 +125,12 @@ class CtxList:
         """
         i = 0
         last_dt_str = None
-        config = self.window.core.config
-        separators = config.get("ctx.records.separators")
-        pinned_separators = config.get("ctx.records.pinned.separators")
         model = self.window.ui.models[id]
-
         for meta_id, meta in data.items():
             gid = meta.group_id
             if (gid is None or gid == 0) and not meta.important:
                 item = self.build_item(meta_id, meta, is_group=False)
-                if separators and (not item.isPinned or pinned_separators):
+                if self._group_separators and (not item.isPinned or self._pinned_separators):
                     if i == 0 or last_dt_str != item.dt:
                         section = self.build_date_section(item.dt, group=False)
                         if section:
@@ -137,15 +149,12 @@ class CtxList:
         i = 0
         last_dt_str = None
         model = self.window.ui.models[id]
-        config = self.window.core.config
-        separators = config.get("ctx.records.separators")
-        pinned_separators = config.get("ctx.records.pinned.separators")
 
         for meta_id, meta in data.items():
             gid = meta.group_id
             if (gid is None or gid == 0) and meta.important:
                 item = self.build_item(meta_id, meta, is_group=False)
-                if separators and pinned_separators:
+                if self._group_separators and self._pinned_separators:
                     if i == 0 or last_dt_str != item.dt:
                         section = self.build_date_section(item.dt, group=False)
                         if section:
@@ -165,10 +174,6 @@ class CtxList:
         model = self.window.ui.models[id]
         groups = self.window.core.ctx.get_groups()
         search_string = self.window.core.ctx.get_search_string()
-        config = self.window.core.config
-        is_group_separators = config.get("ctx.records.groups.separators")
-        is_pinned_separators = config.get("ctx.records.pinned.separators")
-
         grouped = {}
         for meta_id, meta in data.items():
             gid = meta.group_id
@@ -213,7 +218,7 @@ class CtxList:
             i = 0
             for meta_id, meta in items_in_group:
                 item = self.build_item(meta_id, meta, is_group=True)
-                if is_group_separators and (not item.isPinned or is_pinned_separators):
+                if self._group_separators and (not item.isPinned or self._pinned_separators):
                     if i == 0 or last_dt_str != item.dt:
                         section = self.build_date_section(item.dt, group=True)
                         if section:
@@ -259,13 +264,7 @@ class CtxList:
         is_important = data.important
         is_attachment = data.has_additional_ctx()
         in_group = bool(data.group)
-
-        if is_group:
-            if self.window.core.config.get("ctx.records.groups.separators"):
-                append_dt = False
-        else:
-            if self.window.core.config.get("ctx.records.separators"):
-                append_dt = False
+        append_dt = False if (is_group and self._pinned_separators) or ((not is_group) and self._group_separators) else append_dt
 
         dt = self.convert_date(data.updated)
         date_time_str = datetime.fromtimestamp(data.updated).strftime("%Y-%m-%d %H:%M")
@@ -274,15 +273,8 @@ class CtxList:
             title = title[:80] + '...'
         clean_title = title.replace("\n", "")
 
-        if append_dt:
-            name = f"{clean_title} ({dt})"
-        else:
-            name = clean_title
-
-        mode_str = ''
-        if data.last_mode is not None:
-            mode_str = f" ({trans('mode.' + data.last_mode)})"
-
+        name = f"{clean_title} ({dt})" if append_dt else clean_title
+        mode_str = f" ({trans('mode.' + data.last_mode)})" if data.last_mode is not None else ""
         tooltip_text = f"{date_time_str}: {data.name}{mode_str} #{id}"
 
         if is_attachment:
@@ -317,8 +309,7 @@ class CtxList:
         :param group: is group
         :return: SectionItem
         """
-        item = SectionItem(dt, group=group)
-        return item
+        return SectionItem(dt, group=group)
 
     def convert_date(self, timestamp: int) -> str:
         """

pygpt_net/ui/layout/status.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2023.12.25 21:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
 from PySide6.QtCore import Qt
@@ -32,20 +32,29 @@ class Status:
 
         :return: QHBoxLayout
         """
-        self.window.ui.nodes['status'] = QLabel(trans('status.started'))
-        self.window.ui.nodes['status.agent'] = HelpLabel("")
-        self.window.ui.nodes['status.agent'].setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Minimum)
-        self.window.ui.nodes['global.stop'] = QPushButton("STOP")
-        self.window.ui.nodes['global.stop'].setVisible(False)
-        self.window.ui.nodes['global.stop'].clicked.connect(self.window.controller.ui.on_global_stop)
-        self.window.ui.nodes['anim.loading.status'] = Loader()
-        self.window.ui.nodes['anim.loading.status'].hide()
+        nodes = self.window.ui.nodes
+
+        nodes['status'] = QLabel(trans('status.started'))
+        nodes['status'].setParent(self.window)
+
+        nodes['status.agent'] = HelpLabel("")
+        nodes['status.agent'].setParent(self.window)
+        nodes['status.agent'].setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Minimum)
+
+        nodes['global.stop'] = QPushButton("STOP")
+        nodes['global.stop'].setParent(self.window)
+        nodes['global.stop'].hide()
+        nodes['global.stop'].clicked.connect(self.window.controller.ui.on_global_stop)
+
+        nodes['anim.loading.status'] = Loader()
+        nodes['anim.loading.status'].setParent(self.window)
+        nodes['anim.loading.status'].hide()
 
         layout = QHBoxLayout()
-        layout.addWidget(self.window.ui.nodes['anim.loading.status'])
-        layout.addWidget(self.window.ui.nodes['status.agent'])
-        layout.addWidget(self.window.ui.nodes['status'])
-        layout.addWidget(self.window.ui.nodes['global.stop'])
+        layout.addWidget(nodes['anim.loading.status'])
+        layout.addWidget(nodes['status.agent'])
+        layout.addWidget(nodes['status'])
+        layout.addWidget(nodes['global.stop'])
         layout.setAlignment(Qt.AlignLeft)
 
-        return layout
+        return layout

pygpt_net/ui/layout/toolbox/agent.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.24 22:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
 from PySide6.QtWidgets import QVBoxLayout, QLabel, QHBoxLayout, QWidget, QCheckBox
@@ -31,49 +31,38 @@ class Agent:
 
         :return: QWidget
         """
-        # iterations
-        option = self.window.controller.agent.legacy.options["agent.iterations"]
-        self.window.ui.nodes['agent.iterations.label'] = QLabel(trans("toolbox.agent.iterations.label"))
-        self.window.ui.nodes['agent.iterations'] = \
-            OptionSlider(
-                self.window,
-                'global',
-                'agent.iterations',
-                option,
-            )
-        self.window.ui.config['global']['agent.iterations'] = self.window.ui.nodes['agent.iterations']
+        window = self.window
+        ui = window.ui
+        nodes = ui.nodes
+        cfg = ui.config['global']
+        common = window.controller.agent.common
 
-        # auto stop
-        self.window.ui.nodes['agent.auto_stop'] = ToggleLabel(trans("toolbox.agent.auto_stop.label"), parent=self.window)
-        self.window.ui.nodes['agent.auto_stop'].box.stateChanged.connect(
-            lambda:
-            self.window.controller.agent.common.toggle_auto_stop(
-                self.window.ui.config['global']['agent.auto_stop'].box.isChecked())
-        )
-        self.window.ui.config['global']['agent.auto_stop'] = self.window.ui.nodes['agent.auto_stop']
+        container = QWidget(window)
 
-        # continue more
-        self.window.ui.nodes['agent.continue'] = ToggleLabel(trans("toolbox.agent.continue.label"), parent=self.window)
-        self.window.ui.nodes['agent.continue'].box.stateChanged.connect(
-            lambda:
-            self.window.controller.agent.common.toggle_continue(
-                self.window.ui.config['global']['agent.continue'].box.isChecked())
-        )
-        self.window.ui.config['global']['agent.continue'] = self.window.ui.nodes['agent.continue']
+        option = window.controller.agent.legacy.options["agent.iterations"]
+        nodes['agent.iterations.label'] = QLabel(trans("toolbox.agent.iterations.label"), parent=container)
+        nodes['agent.iterations'] = OptionSlider(window, 'global', 'agent.iterations', option)
+        cfg['agent.iterations'] = nodes['agent.iterations']
+
+        nodes['agent.auto_stop'] = ToggleLabel(trans("toolbox.agent.auto_stop.label"), parent=window)
+        nodes['agent.auto_stop'].box.toggled.connect(common.toggle_auto_stop)
+        cfg['agent.auto_stop'] = nodes['agent.auto_stop']
+
+        nodes['agent.continue'] = ToggleLabel(trans("toolbox.agent.continue.label"), parent=window)
+        nodes['agent.continue'].box.toggled.connect(common.toggle_continue)
+        cfg['agent.continue'] = nodes['agent.continue']
 
-        # options
         cols = QHBoxLayout()
-        cols.addWidget(self.window.ui.config['global']['agent.auto_stop'])
-        cols.addWidget(self.window.ui.config['global']['agent.continue'])
+        cols.addWidget(cfg['agent.auto_stop'])
+        cols.addWidget(cfg['agent.continue'])
 
-        # rows
         rows = QVBoxLayout()
-        rows.addWidget(self.window.ui.nodes['agent.iterations.label'])
-        rows.addWidget(self.window.ui.config['global']['agent.iterations'])
+        rows.addWidget(nodes['agent.iterations.label'])
+        rows.addWidget(cfg['agent.iterations'])
         rows.addLayout(cols)
 
-        self.window.ui.nodes['agent.options'] = QWidget()
-        self.window.ui.nodes['agent.options'].setLayout(rows)
-        self.window.ui.nodes['agent.options'].setContentsMargins(0, 0, 0, 0)
+        nodes['agent.options'] = container
+        nodes['agent.options'].setLayout(rows)
+        nodes['agent.options'].setContentsMargins(0, 0, 0, 0)
 
-        return self.window.ui.nodes['agent.options']
+        return nodes['agent.options']

pygpt_net/ui/layout/toolbox/agent_llama.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.07.22 15:00:00                  #
+# Updated Date: 2025.08.24 23:00:00                  #
 # ================================================== #
 
 from PySide6.QtWidgets import QVBoxLayout, QLabel, QHBoxLayout, QWidget, QCheckBox
@@ -32,56 +32,52 @@ class AgentLlama:
 
         :return: QWidget
         """
-        # loop score
-        option = self.window.controller.agent.llama.options["agent.llama.loop.score"]
-        self.window.ui.nodes['agent.llama.loop.score.label'] = QLabel(trans("toolbox.agent.llama.loop.score.label"))
-        self.window.ui.nodes['agent.llama.loop.score'] = \
-            OptionSlider(
-                self.window,
-                'global',
-                'agent.llama.loop.score',
-                option,
-            )
-        self.window.ui.nodes['agent.llama.loop.score'].setToolTip(trans("toolbox.agent.llama.loop.score.tooltip"))
-        self.window.ui.config['global']['agent.llama.loop.score'] = self.window.ui.nodes['agent.llama.loop.score']
-
-        option_mode = self.window.controller.agent.llama.options["agent.llama.loop.mode"]
-        self.window.ui.nodes['agent.llama.loop.mode.label'] = QLabel(trans("toolbox.agent.llama.loop.mode.label"))
-        self.window.ui.nodes['agent.llama.loop.mode'] = \
-            OptionCombo(
-                self.window,
-                'global',
-                'agent.llama.loop.mode',
-                option_mode,
-            )
-        self.window.ui.nodes['agent.llama.loop.mode'].setToolTip(trans("toolbox.agent.llama.loop.mode.tooltip"))
-        self.window.ui.config['global']['agent.llama.loop.mode'] = self.window.ui.nodes['agent.llama.loop.mode']
-
-        # loop enabled
-        self.window.ui.nodes['agent.llama.loop.enabled'] = ToggleLabel(trans("toolbox.agent.llama.loop.enabled.label"), parent=self.window)
-        self.window.ui.nodes['agent.llama.loop.enabled'].box.stateChanged.connect(
-            lambda:
-            self.window.controller.agent.common.toggle_loop(
-                self.window.ui.config['global']['agent.llama.loop.enabled'].box.isChecked())
+        win = self.window
+        ui = win.ui
+        nodes = ui.nodes
+        config_global = ui.config.setdefault('global', {})
+        llama_opts = win.controller.agent.llama.options
+
+        container = QWidget(win)
+
+        nodes['agent.llama.loop.score'] = OptionSlider(
+            win,
+            'global',
+            'agent.llama.loop.score',
+            llama_opts["agent.llama.loop.score"],
+        )
+        nodes['agent.llama.loop.score'].setToolTip(trans("toolbox.agent.llama.loop.score.tooltip"))
+        config_global['agent.llama.loop.score'] = nodes['agent.llama.loop.score']
+
+        nodes['agent.llama.loop.mode.label'] = QLabel(trans("toolbox.agent.llama.loop.mode.label"), parent=win)
+        nodes['agent.llama.loop.mode'] = OptionCombo(
+            win,
+            'global',
+            'agent.llama.loop.mode',
+            llama_opts["agent.llama.loop.mode"],
+        )
+        nodes['agent.llama.loop.mode'].setToolTip(trans("toolbox.agent.llama.loop.mode.tooltip"))
+        config_global['agent.llama.loop.mode'] = nodes['agent.llama.loop.mode']
+
+        nodes['agent.llama.loop.enabled'] = ToggleLabel(trans("toolbox.agent.llama.loop.enabled.label"), parent=win)
+        nodes['agent.llama.loop.enabled'].box.toggled.connect(
+            win.controller.agent.common.toggle_loop
         )
-        self.window.ui.config['global']['agent.llama.loop.enabled'] = self.window.ui.nodes['agent.llama.loop.enabled']
+        config_global['agent.llama.loop.enabled'] = nodes['agent.llama.loop.enabled']
 
-        # label
-        self.window.ui.nodes['agent.llama.loop.label'] = QLabel(trans("toolbox.agent.llama.loop.label"))
+        nodes['agent.llama.loop.label'] = QLabel(trans("toolbox.agent.llama.loop.label"), parent=win)
 
-        # options
         cols = QHBoxLayout()
-        cols.addWidget(self.window.ui.config['global']['agent.llama.loop.enabled'])
-        cols.addWidget(self.window.ui.config['global']['agent.llama.loop.score'])
+        cols.addWidget(config_global['agent.llama.loop.enabled'])
+        cols.addWidget(config_global['agent.llama.loop.score'])
 
-        # rows
         rows = QVBoxLayout()
-        rows.addWidget(self.window.ui.nodes['agent.llama.loop.label'])
+        rows.addWidget(nodes['agent.llama.loop.label'])
         rows.addLayout(cols)
-        rows.addWidget(self.window.ui.nodes['agent.llama.loop.mode'])
+        rows.addWidget(nodes['agent.llama.loop.mode'])
 
-        self.window.ui.nodes['agent_llama.options'] = QWidget()
-        self.window.ui.nodes['agent_llama.options'].setLayout(rows)
-        self.window.ui.nodes['agent_llama.options'].setContentsMargins(0, 0, 0, 0)
+        container.setLayout(rows)
+        container.setContentsMargins(0, 0, 0, 0)
 
-        return self.window.ui.nodes['agent_llama.options']
+        nodes['agent_llama.options'] = container
+        return nodes['agent_llama.options']