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

pygpt_net/controller/presets/editor.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.09.27 00:00:00                  #
+# Updated Date: 2025.09.28 09:35:00                  #
 # ================================================== #
 
 import datetime
@@ -14,8 +14,9 @@ import os
 import shutil
 from typing import Any, Optional, Dict
 
-from PySide6.QtCore import Slot
-from PySide6.QtWidgets import QVBoxLayout, QWidget, QHBoxLayout
+from PySide6.QtCore import Slot, Qt
+from PySide6.QtWidgets import QVBoxLayout, QWidget, QHBoxLayout, QScrollArea, QFrame
+from PySide6.QtGui import QImageReader
 
 from pygpt_net.core.types import (
     MODE_AGENT,
@@ -573,10 +574,19 @@ class Editor:
                 layout.addStretch(1)
                 layout.addLayout(checkbox_layout)
 
-                # as tab
-                tab_widget = QWidget()
-                tab_widget.setLayout(layout)
-                tabs.addTab(tab_widget, title)
+                # wrap the tab content in a scroll area to avoid vertical overlaps
+                tab_content = QWidget()
+                tab_content.setLayout(layout)
+
+                scroll = QScrollArea()
+                scroll.setWidgetResizable(True)
+                scroll.setFrameShape(QFrame.NoFrame)
+                scroll.setWidget(tab_content)
+                # Attach metadata on the tab widget itself for later mapping.
+                scroll.setProperty("agent_id", id)
+                scroll.setProperty("option_tab_id", option_tab_id)
+
+                tabs.addTab(scroll, title)
 
                 # store mapping: agent id -> [tab index]
                 if id not in self.tab_options_idx:
@@ -747,15 +757,20 @@ class Editor:
             layout.addStretch(1)
             layout.addLayout(checkbox_layout)
 
-            # Assemble tab widget and tag it with metadata.
-            tab_widget = QWidget()
-            tab_widget.setLayout(layout)
-            tab_widget.setProperty('agent_id', agent_id)
-            tab_widget.setProperty('option_tab_id', option_tab_id)
+            # Assemble tab widget wrapped into a scroll area.
+            tab_content = QWidget()
+            tab_content.setLayout(layout)
+
+            scroll = QScrollArea()
+            scroll.setWidgetResizable(True)
+            scroll.setFrameShape(QFrame.NoFrame)
+            scroll.setWidget(tab_content)
+            scroll.setProperty('agent_id', agent_id)
+            scroll.setProperty('option_tab_id', option_tab_id)
 
             # Insert at a stable anchor to preserve general ordering between agents.
             insertion_index = min(insertion_index, tabs.count())
-            tabs.insertTab(insertion_index, tab_widget, title)
+            tabs.insertTab(insertion_index, scroll, title)
             new_indices.append(insertion_index)
             insertion_index += 1
 
@@ -1297,17 +1312,26 @@ class Editor:
             os.remove(avatar_path)
         if os.path.exists(file_path):
             shutil.copy(file_path, avatar_path)
+
+            # create thumbnail next to original (max 350px on the longest side, prefix: thumb_)
+            thumb_path = self._create_avatar_thumbnail(avatar_path)
+
             if preset:
                 preset.ai_avatar = store_name
             else:
                 self.tmp_avatar = store_name
+
+            # keep storing original filename in config/preset (UI will prefer thumbnail only for preview if available)
             self.window.controller.config.apply_value(
                 parent_id=self.id,
                 key="ai_avatar",
                 option=self.options["ai_avatar"],
                 value=store_name,
             )
-            avatar_widget.load_avatar(avatar_path)
+
+            # prefer thumbnail in UI preview if it exists, fall back to original
+            preview_path = thumb_path if thumb_path and os.path.exists(thumb_path) else avatar_path
+            avatar_widget.load_avatar(preview_path)
             avatar_widget.enable_remove_button(True)
         return avatar_path
 
@@ -1325,11 +1349,19 @@ class Editor:
                 "avatars",
                 avatar_path,
             )
+            # prefer thumbnail in UI preview if available
+            thumb_path = os.path.join(
+                self.window.core.config.get_user_dir("presets"),
+                "avatars",
+                f"thumb_{avatar_path}",
+            )
+            preview_path = thumb_path if os.path.exists(thumb_path) else file_path
+
             if not os.path.exists(file_path):
                 avatar_widget.remove_avatar()
                 print("Avatar file does not exist:", file_path)
                 return
