narugo1992 commited on
Commit
63a4cc3
1 Parent(s): 1a50e29

dev(narugo): add pixiv and danbooru tags

Browse files
Files changed (1) hide show
  1. app.py +13 -2
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import gradio as gr
2
  from gchar.__main__ import cli
3
  from gchar.games import get_character
 
 
4
 
5
 
6
  def update_database():
@@ -14,7 +16,7 @@ def get_character_info(chtext: str):
14
 
15
  tc = type(char)
16
  fields = {name: getattr(char, name) for name in dir(tc) if isinstance(getattr(tc, name), property)}
17
- visited_names = {'raw_data', 'skins'}
18
  data = []
19
  if 'index' in fields:
20
  data.append(('id', fields['index']))
@@ -40,7 +42,16 @@ def get_character_info(chtext: str):
40
  data.append((name, value))
41
  visited_names.add(name)
42
 
43
- return {'headers': ['field', 'value'], 'data': data}, [(skin.url, skin.name) for skin in char.skins]
 
 
 
 
 
 
 
 
 
44
 
45
 
46
  _STYLE = """
 
1
  import gradio as gr
2
  from gchar.__main__ import cli
3
  from gchar.games import get_character
4
+ from gchar.resources.danbooru import get_danbooru_tag
5
+ from gchar.resources.pixiv import get_pixiv_keywords
6
 
7
 
8
  def update_database():
 
16
 
17
  tc = type(char)
18
  fields = {name: getattr(char, name) for name in dir(tc) if isinstance(getattr(tc, name), property)}
19
+ visited_names = {'raw_data', 'skins', 'names'}
20
  data = []
21
  if 'index' in fields:
22
  data.append(('id', fields['index']))
 
42
  data.append((name, value))
43
  visited_names.add(name)
44
 
45
+ pkeywords = get_pixiv_keywords(char)
46
+ if pkeywords:
47
+ data.append(('pixiv', pkeywords))
48
+ danbooru_tag = get_danbooru_tag(char)
49
+ if danbooru_tag:
50
+ data.append(('danbooru', danbooru_tag))
51
+
52
+ df = {'headers': ['field', 'value'], 'data': data}
53
+ skins = [(skin.url, skin.name) for skin in char.skins]
54
+ return df, skins
55
 
56
 
57
  _STYLE = """