Upload 6 files

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
-from ctag import MODELS, DEFAULT_DF, search_char_dict, on_select_df
+from ctag import (MODELS, DEFAULT_DF, DEFAULT_SERIES_DF, search_char_dict, on_select_df,
+                  search_series_dict, on_select_series_df, update_series_chars, apply_series, on_select_series_gallery)
 from hft2i import (gen_image, save_gallery, get_models, get_def_model, change_model, warm_model, get_model_info_md, get_recom_prompt_mode, update_prompt)
 
 MAX_IMAGES = 6
@@ -23,6 +24,19 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as app:
                     search_input = gr.Textbox(label="Search for characters or series:", placeholder="sousou no frieren")
                     search_detail = gr.Checkbox(label="Show character detail", value=True)
                 search_model = gr.CheckboxGroup(label="Models", choices=MODELS, value=MODELS, scale=1)
+            with gr.Accordion("Search from series", open=False):
+                with gr.Row(equal_height=True):
+                    search_series_input = gr.Textbox(label="Search for series:", placeholder="sousou no frieren")
+                    search_series_detail = gr.Checkbox(label="Show series detail", value=True)
+                with gr.Row(equal_height=True):
+                    search_series = gr.Textbox(label="Selected series", value="", interactive=False)
+                    search_chars = gr.Dropdown(label="Characters", choices=[""], value="", allow_custom_value=True)
+                    search_series_button = gr.Button("Add to search")
+                with gr.Row(equal_height=True):
+                    search_series_output = gr.Dataframe(label="Select series", value=DEFAULT_SERIES_DF, type="pandas", wrap=True, interactive=False)
+                    with gr.Column():
+                        search_series_md = gr.Markdown("<br><br><br>", elem_classes="info")
+                        search_series_gallery = gr.Gallery(label="Characters", allow_preview=False, value=[], columns=4, rows=4, elem_id="gallery", show_share_button=False, interactive=False)
         with gr.Group():
             with gr.Row(equal_height=True):
                 search_tag = gr.Textbox(label="Output tag", value="", show_copy_button=True, interactive=False)
@@ -62,6 +76,11 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as app:
           inputs=[search_input, search_model], outputs=[search_output], trigger_mode="always_last")
     search_output.select(on_select_df, [search_output, search_detail], [search_tag, search_tag_model, search_md], queue=False, show_api=False)
     search_tag.change(update_prompt, [search_tag, search_tag_model], [prompt, model_name[0], recom_prompt_mode], queue=False, show_api=False)
+    search_series_input.change(search_series_dict, [search_series_input], [search_series_output])
+    search_series_output.select(on_select_series_df, [search_series_output, search_series_detail], [search_series, search_series_md, search_series_gallery], queue=False, show_api=False)
+    search_series.change(update_series_chars, [search_series], [search_chars], queue=False, show_api=False)
+    search_series_gallery.select(on_select_series_gallery, [search_series_gallery], [search_input], queue=False, show_api=False)
+    search_series_button.click(apply_series, [search_series, search_chars], [search_input], queue=False, show_api=False)
     for i, o in enumerate(output_images):
         img_i = gr.Number(i, visible=False)
         model_name[i].change(change_model, [model_name[i]], [model_info[i]], queue=False, show_api=False)\

ctag.py

@@ -9,6 +9,7 @@ from pykakasi import kakasi
 
 MODELS = ["Animagine 3.1", "NoobAI XL", "Illustrious"]
 DEFAULT_DF = pd.DataFrame({"Character": ["oomuro sakurako (yuru yuri) (NoobAI XL)", "kafuu chino (gochuumon wa usagi desu ka?) (Animagine 3.1)"]})
+DEFAULT_SERIES_DF = pd.DataFrame({"Series": ["yuru yuri", "gochuumon wa usagi desu ka?"]})
 
 
 kks = kakasi()
@@ -71,6 +72,7 @@ def find_char_info(query: str):
         d["gender"] = i["gender"]
         d["image"] = i["character_image"]
         d["desc"] = i["desc"]
+        d["wiki"] = ""
         aid = str(i["anime_id"])
         if aid in series_wiki_dict.keys(): d["wiki"] = series_wiki_dict[aid]
         return d
@@ -79,6 +81,41 @@ def find_char_info(query: str):
         return None
 
 