-            avatar_widget.load_avatar(file_path)
+            avatar_widget.load_avatar(preview_path)
             avatar_widget.enable_remove_button(True)
         else:
             avatar_widget.remove_avatar()
@@ -1354,10 +1386,25 @@ class Editor:
                 presets_dir = self.window.core.config.get_user_dir("presets")
                 avatars_dir = os.path.join(presets_dir, "avatars")
                 avatar_path = os.path.join(avatars_dir, current)
+                thumb_path = os.path.join(avatars_dir, f"thumb_{current}")
+                # remove original
                 if os.path.exists(avatar_path):
                     os.remove(avatar_path)
+                # remove thumbnail (if exists)
+                if os.path.exists(thumb_path):
+                    os.remove(thumb_path)
             preset.ai_avatar = ""
         else:
+            # if not yet persisted, also drop any staged temp avatar
+            if self.tmp_avatar:
+                presets_dir = self.window.core.config.get_user_dir("presets")
+                avatars_dir = os.path.join(presets_dir, "avatars")
+                avatar_path = os.path.join(avatars_dir, self.tmp_avatar)
+                thumb_path = os.path.join(avatars_dir, f"thumb_{self.tmp_avatar}")
+                if os.path.exists(avatar_path):
+                    os.remove(avatar_path)
+                if os.path.exists(thumb_path):
+                    os.remove(thumb_path)
             self.tmp_avatar = None
 
         self.window.ui.nodes['preset.editor.avatar'].remove_avatar()
@@ -1500,15 +1547,20 @@ class Editor:
                     layout.addStretch(1)
                     layout.addLayout(checkbox_layout)
 
-                    # Create tab widget and tag with metadata
-                    tab_widget = QWidget()
-                    tab_widget.setLayout(layout)
-                    tab_widget.setProperty('agent_id', a_id)
-                    tab_widget.setProperty('option_tab_id', option_tab_id)
+                    # Create tab widget wrapped in a scroll area and tag with metadata
+                    tab_content = QWidget()
+                    tab_content.setLayout(layout)
+
+                    scroll = QScrollArea()
+                    scroll.setWidgetResizable(True)
+                    scroll.setFrameShape(QFrame.NoFrame)
+                    scroll.setWidget(tab_content)
+                    scroll.setProperty('agent_id', a_id)
+                    scroll.setProperty('option_tab_id', option_tab_id)
 
                     # Insert tab and advance anchor
                     insertion_index = min(insertion_index, tabs.count())
-                    tabs.insertTab(insertion_index, tab_widget, title)
+                    tabs.insertTab(insertion_index, scroll, title)
                     insertion_index += 1
 
                     # Apply saved values (if present) or defaults
@@ -1592,4 +1644,67 @@ class Editor:
                         widget.set_value(default_val)
                 except Exception:
                     # Silent fallback; apply_value above should already handle most cases
-                    pass
+                    pass
+
+    # ---------- Avatar thumbnail helpers ----------
+
+    def _create_avatar_thumbnail(self, src_path: str, max_px: int = 350) -> Optional[str]:
+        """
+        Create a thumbnail next to the original avatar file with prefix 'thumb_'.
+
+        Notes:
+        - Respects EXIF orientation via QImageReader.setAutoTransform(True).
+        - Keeps aspect ratio; the longer side is limited to max_px.
+        - Saves using the same extension as the original (derived from destination filename).
+        - Returns full path to the thumbnail or None on failure.
+        """
+        try:
+            if not os.path.exists(src_path):
+                return None
+
+            dir_name, base_name = os.path.split(src_path)
+            thumb_name = f"thumb_{base_name}"
+            thumb_path = os.path.join(dir_name, thumb_name)
+
+            # Clean previous thumbnail (if any) to avoid stale previews
+            if os.path.exists(thumb_path):
+                try:
+                    os.remove(thumb_path)
+                except Exception:
+                    pass
+
+            reader = QImageReader(src_path)
+            reader.setAutoTransform(True)  # apply EXIF rotation if present
+            image = reader.read()
+            if image.isNull():
+                # If reading failed, fallback to copying original (not resized)
+                shutil.copy(src_path, thumb_path)
+                return thumb_path
+
+            w = image.width()
+            h = image.height()
+            if w <= 0 or h <= 0:
+                return None
+
+            # Scale down only when needed; always keep aspect ratio and smooth transformation
+            if max(w, h) > max_px:
+                scaled = image.scaled(max_px, max_px, Qt.KeepAspectRatio, Qt.SmoothTransformation)
+                ok = scaled.save(thumb_path)
+            else:
+                ok = image.save(thumb_path)
+
+            if ok:
+                return thumb_path
+
+            # Last resort: copy original
+            shutil.copy(src_path, thumb_path)
+            return thumb_path
+        except Exception as e:
+            # As a safe fallback try to copy the original; ignore errors silently
+            try:
+                dir_name, base_name = os.path.split(src_path)
+                thumb_path = os.path.join(dir_name, f"thumb_{base_name}")
+                shutil.copy(src_path, thumb_path)
+                return thumb_path
+            except Exception:
+                return None

