pygpt-net 2.4.37__py3-none-any.whl → 2.4.39__py3-none-any.whl

pygpt_net/core/attachments/context.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.29 23:00:00                  #
+# Updated Date: 2024.11.30 04:00:00                  #
 # ================================================== #
 
 import copy
@@ -119,6 +119,8 @@ class Context:
                 if ("type" not in file
                         or file["type"] not in ["local_file", "url"]):
                     continue
+                if not "uuid" in file:
+                    continue
                 file_id = file["uuid"]
                 file_idx_path = os.path.join(meta_path, file_id)
                 text_path = os.path.join(file_idx_path, file_id + ".txt")
@@ -555,17 +557,6 @@ class Context:
         if meta is not None:
             self.delete_index(meta)
 
-    def reset_by_meta_id(self, meta_id: int, delete_files: bool = False):
-        """
-        Delete all attachments for meta by id
-
-        :param meta_id: Meta id
-        :param delete_files: delete files
-        """
-        meta = self.window.core.ctx.get_meta_by_id(meta_id)
-        if meta is not None:
-            self.reset_by_meta(meta, delete_files)
-
     def reset_by_meta(self, meta: CtxMeta, delete_files: bool = False):
         """
         Delete all attachments for meta
@@ -578,6 +569,17 @@ class Context:
         if delete_files:
             self.delete_index(meta)
 
+    def reset_by_meta_id(self, meta_id: int, delete_files: bool = False):
+        """
+        Delete all attachments for meta by id
+
+        :param meta_id: Meta id
+        :param delete_files: delete files
+        """
+        meta = self.window.core.ctx.get_meta_by_id(meta_id)
+        if meta is not None:
+            self.reset_by_meta(meta, delete_files)
+
     def clear(self, meta: CtxMeta, delete_files: bool = False):
         """
         Clear all attachments by ctx meta

pygpt_net/core/audio/__init__.py

@@ -6,10 +6,11 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.26 19:00:00                  #
+# Updated Date: 2024.12.08 00:00:00                  #
 # ================================================== #
 
 import re
+from bs4 import UnicodeDammit
 
 from pygpt_net.provider.audio_input.base import BaseProvider as InputBaseProvider
 from pygpt_net.provider.audio_output.base import BaseProvider as OutputBaseProvider
@@ -27,6 +28,55 @@ class Audio:
             "input": {},
             "output": {},
         }
+        self.last_error = None
+
+    def get_input_devices(self) -> list:
+        """
+        Get input devices
+
+        :return devices list: [(id, name)]
+        """
+        import pyaudio
+        devices = []
+        try:
+            p = pyaudio.PyAudio()
+            num_devices = p.get_device_count()
+            for i in range(num_devices):
+                info = p.get_device_info_by_index(i)
+                if info["maxInputChannels"] > 0:
+                    dammit = UnicodeDammit(info["name"])
+                    devices.append((i, dammit.unicode_markup))
+                    # print(f"Device ID {i}: {info['name']}")
+            p.terminate()
+        except Exception as e:
+            print(f"Audio input devices receive error: {e}")
+        return devices
+
+    def is_device_compatible(self, device_index) -> bool:
+        """
+        Check if device is compatible
+
+        :param device_index: device index
+        :return: True if compatible
+        """
+        import pyaudio
+        rate = int(self.window.core.config.get('audio.input.rate', 44100))
+        channels = int(self.window.core.config.get('audio.input.channels', 1))
+        p = pyaudio.PyAudio()
+        info = p.get_device_info_by_index(device_index)
+        supported = False
+        try:
+            p.is_format_supported(
+                rate=rate,
+                input_device=info['index'],
+                input_channels=channels,
+                input_format=pyaudio.paInt16)
+            supported = True
+        except ValueError as e:
+            self.last_error = str(e)
+            supported = False
+        p.terminate()
+        return supported
 
     def is_registered(self, id: str, type: str = "output") -> bool:
         """
@@ -92,3 +142,11 @@ class Audio:
         :return: cleaned text
         """
         return re.sub(r'~###~.*?~###~', '', str(text))
+
+    def get_last_error(self) -> str:
+        """
+        Return last error
+
+        :return: Error
+        """
+        return self.last_error

