pygpt-net 2.4.42__py3-none-any.whl → 2.4.45__py3-none-any.whl

pygpt_net/provider/loaders/web_microsoft_onedrive.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.17 01:00:00                  #
+# Updated Date: 2024.12.16 01:00:00                  #
 # ================================================== #
 
 import json
@@ -30,18 +30,26 @@ class Loader(BaseLoader):
                     "args": {
                         "folder_id": {
                             "type": "str",
+                            "label": "Folder ID",
                         },
                         "file_ids": {
                             "type": "list",
+                            "label": "File IDs",
+                            "description": "List of file ids, separated by comma (,)",
                         },
                         "mime_types": {
                             "type": "list",
+                            "label": "Mime Types",
+                            "description": "List of mime types, separated by comma (,)",
                         },
                         "folder_path": {
                             "type": "str",
+                            "label": "Folder Path",
                         },
                         "file_paths": {
                             "type": "list",
+                            "label": "File Paths",
+                            "description": "List of file paths, separated by comma (,)",
                         },
                     },
                 }
@@ -98,10 +106,14 @@ class Loader(BaseLoader):
         if "file_ids" in kwargs and kwargs.get("file_ids"):
             if isinstance(kwargs.get("file_ids"), list):
                 args["file_ids"] = kwargs.get("file_ids")  # list of file ids
+            elif isinstance(kwargs.get("file_ids"), str):
+                args["file_ids"] = self.explode(kwargs.get("file_ids"))
 
         if "mime_types" in kwargs and kwargs.get("mime_types"):
             if isinstance(kwargs.get("mime_types"), list):
                 args["mime_types"] = kwargs.get("mime_types")  # list of mime types
+            elif isinstance(kwargs.get("mime_types"), str):
+                args["mime_types"] = self.explode(kwargs.get("mime_types"))
 
         if "folder_path" in kwargs and kwargs.get("folder_path"):
             if isinstance(kwargs.get("folder_path"), str):
@@ -110,4 +122,6 @@ class Loader(BaseLoader):
         if "file_paths" in kwargs and kwargs.get("file_paths"):
             if isinstance(kwargs.get("file_paths"), list):
                 args["file_paths"] = kwargs.get("file_paths")  # list of file paths
+            elif isinstance(kwargs.get("file_paths"), str):
+                args["file_paths"] = self.explode(kwargs.get("file_paths"))
         return args

pygpt_net/provider/loaders/web_page.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.03.12 06:00:00                  #
+# Updated Date: 2024.12.16 01:00:00                  #
 # ================================================== #
 
 from llama_index.core.readers.base import BaseReader
@@ -19,7 +19,7 @@ class Loader(BaseLoader):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.id = "webpage"
-        self.name = "Webpages"
+        self.name = "Webpage"
         self.type = ["web"]
         self.instructions = [
             {
@@ -28,6 +28,8 @@ class Loader(BaseLoader):
                     "args": {
                         "url": {
                             "type": "str",
+                            "label": "URL",
+                            "description": "URL of the webpage to index, e.g. https://www.example.com",
                         },
                     },
                 }

pygpt_net/provider/loaders/web_rss.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.03.12 06:00:00                  #
+# Updated Date: 2024.12.16 01:00:00                  #
 # ================================================== #
 
 from llama_index.core.readers.base import BaseReader
@@ -28,6 +28,8 @@ class Loader(BaseLoader):
                     "args": {
                         "url": {
                             "type": "str",
+                            "label": "URL",
+                            "description": "URL to RSS feed, e.g. https://example.com/feed.xml",
                         },
                     },
                 }

pygpt_net/provider/loaders/web_sitemap.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.17 01:00:00                  #
+# Updated Date: 2024.12.16 01:00:00                  #
 # ================================================== #
 
 from llama_index.core.readers.base import BaseReader