+def find_series_info(query: str):
+    user_agent = get_user_agent()
+    headers = {'User-Agent': user_agent, 'content-type': 'application/json'}
+    base_url = 'https://www.animecharactersdatabase.com/api_series_characters.php'
+    params = {"anime_q": query}
+    session = requests.Session()
+    retries = Retry(total=5, backoff_factor=1, status_forcelist=[500, 502, 503, 504])
+    session.mount("https://", HTTPAdapter(max_retries=retries))
+    try:
+        r = session.get(base_url, params=params, headers=headers, stream=True, timeout=(5.0, 15))
+        if not r.ok or r.json() == -1: return None
+        j = dict(r.json())
+        if "search_results" not in j or len(j["search_results"]) == 0: return None
+        d = {}
+        i = j["search_results"][0]
+        d["name"] = i["anime_name"]
+        d["id"] = i["anime_id"]
+        d["image"] = i["anime_image"]
+        d["wiki"] = ""
+        char_url = i["characters_url"]
+        aid = str(i["anime_id"])
+        if aid in series_wiki_dict.keys(): d["wiki"] = series_wiki_dict[aid]
+        d["chars"] = {}
+        r = session.get(char_url, headers=headers, stream=True, timeout=(5.0, 15))
+        if not r.ok or r.json() == -1: return d
+        j = dict(r.json())
+        if "characters" not in j: return d
+        for c in j["characters"]:
+            d["chars"][c["name"]] = c["character_image"]
+        return d
+    except Exception as e:
+        print(e)
+        return None
+
+
 def load_json(filename: str):
     try:
         with open(filename, encoding="utf-8") as f:
@@ -97,6 +134,17 @@ def load_char_dict():
     return d
 
 
+def create_series_dict(char_dict: dict):
+    d = {}
+    for model, value in char_dict.items():
+        for name, v in value.items():
+            if not v["series"]: continue
+            d[v["series"]] = d[v["series"]] + [v["name"]] if v["series"] in d.keys() and isinstance(d[v["series"]], list) else [v["name"]]
+    for k in sorted(d.keys()):
+        d[k] = sorted(list(set(d[k])))
+    return d
+
+
 def create_char_df(char_dict: dict):
     d = {}
     m = {}
@@ -112,9 +160,55 @@ def create_char_df(char_dict: dict):
 
 
 char_dict = load_char_dict()
+series_dict = create_series_dict(char_dict)
+series_list = list(sorted(series_dict.keys()))
 char_df, tag_dict, model_dict = create_char_df(char_dict)
 
 
+def search_series_dict(q: str, progress=gr.Progress(track_tqdm=True)):
+    try:
+        MAX_COLS = 20
+        if q.strip():
+            rq = to_roman(q).lower()
+            search_results = pd.DataFrame({"Series": [s for s in series_list if rq in s]})
+        else:
+            return DEFAULT_SERIES_DF
+        if len(search_results.columns) > MAX_COLS:
+            search_results = search_results.iloc[:, :MAX_COLS]
+        return search_results
+    except Exception as e:
+        print(e)
+
+
+def on_select_series_df(df: pd.DataFrame, is_detail: bool, evt: gr.SelectData):
+    gallery = []
+    if is_detail: info = find_series_info(evt.value.split("(")[0].strip())
+    else: info = None
+    if info is not None:
+        md = f'## [{info["name"]}]({info["wiki"]})\n![{info["name"]}]({info["image"]}#center)'
+        for cap, image in info["chars"].items():
+            gallery.append((image, cap))
+    else: md = f'## {evt.value}'
+    return evt.value, md, gallery
+
+
+def on_select_series_gallery(gallery: list, evt: gr.SelectData):
+    return " ".join(list(reversed(gallery[evt.index][1].lower().split(" "))))
+
+ 
+def update_series_chars(series: str):
+    choices = list(series_dict.get(series, []))
+    return gr.update(choices=choices, value=choices[0] if len(choices) > 0 else "")
+
+
+def apply_series(series: str, char: str):
+    q = char if char else ""
+    q = q + " (" + series + ")" if series else q
+    q = q.strip()
+    if not q: return gr.update()
+    else: return q
+
+
 def search_char_dict(q: str, models: list[str], progress=gr.Progress(track_tqdm=True)):
     try:
         MAX_COLS = 50
@@ -135,7 +229,7 @@ def search_char_dict(q: str, models: list[str], progress=gr.Progress(track_tqdm=
 
 def on_select_df(df: pd.DataFrame, is_detail: bool, evt: gr.SelectData):
     d = tag_dict.get(evt.value, None)
-    if d is None: return ""
+    if d is None: return "", "", "<br><br><br>"
     #print(d) #
     if is_detail: info = find_char_info(d["name"])
     else: info = None

data/noob_e621.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8597a1068152368c4f5f59f4fab46669175e7685a7d2edce85239c36c227101c
-size 38623302
+oid sha256:932d512d646a1df0a54f588e6dd1153be3f1176b76736e4611c9a240ad3987e7
+size 38623283