pygpt_net/core/ctx/__init__.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 19:00:00                  #
+# Updated Date: 2024.12.09 00:00:00                  #
 # ================================================== #
 
 import copy
@@ -490,6 +490,16 @@ class Ctx:
         for item in self.get_items():
             if item.id == id:
                 return item
+        return self.fetch_item_by_id(id)   # if no item found, try to fetch from DB
+
+    def fetch_item_by_id(self, id: int) -> CtxItem:
+        """
+        Fetch ctx item by id
+
+        :param id: item id
+        :return: context item
+        """
+        return self.provider.get_item_by_id(id)
 
     def get_meta(self, reload: bool = False) -> dict:
         """

pygpt_net/core/ctx/container.py

@@ -6,39 +6,44 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.07 23:00:00                  #
+# Updated Date: 2024.12.09 00:00:00                  #
 # ================================================== #
 
 from PySide6.QtWidgets import QVBoxLayout
 
 from .bag import Bag
 from pygpt_net.ui.widget.textarea.output import ChatOutput
+from ..tabs import Tab
 
 
 class Container:
     def __init__(self, window=None):
         """
-        Context container
+        Context output container
+
+        :param window: Window
         """
         self.window = window
         self.bags = {}
         self.bags[0] = Bag(window)  # always create initial bag
 
-    def register_output(self, id: int = 0):
+    def get(self, tab: Tab):
         """
-        Register output
+        Register and return output
 
-        :param id: ID
+        :param tab: Tab
         :return: Widget
         """
         # plain output
         output_plain = ChatOutput(self.window)
+        output_plain.set_tab(tab)
 
         # web
         if self.window.core.config.get("render.engine") == "web":
             from pygpt_net.ui.widget.textarea.web import ChatWebOutput, CustomWebEnginePage
             # build output
             output_html = ChatWebOutput(self.window)
+            output_html.set_tab(tab)
             output_html.setPage(
                 CustomWebEnginePage(self.window, output_html)
             )
@@ -48,13 +53,15 @@ class Container:
         else:
             # legacy
             output_html = ChatOutput(self.window)
+            output_html.set_tab(tab)
 
         if 'output_plain' not in self.window.ui.nodes:
             self.window.ui.nodes['output_plain'] = {}
         if 'output' not in self.window.ui.nodes:
             self.window.ui.nodes['output'] = {}
-        self.window.ui.nodes['output_plain'][id] = output_plain
-        self.window.ui.nodes['output'][id] = output_html
+
+        self.window.ui.nodes['output_plain'][tab.pid] = output_plain
+        self.window.ui.nodes['output'][tab.pid] = output_html
 
         # show/hide plain/html
         if self.window.core.config.get("render.plain") is True:
@@ -66,8 +73,8 @@ class Container:
 
         # build layout
         layout = QVBoxLayout()
-        layout.addWidget(self.window.ui.nodes['output_plain'][id])
-        layout.addWidget(self.window.ui.nodes['output'][id])
+        layout.addWidget(self.window.ui.nodes['output_plain'][tab.pid])
+        layout.addWidget(self.window.ui.nodes['output'][tab.pid])
         layout.setContentsMargins(0, 0, 0, 0)
         widget = self.window.core.tabs.from_layout(layout)
         return widget

pygpt_net/core/ctx/output.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.05 23:00:00                  #
+# Updated Date: 2024.12.09 03:00:00                  #
 # ================================================== #
 
 from pygpt_net.item.ctx import CtxItem, CtxMeta
@@ -16,11 +16,28 @@ class Output:
     def __init__(self, window=None):
         """
         Context output mapping
+
+        :param window: Window
         """
         self.window = window
-        self.mapping = {}  # pid => meta.id
+        self.mapping = {}  # [column_idx => [pid => meta.id]]
         self.last_pids = {}  # meta.id => pid
         self.last_pid = 0 # last used PID
+        self.initialized = False
+
+    def init(self, force: bool = False):
+        """
+        Initialize mappings
+
+        :param force: Force reinitialize
+        """
+        if self.initialized and not force:
+            return
+        self.clear()
+        for n in range(0, self.window.core.tabs.NUM_COLS):
+            self.mapping[n] = {}
+            self.last_pids[n] = {}
+        self.initialized = True
 
     def store(self, meta: CtxMeta) -> int:
         """
