dev(narugo): add civitai

dev(narugo): add caption for skins

app.py

@@ -9,8 +9,11 @@ from gchar.games.dispatch.access import get_character
 from gchar.resources.pixiv import get_pixiv_keywords, get_pixiv_posts
 from gchar.resources.sites import list_available_sites, get_site_tag
 from huggingface_hub import hf_hub_url
+from pycivitai import civitai_find_online
+from pycivitai.client import ModelNotFound
 
 from character import get_ch_name
+from civitai import try_find_title
 from huggingface import get_hf_fs
 
 hf_fs = get_hf_fs()
@@ -47,11 +50,13 @@ def query(chr_name):
     if hf_fs.exists(meta_json):
         meta = json.loads(hf_fs.read_text(meta_json))
         for item in meta['files']:
-            skin_urls.append(hf_hub_url(
+            skin_url = hf_hub_url(
                 'deepghs/game_character_skins',
                 filename=f'{ch.__game_name__}/{ch.index}/{item["name"]}',
                 repo_type='dataset',
-            ))
+            )
+            skin_name = item['metadata']['name']
+            skin_urls.append((skin_url, skin_name))
 
     # get repo info
     repo = f'CyberHarem/{get_ch_name(ch)}'
@@ -69,6 +74,16 @@ def query(chr_name):
         else:
             print('Dataset not found.', file=sf)
         print(file=sf)
+
+        try:
+            model_name = try_find_title(str(ch.enname), ch.__game_name__)
+            resource = civitai_find_online(model_name)
+            civit_url = f'https://civitai.com/models/{resource.model_id}'
+            print(f'CivitAI Model: [{civit_url}]({civit_url})', file=sf)
+        except ModelNotFound:
+            print('No CivitAI published model found.', file=sf)
+        print(file=sf)
+
         html = markdown.markdown(sf.getvalue())
 
     # get tags on all sites

civitai.py

@@ -0,0 +1,26 @@
+from gchar.games.dispatch.access import GAME_CHARS
+
+
+def try_find_title(char_name, game_name):
+    try:
+        game_cls = GAME_CHARS[game_name.lower()]
+        ch = game_cls.get(char_name)
+        if ch:
+            names = []
+            if ch.enname:
+                names.append(str(ch.enname))
+            if ch.jpname:
+                names.append(str(ch.jpname))
+            if ch.cnname:
+                names.append(str(ch.cnname))
+            if hasattr(ch, 'krname') and ch.krname:
+                names.append(str(ch.krname))
+
+            return f"{'/'.join(names)} ({game_cls.__official_name__})"
+
+        else:
+            cname = ' '.join(list(map(str.capitalize, char_name.split(' '))))
+            return f'{cname} ({game_cls.__official_name__})'
+
+    except KeyError:
+        return None