pygpt_net/controller/presets/presets.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.09.26 03:00:00                  #
+# Updated Date: 2025.12.25 20:00:00                  #
 # ================================================== #
 
 import re
@@ -497,8 +497,10 @@ class Presets:
 
         :param no_scroll: do not scroll to current
         """
+        self.locked = True
         w = self.window
         if w.core.config.get('mode') == MODE_ASSISTANT:
+            self.locked = False
             return
         if no_scroll:
             w.ui.nodes['preset.presets'].store_scroll_position()
@@ -511,6 +513,7 @@ class Presets:
         if no_scroll:
             w.ui.nodes['preset.presets'].restore_scroll_position()
         self.on_changed()
+        self.locked = False
 
     def update_list(self):
         """Update presets list"""

pygpt_net/controller/ui/mode.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.09.27 15:00:00                  #
+# Updated Date: 2025.12.25 20:00:00                  #
 # ================================================== #
 
 from pygpt_net.core.types import (
@@ -39,7 +39,8 @@ class Mode:
     def update(self):
         """Update mode, model, preset and rest of the toolbox"""
 
-        mode = self.window.core.config.data['mode']
+        mode = self.window.core.config.get("mode")
+        model = self.window.core.config.get("model")
 
         ui_nodes = self.window.ui.nodes
         ui_tabs = self.window.ui.tabs
@@ -53,7 +54,7 @@ class Mode:
         is_agent_llama = mode == MODE_AGENT_LLAMA
         is_agent_openai = mode == MODE_AGENT_OPENAI
         is_expert = mode == MODE_EXPERT
-        is_image = mode == MODE_IMAGE
+        is_media = mode == MODE_IMAGE
         is_llama_index = mode == MODE_LLAMA_INDEX
         is_completion = mode == MODE_COMPLETION
         is_audio = mode == MODE_AUDIO
@@ -161,15 +162,18 @@ class Mode:
             ui_nodes['preset.editor.modes'].setVisible(True)
             ui_tabs['preset.editor.extra'].setTabText(0, trans("preset.prompt"))
 
-        # image options visibility
-        if is_image:
+        # media options visibility
+        if is_media:
             ui_nodes['media.raw'].setVisible(True)
-            if ctrl.media.is_video_model():
+            if ctrl.media.is_video_model() and ctrl.media.get_mode() == "video":
                 ui_nodes['video.options'].setVisible(True)
                 ui_nodes['dalle.options'].setVisible(False)
-            elif ctrl.media.is_image_model():
+            elif ctrl.media.is_image_model() and ctrl.media.get_mode() == "image":
                 ui_nodes['dalle.options'].setVisible(True)
                 ui_nodes['video.options'].setVisible(False)
+            elif ctrl.media.get_mode() == "music":
+                ui_nodes['dalle.options'].setVisible(False)
+                ui_nodes['video.options'].setVisible(False)
             else:
                 ui_nodes['media.raw'].setVisible(False)
                 ui_nodes['dalle.options'].setVisible(False)
@@ -199,7 +203,7 @@ class Mode:
         else:
             ui_nodes['idx.options'].setVisible(False)
 
-        if is_image:
+        if is_media:
             ui_nodes['input.stream'].setVisible(False)
         else:
             ui_nodes['input.stream'].setVisible(True)
@@ -213,13 +217,13 @@ class Mode:
         ui_tabs['input'].setTabVisible(1, show)
 
         # remote tools icon visibility
-        if not is_image and not is_completion:
+        if not is_media and not is_completion:
             self.window.ui.nodes['input'].set_icon_visible("web", True)
         else:
             self.window.ui.nodes['input'].set_icon_visible("web", False)
 
         ui_tabs['input'].setTabVisible(2, is_assistant)
-        ui_tabs['input'].setTabVisible(3, (not is_assistant) and (not is_image))
+        ui_tabs['input'].setTabVisible(3, (not is_assistant) and (not is_media))
 
         presets_editor.toggle_extra_options()
 

pygpt_net/controller/ui/ui.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2025.09.26 17:00:00                  #
+# Updated Date: 2025.12.25 20:00:00                  #
 # ================================================== #
 
 from typing import Optional
@@ -68,6 +68,7 @@ class UI:
         self.update_tokens()
         self.vision.update()
         self.window.controller.agent.legacy.update()
+        self.img_update_available_modes()
         self.img_update_available_resolutions()
 
     def handle(self, event: BaseEvent):
@@ -258,4 +259,20 @@ class UI:
             key="img_resolution",
             option=self.window.core.image.get_resolution_option(),
             value=current,
+        )
+
+    def img_update_available_modes(self):
+        """Update available modes for images"""
+        mode = self.window.core.config.get('mode')
+        if mode != MODE_IMAGE:
+            return
+        model = self.window.core.config.get('model')
+        keys = self.window.core.image.get_available_modes(model)
+        current = self.window.core.config.get('img_mode', 'image')
+        self.window.ui.config['global']['img_mode'].set_keys(keys, lock=False)
+        self.window.controller.config.apply_value(
+            parent_id="global",
+            key="img_mode",
+            option=self.window.core.image.get_mode_option(),
+            value=current,
         )

pygpt_net/core/agents/custom/__init__.py

@@ -184,8 +184,9 @@ class Custom:
                                 tab["label"] = slots["name"]
                             if "role" in slots:
                                 opts["role"] = {
-                                    "type": "str",
-                                    "label": trans("agent.option.role"),
+                                    "type": "text",
+                                    "label": trans("agent.option.role.label"),
+                                    "description": trans("agent.option.role"),
                                     "default": slots["role"],
                                 }
                             if "instruction" in slots:
@@ -213,6 +214,21 @@ class Custom:
             except Exception as e:
                 self.window.core.debug.log(f"Failed to build options for custom agent '{agent_id}': {e}")
                 continue
+
+        # debug tab - trace_id, etc.
+        """
+        options["debug"] = {
+            "label": trans("agent.tab.debug"),
+            "options": {
+                "trace_id": {
+                    "type": "text",
+                    "label": trans("agent.option.debug.trace_id"),
+                    "description": trans("agent.option.debug.trace_id.desc"),
+                    "default": "",
+                }
+            }
+        }
+        """
         return options
 
     def new_agent(self, name: str):

pygpt_net/core/agents/custom/runner.py

@@ -337,8 +337,8 @@ class FlowOrchestrator:
                 "input": prepared_items,
                 "max_turns": int(agent_kwargs.get("max_iterations", max_iterations)),
             }
-            if trace_id:
-                run_kwargs["trace_id"] = trace_id
+            # if trace_id:
+                # run_kwargs["trace_id"] = trace_id
 
             # Header for UI
             ctx.set_agent_name(agent.name)

pygpt_net/core/attachments/clipboard.py

@@ -0,0 +1,146 @@
+#!/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 08:00:00                  #
+# ================================================== #
+
+from PySide6.QtCore import QObject, QEvent, Qt
+
+
+class AttachmentDropHandler(QObject):
+    """
+    Generic drag & drop handler for attaching files/images/urls/text.
+
+    Policies:
+    - SWALLOW_ALL: always consume the drop (e.g., attachments list).
+    - INPUT_MIX  : for ChatInput; process attachments and:
+        * swallow image payloads (no text insert),
+        * allow default handling for non-image payloads so paths/text get inserted.
+    """
+    SWALLOW_ALL = 0
+    INPUT_MIX = 1
+
+    def __init__(self, window, target_widget, policy=SWALLOW_ALL):
+        super().__init__(target_widget)
+        self.window = window
+        self._target = target_widget
+        self._policy = policy
+
+        # Accept drops on target and its viewport (important for QTextEdit/QAbstractScrollArea)
+        self._enable_drops(self._target)
+        vp = self._get_viewport(self._target)
+        if vp is not None:
+            self._enable_drops(vp)
+
+        # Install filters on both
+        self._target.installEventFilter(self)
+        if vp is not None:
+            vp.installEventFilter(self)
+
+    def _enable_drops(self, w):
+        try:
+            w.setAcceptDrops(True)
+        except Exception:
+            pass
+
+    def _get_viewport(self, w):
+        try:
+            vp = getattr(w, "viewport", None)
+            if callable(vp):
+                return vp()
+            return None
+        except Exception:
+            return None
+
+    def _mime_supported(self, md) -> bool:
+        try:
+            if md is None:
+                return False
+            return md.hasUrls() or md.hasImage() or md.hasText()
+        except Exception:
+            return False
+
+    def _process_drop(self, md):
+        """
+        Route to ChatInput.handle_clipboard() to reuse existing attach pipeline.
+        """
+        try:
+            chat_input = self.window.ui.nodes.get('input')
+        except Exception:
+            chat_input = None
+
+        if chat_input is not None and hasattr(chat_input, 'handle_clipboard'):
+            try:
+                chat_input.handle_clipboard(md)
+                return chat_input
+            except Exception as e:
+                try:
+                    self.window.core.debug.log(e)
+                except Exception:
+                    pass
+        return None
+
+    def _allow_default_text_insert_for_non_image(self, md) -> bool:
+        try:
+            return not (md and md.hasImage())
+        except Exception:
+            return True
+
+    def eventFilter(self, obj, event):
+        # Only handle events coming to the target or its viewport
+        if obj is not self._target and obj is not self._get_viewport(self._target):
+            return False
+
+        et = event.type()
+
+        if et in (QEvent.DragEnter, QEvent.DragMove):
+            md = getattr(event, 'mimeData', lambda: None)()
+            if self._mime_supported(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_supported(md):
+                return False
+
+            chat_input = self._process_drop(md)
+
+            try:
+                event.setDropAction(Qt.CopyAction)
+                event.acceptProposedAction()
+            except Exception:
+                event.accept()
+
+            # Policy decision:
+            if self._policy == self.SWALLOW_ALL:
+                # Consume the event; nothing else should handle it.
+                return True
+
+            if self._policy == self.INPUT_MIX:
+                # For non-image payloads we allow default to insert text/paths into input.
+                # To avoid duplicate attachments (insertFromMimeData calls handle_clipboard),
+                # set a one-shot guard flag.
+                if chat_input is not None and self._allow_default_text_insert_for_non_image(md):
+                    try:
+                        chat_input._skip_clipboard_on_next_insert = True
+                    except Exception:
+                        pass
+                    return False  # let default drop insert text/paths
+                else:
+                    return True  # swallow images
+
+            # Default: swallow
+            return True
+
+        return False

pygpt_net/core/image/image.py

@@ -6,7 +6,7 @@
 # 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                  #
 # ================================================== #
 
 import uuid
@@ -158,6 +158,39 @@ class Image(QObject):
             "keys": self.get_available_resolutions(),
         }
 
+    def get_mode_option(self) -> dict:
+        """
+        Get image mode option for UI
+
+        :return: dict
+        """
+        return {
+            "type": "combo",
+            "slider": True,
+            "label": "img_mode",
+            "value": "image",
+            "keys": self.get_available_modes(),
+        }
+
+    def get_available_modes(self, model_name: str = None) -> Dict[str, str]:
+        """
+        Get available modes (image / video)
+
+        :param model_name: model name
+        :return: dict of available modes
+        """
+        options = {}
+        if model_name:
+            model = self.window.core.models.get(model_name)
+            if model:
+                if model.is_image_output():
+                    options["image"] = trans("mode.img.image")
+                if model.is_video_output():
+                    options["video"] = trans("mode.img.video")
+                if model.is_music_output():
+                    options["music"] = trans("mode.img.music")
+        return options
+
     def get_available_resolutions(self, model: str = None) -> Dict[str, str]:
         """
         Get available image resolutions