[UI] minor bugfix (missing default category tag) and refactoring in AssetBrowser

Author: xushengj

src/preppipe_gui_pyside6/toolwidgets/assetbrowser.py

@@ -127,32 +127,14 @@ def load_tags(self):
     asset_manager = AssetManager.get_instance()
     self.assets_by_tag.clear()
 
-    background_tag = self.tag_manager.get_tr_background()
-    character_tag = self.tag_manager.get_tr_character()
-    other_tag = self.tag_manager.get_tr_other()
-
     # 初始化基于资产类型的分类标签
     for asset_id in self.all_asset_ids:
-      try:
-        asset = asset_manager.get_asset(asset_id)
-        if isinstance(asset, ImagePack):
-          descriptor = ImagePack.get_descriptor_by_id(asset_id)
-          if descriptor:
-            pack_type = descriptor.get_image_pack_type()
-            if pack_type == ImagePackDescriptor.ImagePackType.BACKGROUND:
-              category_tag = background_tag
-            elif pack_type == ImagePackDescriptor.ImagePackType.CHARACTER:
-              category_tag = character_tag
-            else:
-              category_tag = other_tag
-
-            # 只添加资产到对应分类，不修改tags_dict
-            if category_tag not in self.assets_by_tag:
-              self.assets_by_tag[category_tag] = (None, [asset_id])
-              continue
-            self.assets_by_tag[category_tag][1].append(asset_id)
-      except Exception:
-        continue
+      category_tag = str(self.tag_manager.get_asset_type_tag(asset_id).translatable)
+      # 只添加资产到对应分类，不修改tags_dict
+      if category_tag not in self.assets_by_tag:
+        self.assets_by_tag[category_tag] = (None, [asset_id])
+      else:
+        self.assets_by_tag[category_tag][1].append(asset_id)
 
     # 基于现有标签构建标签-资产对应关系
     for asset_id, tags in tags_dict.items():

src/preppipe_gui_pyside6/util/assettagmanager.py

@@ -5,6 +5,8 @@
 from typing import Optional, Any
 from PySide6.QtCore import Signal, QObject
 from ..settingsdict import SettingsDict
+from preppipe.assets.assetmanager import AssetManager
+from preppipe.util.imagepack import *
 from preppipe.language import TranslationDomain, Translatable
 
 TR_gui_util_assettagmanager = TranslationDomain("gui_util_assettagmanager")
@@ -186,7 +188,22 @@ def get_recent_tags_display(self) -> list[str]:
     return display_tags
 
   def get_asset_tags(self, asset_id: str) -> set[str]:
-    return self.get_tags_dict().get(asset_id, set())
+    result = set()
+    result.update(self.get_tags_dict().get(asset_id, set()))
+    result.add(self.get_asset_type_tag(asset_id).semantic)
+    return result
+
+  def get_asset_type_tag(self, asset_id: str) -> AssetTagType:
+    asset = AssetManager.get_instance().get_asset(asset_id)
+    if isinstance(asset, ImagePack):
+      descriptor = ImagePack.get_descriptor_by_id(asset_id)
+      if descriptor:
+        pack_type = descriptor.get_image_pack_type()
+        if pack_type == ImagePackDescriptor.ImagePackType.BACKGROUND:
+          return AssetTagType.BACKGROUND
+        elif pack_type == ImagePackDescriptor.ImagePackType.CHARACTER:
+          return AssetTagType.CHARACTER_SPRITE
+    return AssetTagType.OTHER
 
   def get_all_tags(self) -> set[str]:
     return {