@@ -29,31 +46,20 @@ class Output:
         :param meta: Meta
         :return: PID
         """
+        self.init()
         pid = self.window.core.tabs.get_active_pid()
-        # check if allowed tab
         tab = self.window.core.tabs.get_tab_by_pid(pid)
         if tab is None or tab.type != Tab.TAB_CHAT:
             return 0
-        self.mapping[pid] = meta.id
-        self.last_pids[meta.id] = pid
+        col_idx = tab.column_idx
+        self.mapping[col_idx][pid] = meta.id
+        self.last_pids[col_idx][meta.id] = pid
         self.last_pid = pid
         tab = self.window.core.tabs.get_tab_by_pid(pid)
         if tab is not None:
             tab.data_id = meta.id  # store meta id in tab data
         return pid
 
-    def meta_id_in_tab(self, meta_id: int) -> bool:
-        """
-        Check if meta id is in tab
-
-        :param meta_id: Meta ID
-        :return: bool
-        """
-        for pid, mid in self.mapping.items():
-            if mid == meta_id:
-                return True
-        return False
-
     def is_mapped(self, meta: CtxMeta) -> bool:
         """
         Check if meta is mapped
@@ -61,19 +67,46 @@ class Output:
         :param meta: Meta
         :return: bool
         """
-        return meta.id in self.mapping.values()
+        self.init()
+        for col_idx in self.mapping:
+            if meta.id in self.mapping[col_idx].values():
+                return True
+        return False
 
-    def get_meta(self, pid: int) -> CtxMeta or None:
+    def get_meta(self, pid: int) -> int or None:
         """
         Get meta by PID
 
         :param pid: PID
-        :return: Meta or None
+        :return: meta ID or None
         """
-        if pid in self.mapping:
-            return self.mapping[pid]
+        self.init()
+        for col_idx in self.mapping:
+            if pid in self.mapping[col_idx]:
+                return self.mapping[col_idx][pid]
         return
 
+    def prepare_meta(self, tab: Tab) -> int or None:
+        """
+        Get meta ID by PID
+
+        :param tab: Tab
+        :return: Meta ID or None
+        """
+        self.init()
+        pid = tab.pid
+        col_idx = tab.column_idx
+        meta_id = None
+        if pid in self.mapping[col_idx]:
+            return self.mapping[col_idx][pid]
+        if meta_id is None:
+            meta_id = tab.data_id
+            if meta_id is not None:
+                self.mapping[col_idx][pid] = meta_id
+                self.last_pids[col_idx][meta_id] = pid
+                self.last_pid = pid
+        return meta_id
+
     def get_mapped(self, meta: CtxMeta) -> int or None:
         """
         Get PID by meta
@@ -81,75 +114,59 @@ class Output:
         :param meta: Meta
         :return: PID or None
         """
+        self.init()
         all = []
         pid = None
         active_pid = self.window.core.tabs.get_active_pid()
-        for pid, meta_id in self.mapping.items():
+        tab = self.window.core.tabs.get_tab_by_pid(active_pid)
+        if tab is None:
+            return None
+        col_idx = tab.column_idx
+        for pid, meta_id in self.mapping[col_idx].items():
             if meta_id == meta.id:
                 all.append(pid)
         for pid in all:
-            if pid == active_pid:  # at first check if active
+            if pid == active_pid:  # at first check for current tab
                 return pid
         return pid
 
-    def get_mapped_last(self, meta: CtxMeta) -> int or None:
-        """
-        Get last PID by meta
-
-        :param meta: Meta
-        :return: PID or None
-        """
-        active_pid = self.window.core.tabs.get_active_pid()
-        if self.mapping[active_pid] == meta.id:
-            return active_pid  # return active PID at first
-        if meta.id in self.last_pids:
-            return self.last_pids[meta.id]
-        return
-
-    def prepare(self, meta: CtxMeta):
-        """
-        Clear mapping
-
-        :param meta: meta to prepare
-        """
-        active_pid = self.window.core.tabs.get_active_pid()
-        if active_pid in self.mapping:
-            del self.mapping[active_pid]  # remove from mapping
-
-    def clear(self):
-        """
-        Clear mapping
-        """
-        self.mapping = {}
-        self.last_pids = {}
-        self.last_pid = 0
-
     def is_empty(self) -> bool:
         """
         Check if mapping is empty
 
         :return: bool
         """
+        self.init()
         active_pid = self.window.core.tabs.get_active_pid()
-        if active_pid in self.mapping:
-            return False
+        for col_idx in list(self.mapping.keys()):
+            if active_pid in self.mapping[col_idx]:
+                return False
         return True
 
-    def get_pid(self, meta: CtxMeta = None) -> int:
+    def get_pid(self, meta: CtxMeta = None) -> int or None:
         """
         Get PID by meta
 
         :param meta: Meta
         :return: PID
         """
+        self.init()
+        active_pid = self.window.core.tabs.get_active_pid()
         if meta is None:
-            return self.window.core.tabs.get_active_pid()
+            return None
         if self.is_mapped(meta):
-            if self.window.core.tabs.get_active_pid() != self.get_mapped(meta):  # if loaded in another tab
-                return self.store(meta)  # update to current
-            return self.get_mapped_last(meta)
-        else:
-            return self.store(meta)
+            if active_pid == self.get_mapped(meta):  # if loaded in current tab
+                return active_pid  # return current
+        return self.store(meta)
+
+    def clear(self):
+        """
+        Clear mapping
+        """
+        self.mapping = {}
+        self.last_pids = {}
+        self.last_pid = 0
+        self.initialized = False
 
     def get_current(self, meta: CtxMeta = None):
         """