@@ -23,11 +23,13 @@ class Loader(BaseLoader):
         self.type = ["web"]
         self.instructions = [
             {
-                "rss": {
-                    "description": "read sitemap XML from URL",
+                "sitemap": {
+                    "description": "read all web pages from sitemap.xml",
                     "args": {
                         "url": {
                             "type": "str",
+                            "label": "URL",
+                            "description": "URL to sitemap XML, e.g. https://example.com/sitemap.xml, all pages will be read",
                         },
                     },
                 }
@@ -41,6 +43,10 @@ class Loader(BaseLoader):
             "html_to_text": "bool",
             "limit": "int",
         }
+        self.init_args_desc = {
+            "html_to_text": "Whether to convert HTML to text",
+            "limit": "Maximum number of concurrent requests",
+        }
 
     def get(self) -> BaseReader:
         """

pygpt_net/provider/loaders/web_twitter.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.04.17 01:00:00                  #
+# Updated Date: 2024.12.16 01:00:00                  #
 # ================================================== #
 
 import json
@@ -30,9 +30,11 @@ class Loader(BaseLoader):
                     "args": {
                         "users": {
                             "type": "list",
+                            "label": "Twitter/X usernames",
                         },
                         "max_tweets": {
                             "type": "int",
+                            "label": "Max tweets",
                         },
                     },
                 }
@@ -80,7 +82,7 @@ class Loader(BaseLoader):
             if isinstance(kwargs.get("users"), list):
                 args["twitterhandles"] = kwargs.get("users")  # usernames
             elif isinstance(kwargs.get("users"), str):
-                args["twitterhandles"] = [kwargs.get("users")]
+                args["twitterhandles"] = self.explode(kwargs.get("users"))
 
         if "max_tweets" in kwargs and kwargs.get("max_tweets"):
             if isinstance(kwargs.get("max_tweets"), int):

pygpt_net/provider/loaders/web_yt.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.16 01:00:00                  #
 # ================================================== #
 
 from llama_index.core.readers.base import BaseReader
@@ -28,11 +28,25 @@ class Loader(BaseLoader):
                     "args": {
                         "url": {
                             "type": "str",
+                            "label": "Youtube URL",
+                            "description": "URL of the YouTube video, e.g. https://www.youtube.com/watch?v=CRRlbK5w8AE",
                         },
                     },
                 }
             }
         ]
+        self.init_args = {
+            "languages": ["en"],
+        }
+        self.init_args_types = {
+            "languages": "list",
+        }
+        self.init_args_labels = {
+            "languages": "Languages",
+        }
+        self.init_args_desc = {
+            "languages": "List of languages to extract from the video, separated by comma (,), e.g. 'en,de,fr'. Default is 'en'",
+        }
 
     def get(self) -> BaseReader:
         """
@@ -40,7 +54,8 @@ class Loader(BaseLoader):
 
         :return: Data reader instance
         """
-        return YoutubeTranscriptReader()
+        args = self.get_args()
+        return YoutubeTranscriptReader(**args)
 
     def prepare_args(self, **kwargs) -> dict:
         """

pygpt_net/provider/vector_stores/ctx_attachment.py

@@ -50,7 +50,7 @@ class CtxAttachmentProvider(BaseStore):
 
         :return: True if exists
         """
-        path = self.get_path()
+        path = self.get_path("")
         if os.path.exists(path):
             store = os.path.join(path, "docstore.json")
             if os.path.exists(store):

pygpt_net/tools/indexer/__init__.py

@@ -6,7 +6,7 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.12.14 22:00:00                  #
+# Updated Date: 2024.12.16 01:00:00                  #
 # ================================================== #
 
 import datetime
@@ -311,48 +311,16 @@ class IndexerTool(BaseTool):
 
         :param force: force indexing
         """
-        input_params = {}
-        input_config = {}
-        is_replace = self.window.ui.nodes["tool.indexer.web.options.replace"].isChecked()
-        loader = self.window.ui.nodes["tool.indexer.web.loader"].get_value()
-        if not loader:
+        result, loader, input_params, input_config = self.window.core.idx.ui.loaders.handle_options(
+            self.window.ui.nodes["tool.indexer.web.loader"],
+            "tool.indexer.web.loader.option",
+            "tool.indexer.web.loader.config",
+        )
+        if not result:
             self.window.ui.dialogs.alert(trans("tool.indexer.alert.no_loader"))
             return
-        loaders = self.window.core.idx.indexing.get_external_instructions()
-        if loader in loaders:
-            params = loaders[loader]
-            for k in params["args"]:
-                key_path = "tool.indexer.web.loader.option." + loader + "." + k
-                if key_path in self.window.ui.nodes:
-                    input_params[k] = self.window.ui.nodes[key_path].text()
 
-        loaders = self.window.core.idx.indexing.get_external_config()
-        if loader in loaders:
-            params = loaders[loader]
-            for k in params:
-                key_path = "tool.indexer.web.loader.config." + loader + "." + k
-                type = params[k]["type"]
-                if key_path in self.window.ui.nodes:
-                    tmp_value = self.window.ui.nodes[key_path].text()
-                    try:
-                        if tmp_value:
-                            if type == "int":
-                                tmp_value = int(tmp_value)
-                            elif type == "float":
-                                tmp_value = float(tmp_value)
-                            elif type == "bool":
-                                if tmp_value.lower() in ["true", "1"]:
-                                    tmp_value = True
-                                else:
-                                    tmp_value = False
-                            elif type == "list":
-                                tmp_value = tmp_value.split(",")
-                            elif type == "dict":
-                                tmp_value = json.loads(tmp_value)
-                            input_config[k] = tmp_value
-                    except Exception as e:
-                        self.window.core.debug.log(e)
-                        self.window.ui.dialogs.alert(e)
+        is_replace = self.window.ui.nodes["tool.indexer.web.options.replace"].isChecked()
         if not force:
             self.window.ui.dialogs.confirm(
                 type="idx.tool.index",

pygpt_net/tools/indexer/ui/web.py

@@ -6,16 +6,18 @@
 # GitHub:  https://github.com/szczyglis-dev/py-gpt   #
 # MIT License                                        #
 # Created By  : Marcin Szczygliński                  #
-# Updated Date: 2024.08.19 20:00:00                  #
+# Updated Date: 2024.12.16 01:00:00                  #
 # ================================================== #
 
 import json
 
 from PySide6.QtCore import Qt
+from PySide6.QtGui import QIcon
 from PySide6.QtWidgets import QLabel, QVBoxLayout, QWidget, QCheckBox, QHBoxLayout, QScrollArea, \
     QSizePolicy
 
-from pygpt_net.ui.widget.element.labels import HelpLabel, UrlLabel
+from pygpt_net.ui.widget.element.group import QVLine, QHLine
+from pygpt_net.ui.widget.element.labels import HelpLabel, UrlLabel, IconLabel
 from pygpt_net.ui.widget.option.combo import OptionCombo
 from pygpt_net.ui.widget.option.input import OptionInput
 from pygpt_net.utils import trans
@@ -33,15 +35,17 @@ class WebTab:
         self.params_widget = None
 
     def setup(self):
-        """
-        Setup tab widget
-        """
+        """Setup tab widget"""
         # get loaders list
         loaders = self.window.controller.config.placeholder.apply_by_id("llama_index_loaders_web")
         loaders_list = []
         for loader in loaders:
-            key = list(loader.keys())[0]
-            loaders_list.append(key.replace("web_", ""))
+            k = list(loader.keys())[0]
+            key = k.replace("web_", "")
+            value = loader[k]
+            loaders_list.append({
+                key: value,
+            })
 
         self.window.ui.nodes["tool.indexer.web.loader"] = OptionCombo(
             self.window,
@@ -55,14 +59,25 @@ class WebTab:
         )
 
         self.window.ui.nodes["tool.indexer.web.loader"].layout.setContentsMargins(0, 0, 0, 0)
-        self.window.ui.nodes["tool.indexer.web.loader.label"] = QLabel(trans("tool.indexer.tab.web.loader"))
+        self.window.ui.nodes["tool.indexer.web.loader.label"] = HelpLabel(trans("tool.indexer.tab.web.loader"))
         self.window.ui.add_hook("update.tool.indexer.web.loader", self.hook_loader_change)
 
         self.window.ui.nodes["tool.indexer.web.options.label"] = HelpLabel(trans("tool.indexer.tab.web.source"))
-        self.window.ui.nodes["tool.indexer.web.config.label"] = HelpLabel(trans("tool.indexer.tab.web.cfg"))
+
+        config_label = HelpLabel(trans("tool.indexer.tab.web.cfg"))
+        config_label.setWordWrap(False)
+
+        config_label_layout = QHBoxLayout()
+        config_label_layout.addWidget(IconLabel(QIcon(":/icons/settings_filled.svg")))
+        config_label_layout.addWidget(config_label)
+        config_label_layout.setAlignment(Qt.AlignLeft)
+
+        self.window.ui.nodes["tool.indexer.web.config.label"] = QWidget()
+        self.window.ui.nodes["tool.indexer.web.config.label"].setLayout(config_label_layout)
+
         self.window.ui.nodes["tool.indexer.web.config.help"] = UrlLabel(
             trans("tool.indexer.tab.web.help"),
-            "https://pygpt.readthedocs.io/en/latest/modes.html#chat-with-files-llama-index")
+            "https://pygpt.readthedocs.io/en/latest/configuration.html#data-loaders")
 
         # --------------------------------------------------
 
@@ -78,7 +93,7 @@ class WebTab:
 
         # params
         params_layout.addWidget(self.window.ui.nodes["tool.indexer.web.options.label"])
-        inputs, groups = self.setup_loader_options()
+        inputs, groups = self.window.core.idx.ui.loaders.setup_loader_options()
         for loader in inputs:
             for k in inputs[loader]:
                 self.window.ui.nodes["tool.indexer.web.loader.option." + loader + "." + k] = inputs[loader][k]
@@ -87,9 +102,12 @@ class WebTab:
             params_layout.addWidget(self.window.ui.nodes["tool.indexer.web.loader.option_group"][loader])
             self.window.ui.nodes["tool.indexer.web.loader.option_group"][loader].hide()  # hide on start
 
+        # separator
+        params_layout.addWidget(QHLine())
+
         # config
         params_layout.addWidget(self.window.ui.nodes["tool.indexer.web.config.label"])
-        inputs, groups = self.setup_loader_config()
+        inputs, groups = self.window.core.idx.ui.loaders.setup_loader_config()
         for loader in inputs:
             for k in inputs[loader]:
                 self.window.ui.nodes["tool.indexer.web.loader.config." + loader + "." + k] = inputs[loader][k]
@@ -99,6 +117,7 @@ class WebTab:
             self.window.ui.nodes["tool.indexer.web.loader.config_group"][loader].hide()  # hide on start
         params_layout.addWidget(self.window.ui.nodes["tool.indexer.web.config.help"], alignment=Qt.AlignCenter)
 
+        # stretch
         params_layout.addStretch(1)
 
         self.params_widget = QWidget()
@@ -117,9 +136,10 @@ class WebTab:
 
         self.window.ui.nodes["tool.indexer.provider"] = HelpLabel(self.window.core.config.get("llama.idx.storage"))
 
-        loader_layout = QHBoxLayout()
+        loader_layout = QVBoxLayout()
         loader_layout.addWidget(self.window.ui.nodes["tool.indexer.web.loader.label"])
         loader_layout.addWidget(self.window.ui.nodes["tool.indexer.web.loader"])
+        loader_layout.setContentsMargins(0, 10, 0, 0)
 
         options_layout = QVBoxLayout()
         options_layout.addWidget(self.window.ui.nodes["tool.indexer.web.options.replace"])
@@ -177,70 +197,3 @@ class WebTab:
 
         self.params_widget.adjustSize()
         self.params_scroll.update()
-
-    def setup_loader_options(self):
-        """
-        Setup loader options
-        """
-        inputs = {}
-        groups = {}
-        loaders = self.window.core.idx.indexing.get_external_instructions()
-        for loader in loaders:
-            params = loaders[loader]
-            inputs[loader] = {}
-            group = QVBoxLayout()
-            for k in params["args"]:
-                widget = OptionInput(self.window, "tool.indexer", f"web.loader.{loader}.option.{k}", {
-                    "label": k,
-                    "value": "",
-                })
-                widget.setPlaceholderText(params["args"][k]["type"])
-                inputs[loader][k] = widget
-                row = QHBoxLayout()  # cols
-                row.addWidget(QLabel(k))
-                row.addWidget(widget)
-                group.addLayout(row)
-            group_widget = QWidget()
-            group_widget.setLayout(group)
-            groups[loader] = group_widget
-
-        return inputs, groups
-
-    def setup_loader_config(self):
-        """
-        Setup loader config
-        """
-        inputs = {}
-        groups = {}
-        loaders = self.window.core.idx.indexing.get_external_config()
-        for loader in loaders:
-            params = loaders[loader]
-            inputs[loader] = {}
-            group = QVBoxLayout()
-            for k in params:
-                widget = OptionInput(self.window, "tool.indexer", f"web.loader.{loader}.config.{k}", {
-                    "label": k,
-                    "value": params[k]["value"],
-                })
-                try:
-                    if params[k]["value"] is not None:
-                        if params[k]["type"] == "list" and isinstance(params[k]["value"], list):
-                            widget.setText(", ".join(params[k]["value"]))
-                        elif params[k]["type"] == "dict" and isinstance(params[k]["value"], dict):
-                            widget.setText(json.dumps(params[k]["value"]))
-                        else:
-                            widget.setText(str(params[k]["value"]))
-                except Exception as e:
-                    self.window.core.debug.log(e)
-
-                widget.setPlaceholderText(params[k]["type"])
-                inputs[loader][k] = widget
-                row = QHBoxLayout()  # cols
-                row.addWidget(QLabel(k))
-                row.addWidget(widget)
-                group.addLayout(row)
-            group_widget = QWidget()
-            group_widget.setLayout(group)
-            groups[loader] = group_widget
-
-        return inputs, groups

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: 2024.12.12 04:00:00                  #
+# Updated Date: 2024.12.16 01:00:00                  #
 # ================================================== #
 
 from PySide6 import QtCore
@@ -81,9 +81,11 @@ class CtxList:
         self.window.ui.models[id].removeRows(0, self.window.ui.models[id].rowCount())
 
         if self.window.core.config.get("ctx.records.folders.top"):
+            self.update_items_pinned(id, data)
             self.update_groups(id, data)
             self.update_items(id, data)
         else:
+            self.update_items_pinned(id, data)
             self.update_items(id, data)
             self.update_groups(id, data)
 
@@ -96,13 +98,43 @@ class CtxList:
         """
         i = 0
         last_dt_str = None
+        separators = self.window.core.config.get("ctx.records.separators")
+        pinned_separators = self.window.core.config.get("ctx.records.pinned.separators")
         for meta_id in data:
             if data[meta_id].group_id is None or data[meta_id].group_id == 0:
+                if data[meta_id].important:
+                    continue
                 item = self.build_item(meta_id, data[meta_id], is_group=False)
-                if self.window.core.config.get("ctx.records.separators"):
-                    if not item.isPinned or self.window.core.config.get("ctx.records.pinned.separators"):
+                if separators:
+                    if not item.isPinned or pinned_separators:
                         if i == 0 or last_dt_str != item.dt:
-                            section = self.build_date_section(item.dt)
+                            section = self.build_date_section(item.dt, group=False)
+                            if section:
+                                self.window.ui.models[id].appendRow(section)
+                        last_dt_str = item.dt
+                self.window.ui.models[id].appendRow(item)
+                i += 1
+
+    def update_items_pinned(self, id, data):
+        """
+        Update items pinned
+
+        :param id: ID of the list
+        :param data: Data to update
+        """
+        i = 0
+        last_dt_str = None
+        separators = self.window.core.config.get("ctx.records.separators")
+        pinned_separators = self.window.core.config.get("ctx.records.pinned.separators")
+        for meta_id in data:
+            if data[meta_id].group_id is None or data[meta_id].group_id == 0:
+                if not data[meta_id].important:
+                    continue
+                item = self.build_item(meta_id, data[meta_id], is_group=False)
+                if separators:
+                    if pinned_separators:
+                        if i == 0 or last_dt_str != item.dt:
+                            section = self.build_date_section(item.dt, group=False)
                             if section:
                                 self.window.ui.models[id].appendRow(section)
                         last_dt_str = item.dt
@@ -116,21 +148,37 @@ class CtxList:
         :param id: ID of the list
         :param data: Data to update
         """
-        # get groups
-        groups = self.window.core.ctx.get_groups()
-
+        groups = self.window.core.ctx.get_groups()  # get groups
         for group_id in groups:
             last_dt_str = None
             group = groups[group_id]
             c = self.count_in_group(group.id, data)
-            if c == 0 and self.window.core.ctx.get_search_string() is not None and self.window.core.ctx.get_search_string() != "":
+            if (c == 0 and self.window.core.ctx.get_search_string() is not None
+                    and self.window.core.ctx.get_search_string() != ""):
                 continue  # skip empty groups when searching
 
             suffix = ""
             if c > 0:
                 suffix = " (" + str(c) + ")"
+            is_attachment = group.has_additional_ctx()
             group_name = group.name + suffix
             group_item = GroupItem(QIcon(":/icons/folder_filled.svg"), group_name, group.id)
+            group_item.hasAttachments = group.has_additional_ctx()
+            custom_data = {
+                "is_group": True,
+                "is_attachment": is_attachment,
+            }
+            if is_attachment:
+                files = group.get_attachment_names()
+                num = len(files)
+                files_str = ", ".join(files)
+                if len(files_str) > 40:
+                    files_str = files_str[:40] + '...'
+                tooltip_str = trans("attachments.ctx.tooltip.list").format(num=num) + ": " + files_str
+                group_item.setToolTip(tooltip_str)
+
+            group_item.setData(custom_data, QtCore.Qt.ItemDataRole.UserRole)
+
             i = 0
             for meta_id in data:
                 if data[meta_id].group_id != group.id:
@@ -139,7 +187,7 @@ class CtxList:
                 if self.window.core.config.get("ctx.records.groups.separators"):
                     if not item.isPinned or self.window.core.config.get("ctx.records.pinned.separators"):
                         if i == 0 or last_dt_str != item.dt:
-                            section = self.build_date_section(item.dt)
+                            section = self.build_date_section(item.dt, group=True)
                             if section:
                                 group_item.appendRow(section)
                         last_dt_str = item.dt
@@ -178,6 +226,17 @@ class CtxList:
         :return: Item
         """
         append_dt = True
+        is_important = False
+        is_attachment = False
+        in_group = False
+        label = data.label
+        if data.important:
+            is_important = True
+        if data.has_additional_ctx():
+            is_attachment = True
+        if data.group:
+            in_group = True
+
         if is_group:
             if self.window.core.config.get("ctx.records.groups.separators"):
                 append_dt = False
@@ -206,35 +265,44 @@ class CtxList:
             mode_str,
             id,
         )
+
+        # append attachments to tooltip
+        if is_attachment:
+            files = data.get_attachment_names()
+            num = len(files)
+            files_str = ", ".join(files)
+            if len(files_str) > 40:
+                files_str = files_str[:40] + '...'
+            tooltip_str = trans("attachments.ctx.tooltip.list").format(num=num) + ": " + files_str
+            tooltip_text += "\n" + tooltip_str
+
         item = Item(name, id)
         item.id = id
         item.dt = dt
         item.isPinned = data.important
         item.setData(tooltip_text, QtCore.Qt.ToolTipRole)
-        is_important = False
-        is_attachment = False
-        label = data.label
-        if data.important:
-            is_important = True
-        if data.additional_ctx and len(data.additional_ctx) > 0:
-            is_attachment = True
+
         custom_data = {
             "label": label,
             "is_important": is_important,
             "is_attachment": is_attachment,
+            "in_group": in_group,
         }
         item.setData(custom_data, QtCore.Qt.ItemDataRole.UserRole)
         item.setData(name)
         return item
 
-    def build_date_section(self, dt: str) -> SectionItem:
+    def build_date_section(self, dt: str, group: bool = False) -> SectionItem:
         """
         Build date section
 
         :param dt: date section string
+        :param group: is group
         :return: SectionItem
         """
-        return SectionItem(dt)
+        item = SectionItem(dt, group=group)
+        # item.setToolTip(dt)
+        return item
 
     def convert_date(self, timestamp: int) -> str:
         """