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

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/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):

pygpt_net/core/render/plain/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 03:00:00                  #
 # ================================================== #
 
 from datetime import datetime
@@ -128,7 +128,7 @@ class Renderer(BaseRenderer):
         :param clear: True if clear all output before append
         """
         if clear:
-            self.clear_output()
+            self.clear_output(meta)
 
         i = 0
         for item in items:
@@ -399,7 +399,7 @@ class Renderer(BaseRenderer):
         :param meta: context meta
         """
         self.reset()
-        self.get_output_node().clear()
+        self.get_output_node(meta).clear()
 
     def clear_input(self):
         """Clear input"""

pygpt_net/core/render/web/body.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.12.07 21:00:00                  #
+# Updated Date: 2024.12.09 03:00:00                  #
 # ================================================== #
 
 import os
@@ -325,7 +325,7 @@ class Body:
             html += "</div>"
         return html
 
-    def get_html(self) -> str:
+    def get_html(self, pid: int) -> str:
         """
         Build webview HTML code
 
@@ -370,13 +370,14 @@ class Body:
         let scrollTimeout = null;
         let prevScroll = 0;
         let bridge;
+        let pid = """ + str(pid) + """
         new QWebChannel(qt.webChannelTransport, function (channel) {
             bridge = channel.objects.bridge;
         });
         history.scrollRestoration = "manual";
         document.addEventListener('keydown', function(event) {
             if (event.ctrlKey && event.key === 'f') {
-                window.location.href = 'bridge://open_find'; // send to bridge
+                window.location.href = 'bridge://open_find:' + pid; // send to bridge
                 event.preventDefault();
             }
             if (event.key === 'Escape') {
@@ -384,6 +385,11 @@ class Body:
                 event.preventDefault();
             }
         });
+        document.addEventListener('click', function(event) {
+            if (event.target.tagName !== 'A' && !event.target.closest('a')) {
+                window.location.href = 'bridge://focus';
+            }
+        });
         function highlightCode() {
             document.querySelectorAll('pre code').forEach(el => {
                 if (!el.classList.contains('hljs')) hljs.highlightElement(el);

pygpt_net/core/render/web/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.26 04:00:00                  #
+# Updated Date: 2024.12.09 03:00:00                  #
 # ================================================== #
 
 import json
@@ -19,6 +19,7 @@ from pygpt_net.core.text.utils import has_unclosed_code_tag
 from pygpt_net.item.ctx import CtxItem, CtxMeta
 from pygpt_net.ui.widget.textarea.input import ChatInput
 from pygpt_net.utils import trans
+from pygpt_net.core.tabs import Tab
 
 from .body import Body
 from .helpers import Helpers
@@ -59,16 +60,17 @@ class Renderer(BaseRenderer):
         node.set_meta(meta)
         self.reset(meta)
 
-    def on_page_loaded(self, meta: CtxMeta = None):
+    def on_page_loaded(self, meta: CtxMeta = None, tab: Tab = None):
         """
         On page loaded callback from WebEngine widget
 
         :param meta: context meta
+        :param tab: Tab
         """
         if meta is None:
             return
-        pid = self.get_or_create_pid(meta)
-        if pid is None:
+        pid = tab.pid
+        if pid is None or pid not in self.pids:
             return
         self.pids[pid].loaded = True
         if self.pids[pid].html != "" and not self.pids[pid].use_buffer:
@@ -818,7 +820,7 @@ class Renderer(BaseRenderer):
         if self.pids[pid].loaded:
             return  # wait for page load
 
-        html = self.body.get_html()
+        html = self.body.get_html(pid)
         self.pids[pid].document = html
         self.get_output_node_by_pid(pid).setHtml(html, baseUrl="file://")