@@ -176,7 +193,8 @@ class Output:
         if pid in self.window.ui.nodes['output_plain']:
             return self.window.ui.nodes['output_plain'][pid]
         else:
-            return self.window.ui.nodes['output_plain'][0]
+            for pid in self.window.ui.nodes['output_plain']:
+                return self.window.ui.nodes['output_plain'][pid]
 
     def get_by_pid(self, pid = None):
         """
@@ -201,7 +219,8 @@ class Output:
         if pid in self.window.ui.nodes['output_plain']:
             return self.window.ui.nodes['output_plain'][pid]
         else:
-            return self.window.ui.nodes['output_plain'][0]
+            for pid in self.window.ui.nodes['output_plain']:
+                return self.window.ui.nodes['output_plain'][pid]
 
     def get_all(self) -> list:
         """

pygpt_net/core/debug/tabs.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.07 23:00:00                  #
+# Updated Date: 2024.12.09 00:00:00                  #
 # ================================================== #
 
 class TabsDebug:
@@ -25,6 +25,7 @@ class TabsDebug:
         self.window.core.debug.add(self.id, 'current PID', str(self.window.controller.ui.tabs.get_current_pid()))
         self.window.core.debug.add(self.id, 'current IDX', str(self.window.controller.ui.tabs.get_current_idx()))
         self.window.core.debug.add(self.id, 'current Type', str(self.window.controller.ui.tabs.get_current_type()))
+        self.window.core.debug.add(self.id, 'current Column', str(self.window.controller.ui.tabs.get_current_column_idx()))
         self.window.core.debug.add(self.id, 'last_pid', str(self.window.core.tabs.last_pid))
         self.window.core.debug.add(self.id, 'count(pids)', str(len(self.window.core.tabs.pids)))
         self.window.core.debug.add(self.id, 'count(ctx bags)', str(len(self.window.core.ctx.container.bags)))
@@ -38,6 +39,6 @@ class TabsDebug:
 
         # mapping PID => meta.id
         self.window.core.debug.add(self.id, 'PID => meta.id', str(self.window.core.ctx.output.mapping))
-        self.window.core.debug.add(self.id, '(last) PID => meta.id', str(self.window.core.ctx.output.last_pids))
+        self.window.core.debug.add(self.id, '(last) meta.id => PID', str(self.window.core.ctx.output.last_pids))
         self.window.core.debug.add(self.id, '(last) PID', str(self.window.core.ctx.output.last_pid))
         self.window.core.debug.end(self.id)

pygpt_net/core/filesystem/__init__.py

@@ -6,13 +6,12 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.23 21:00:00                  #
+# Updated Date: 2024.12.07 21:00:00                  #
 # ================================================== #
 
 import os
 import shutil
 
-from datetime import datetime
 from pathlib import PurePath
 from uuid import uuid4
 
@@ -38,18 +37,6 @@ class Filesystem:
         self.types = Types(window)
         self.url = Url(window)
         self.workdir_placeholder = "%workdir%"
-        self.styles = [
-            "style.css",
-            "style.dark.css",
-            "style.light.css",
-            "markdown.css",
-            "markdown.dark.css",
-            "markdown.light.css",
-            "web.css",
-            "web.dark.css",
-            "web.light.css",
-            "fix_windows.css",
-        ]
 
     def install(self):
         """Install provider data"""
@@ -77,9 +64,9 @@ class Filesystem:
 
         src_dir = os.path.join(self.window.core.config.get_app_path(), 'data', 'css')
         dst_dir = os.path.join(self.window.core.config.path, 'css')
-
+        app_styles = os.listdir(src_dir)
         try:
-            for style in self.styles:
+            for style in app_styles:
                 src = os.path.join(src_dir, style)
                 dst = os.path.join(dst_dir, style)
                 if (not os.path.exists(dst) or force) and os.path.exists(src):
@@ -91,7 +78,8 @@ class Filesystem:
         """Backup user custom css styles"""
         css_dir = os.path.join(self.window.core.config.path, 'css')
         backup_file_extension = '.backup'
-        for style in self.styles:
+        user_styles = os.listdir(css_dir)
+        for style in user_styles:
             src = os.path.join(css_dir, style)
             dst = os.path.join(css_dir, style + backup_file_extension)
             if os.path.exists(src):
@@ -457,5 +445,3 @@ class Filesystem:
         else:
             files = [os.path.join(path, f) for f in os.listdir(path) if os.path.isfile(os.path.join(path, f))]
         return files
-
-

pygpt_net/core/filesystem/url.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.08 06:00:00                  #
+# Updated Date: 2024.12.09 03:00:00                  #
 # ================================================== #
 
 from PySide6.QtCore import QUrl
@@ -38,14 +38,18 @@ class Url:
         ]
 
         # JS bridge
-        if url.toString() == 'bridge://open_find':
-            pid = self.window.controller.ui.tabs.get_current_pid()
+        if url.toString().startswith('bridge://open_find'):
+            pid = int(url.toString().split(':')[2])
             if pid in self.window.ui.nodes['output']:
                 self.window.ui.nodes['output'][pid].find_open()
             return
         elif url.toString() == 'bridge://escape':
             self.window.controller.access.on_escape()
             return
+        elif url.toString() == 'bridge://focus':
+            pid = self.window.controller.ui.tabs.get_current_pid()
+            if pid in self.window.ui.nodes['output']:
+                self.window.ui.nodes['output'][pid].on_focus_js()
 
         # -------------
 

pygpt_net/core/render/base.py

@@ -6,9 +6,10 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.07 23:00:00                  #
+# Updated Date: 2024.12.09 00:00:00                  #
 # ================================================== #
 
+from pygpt_net.core.tabs import Tab
 from pygpt_net.item.ctx import CtxItem, CtxMeta
 
 
@@ -20,6 +21,15 @@ class BaseRenderer:
         :param window: Window instance
         """
         self.window = window
+        self.tab = None
+
+    def set_tab(self, tab: Tab):
+        """
+        Append tab
+
+        :param tab: Tab
+        """
+        self.tab = tab
 
     def prepare(self):
         """
@@ -218,13 +228,14 @@ class BaseRenderer:
         """
         pass
 
-    def on_page_loaded(self, meta: CtxMeta = None):
+    def on_page_loaded(self, meta: CtxMeta = None, tab: Tab = None):
         """
         On page loaded callback
 
         :param meta: context meta
+        :param tab: Tab
         """
-        pass
+        self.tab = tab
 
     def on_enable_edit(self, live: bool = True):
         """

pygpt_net/core/render/markdown/renderer.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.11.05 23:00:00                  #
+# Updated Date: 2024.12.09 00:00:00                  #
 # ================================================== #
 
 import re
@@ -50,6 +50,8 @@ class Renderer(BaseRenderer):
         
         :param meta: context PID
         """
+        if self.tab is not None:
+            return self.tab.pid  # get PID from tab if exists
         return self.window.core.ctx.output.get_pid(meta)
 
     def get_or_create_pid(self, meta: CtxMeta):