Upload 3 files

app.py

@@ -413,8 +413,10 @@ from tagger import (
     insert_recom_prompt,
     compose_prompt_to_copy,
     translate_prompt,
+    select_random_character,
 )
 from modutils import (
+    change_interface_mode,
     get_t2i_model_info,
     get_tupled_model_list,
     save_gallery_images,
@@ -424,6 +426,8 @@ from modutils import (
     set_lora_prompt,
     get_lora_tupled_list,
     apply_lora_prompt,
+    search_civitai_lora,
+    select_civitai_lora,
     set_textual_inversion_prompt,
     get_model_pipeline,
     set_optimization,
@@ -873,7 +877,8 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
         with gr.Tab("Generation"):
             v2b = V2UI()
             with gr.Column(scale=2):
-                with gr.Accordion("Model and Task", open=False):
+                interface_mode_gui = gr.Radio(label="Quick settings", choices=["Simple", "Standard", "Fast", "LoRA"], value="Standard")
+                with gr.Accordion("Model and Task", open=False) as menu_model:
                     task_gui = gr.Dropdown(label="Task", choices=sdxl_task, value=task_model_list[0])
                     model_name_gui = gr.Dropdown(label="Model", info="You can enter a huggingface model repo_id to want to use.", choices=get_tupled_model_list(model_list), value="votepurchase/animagine-xl-3.1", allow_custom_value=True)
                     model_info_gui = gr.Markdown()
@@ -881,12 +886,13 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
                         quick_model_type_gui = gr.Radio(label="Model Type", choices=["None", "Animagine", "Pony"], value="None", interactive=True)
                         quick_genre_gui = gr.Radio(label="Genre", choices=["Anime", "Photo"], value="Anime", interactive=True)
                         quick_speed_gui = gr.Radio(label="Speed", choices=["Fast", "Standard", "Heavy"], value="Standard", interactive=True)
-                        quick_aspect_gui = gr.Radio(label="Aspect Ratio", choices=["1:1", "3:4"], value="3:4", interactive=True)
-                    quality_selector_gui = gr.Dropdown(label="Quality Tags Presets", interactive=True, choices=list(preset_quality.keys()), value="None")
-                    style_selector_gui = gr.Dropdown(label="Style Preset", interactive=True, choices=list(preset_styles.keys()), value="None")
-                    sampler_selector_gui = gr.Dropdown(label="Sampler Quick Settings", interactive=True, choices=list(preset_sampler_setting.keys()), value="None")
-                    optimization_gui = gr.Dropdown(label="Optimization for SDXL", choices=list(optimization_list.keys()), value="None", interactive=True)
-                with gr.Accordion("Generate prompt from Image", open=False):
+                        quick_aspect_gui = gr.Radio(label="Aspect Ratio", choices=["1:1", "3:4"], value="1:1", interactive=True)
+                    with gr.Accordion(label="Advanced options", open=False):
+                        quality_selector_gui = gr.Dropdown(label="Quality Tags Presets", interactive=True, choices=list(preset_quality.keys()), value="None")
+                        style_selector_gui = gr.Dropdown(label="Style Preset", interactive=True, choices=list(preset_styles.keys()), value="None")
+                        sampler_selector_gui = gr.Dropdown(label="Sampler Quick Settings", interactive=True, choices=list(preset_sampler_setting.keys()), value="None")
+                        optimization_gui = gr.Dropdown(label="Optimization for SDXL", choices=list(optimization_list.keys()), value="None", interactive=True)
+                with gr.Accordion("Generate prompt from Image", open=False) as menu_from_image:
                     input_image_gui = gr.Image(label="Input image", type="pil", sources=["upload", "clipboard"], height=256)
                     with gr.Accordion(label="Advanced options", open=False):
                         general_threshold_gui = gr.Slider(label="Threshold", minimum=0.0, maximum=1.0, value=0.3, step=0.01, interactive=True)
@@ -898,16 +904,17 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
                     generate_from_image_btn_gui = gr.Button(value="GENERATE TAGS FROM IMAGE", size="lg", variant="primary")
                 with gr.Group():
                     prompt_gui = gr.Textbox(lines=6, placeholder="1girl, solo, ...", label="Prompt", show_copy_button=True)
-                    with gr.Accordion("Negative prompt, etc.", open=False):
+                    with gr.Accordion("Negative prompt, etc.", open=False) as menu_negative:
                         neg_prompt_gui = gr.Textbox(lines=3, placeholder="lowres, (bad), ...", label="Negative prompt", show_copy_button=True)
                         translate_prompt_button = gr.Button(value="Translate prompt to English", size="sm", variant="secondary")
-                        insert_prompt_gui = gr.Radio(label="Insert reccomended positive / negative prompt", choices=["None", "Animagine", "Pony"], value="None", interactive=True, scale=2)
-                        prompt_type_gui = gr.Radio(label="Convert tags to", choices=["danbooru", "e621"], value="e621", visible=False)
-                        prompt_type_button = gr.Button(value="Convert prompt to Pony e621 style", size="sm", variant="secondary")
+                        with gr.Accordion(label="Advanced options", open=False):
+                            insert_prompt_gui = gr.Radio(label="Insert reccomended positive / negative prompt", choices=["None", "Animagine", "Pony"], value="None", interactive=True, scale=2)
+                            prompt_type_gui = gr.Radio(label="Convert tags to", choices=["danbooru", "e621"], value="e621", visible=False)
+                            prompt_type_button = gr.Button(value="Convert prompt to Pony e621 style", size="sm", variant="secondary")
                         with gr.Row():
                             character_dbt = gr.Textbox(lines=1, placeholder="kafuu chino, ...", label="Character names")
                             series_dbt = gr.Textbox(lines=1, placeholder="Is the order a rabbit?, ...", label="Series names")
-                            generate_db_random_button = gr.Button(value="Generate random prompt from character", size="sm", variant="secondary")
+                            random_character_gui = gr.Button(value="Pick a random character 🎲", size="sm")
                             model_name_dbt = gr.Dropdown(label="Model", choices=list(V2_ALL_MODELS.keys()), value=list(V2_ALL_MODELS.keys())[0], visible=False)
                             aspect_ratio_dbt = gr.Radio(label="Aspect ratio", choices=list(V2_ASPECT_RATIO_OPTIONS), value="square", visible=False)
                             length_dbt = gr.Radio(label="Length", choices=list(V2_LENGTH_OPTIONS), value="very_long", visible=False)
@@ -915,7 +922,8 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
                             ban_tags_dbt = gr.Textbox(label="Ban tags", placeholder="alternate costumen, ...", value="futanari, censored, furry, furrification", visible=False)
                             elapsed_time_dbt = gr.Markdown(label="Elapsed time", value="", visible=False)
                             copy_button_dbt = gr.Button(value="Copy to clipboard", visible=False)
-                        rating_dbt = gr.Radio(label="Rating", choices=list(V2_RATING_OPTIONS), value="explicit")
+                        generate_db_random_button = gr.Button(value="Generate random prompt 🎲", size="sm", variant="secondary")
+                        rating_dbt = gr.Radio(label="Rating", choices=list(V2_RATING_OPTIONS), value="sfw")
                         with gr.Row():
                             set_params_gui = gr.Button(value="↙️")
                             clear_prompt_gui = gr.Button(value="🗑️")
@@ -951,7 +959,7 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
 
                 actual_task_info = gr.HTML()
 
-                with gr.Accordion("Generation settings", open=False, visible=True):
+                with gr.Accordion("Generation settings", open=False, visible=True) as menu_gen:
                     steps_gui = gr.Slider(minimum=1, maximum=100, step=1, value=28, label="Steps")
                     cfg_gui = gr.Slider(minimum=0, maximum=30, step=0.5, value=7.0, label="CFG")
                     sampler_gui = gr.Dropdown(label="Sampler", choices=scheduler_names, value="Euler a")
@@ -1043,7 +1051,7 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
                     prompt_syntax_gui = gr.Dropdown(label="Prompt Syntax", choices=prompt_s_options, value=prompt_s_options[1][1])
                     vae_model_gui = gr.Dropdown(label="VAE Model", choices=vae_model_list)
 
-                with gr.Accordion("Hires fix", open=False, visible=True):
+                with gr.Accordion("Hires fix", open=False, visible=True) as menu_hires:
 
                     upscaler_keys = list(upscaler_dict_gui.keys())
 
@@ -1057,7 +1065,7 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
                     hires_prompt_gui = gr.Textbox(label="Hires Prompt", placeholder="Main prompt will be use", lines=3)
                     hires_negative_prompt_gui = gr.Textbox(label="Hires Negative Prompt", placeholder="Main negative prompt will be use", lines=3)
 
-                with gr.Accordion("LoRA", open=False, visible=True):
+                with gr.Accordion("LoRA", open=False, visible=True) as menu_lora:
                     lora1_gui = gr.Dropdown(label="Lora1", choices=get_lora_tupled_list(lora_model_list), allow_custom_value=True)
                     lora_scale_1_gui = gr.Slider(minimum=-2, maximum=2, step=0.01, value=1.00, label="Lora Scale 1")
                     with gr.Row():
@@ -1102,6 +1110,12 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
                             [text_lora],
                             [lora1_gui, lora2_gui, lora3_gui, lora4_gui, lora5_gui]
                         )
+                        with gr.Row():
+                            search_civitai_query_lora = gr.Textbox(label="Query", placeholder="oomuro sakurako...", lines=1)
+                            search_civitai_basemodel_lora = gr.CheckboxGroup(label="Search LoRA for", choices=["Pony", "SD 1.5", "SDXL 1.0"], value=["Pony", "SDXL 1.0"])
+                            search_civitai_button_lora = gr.Button("Search on Civitai")
+                        search_civitai_result_lora = gr.Dropdown(label="Search Results", choices=[("", "")], value="", allow_custom_value=True, visible=False)
+                        search_civitai_desc_lora = gr.Markdown(value="", visible=False)
 
                     with gr.Accordion("From Local", open=True, visible=True):
                         file_output_lora = gr.File(label="Uploaded LoRA", file_types=['.ckpt', '.pt', '.pth', '.safetensors', '.bin'], file_count="multiple", interactive=False, visible=False)
@@ -1111,8 +1125,8 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
                             [file_output_lora],
                             [lora1_gui, lora2_gui, lora3_gui, lora4_gui, lora5_gui]
                         )
-## END MOD
-                with gr.Accordion("IP-Adapter", open=False, visible=True):##############
+
+                with gr.Accordion("IP-Adapter", open=False, visible=True) as menu_ipa:##############
 
                     IP_MODELS = sorted(list(set(IP_ADAPTERS_SD + IP_ADAPTERS_SDXL)))
                     MODE_IP_OPTIONS = ["original", "style", "layout", "style+layout"]
@@ -1130,7 +1144,7 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
                         mode_ip2 = gr.Dropdown(value="style", label="Mode", choices=MODE_IP_OPTIONS)
                         scale_ip2 = gr.Slider(minimum=0., maximum=2., step=0.01, value=0.7, label="Scale")
 
-                with gr.Accordion("ControlNet / Img2img / Inpaint", open=False, visible=True):
+                with gr.Accordion("ControlNet / Img2img / Inpaint", open=False, visible=True) as menu_i2i:
                     image_control = gr.Image(label="Image ControlNet/Inpaint/Img2img", type="filepath")
                     image_mask_gui = gr.Image(label="Image Mask", type="filepath")
                     strength_gui = gr.Slider(
@@ -1162,12 +1176,12 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
                     control_net_start_threshold_gui = gr.Slider(minimum=0, maximum=1, step=0.01, value=0, label="ControlNet Start Threshold (%)")
                     control_net_stop_threshold_gui = gr.Slider(minimum=0, maximum=1, step=0.01, value=1, label="ControlNet Stop Threshold (%)")
 
-                with gr.Accordion("T2I adapter", open=False, visible=True):
+                with gr.Accordion("T2I adapter", open=False, visible=True) as menu_t2i:
                     t2i_adapter_preprocessor_gui = gr.Checkbox(value=True, label="T2i Adapter Preprocessor")
                     adapter_conditioning_scale_gui = gr.Slider(minimum=0, maximum=5., step=0.1, value=1, label="Adapter Conditioning Scale")
                     adapter_conditioning_factor_gui = gr.Slider(minimum=0, maximum=1., step=0.01, value=0.55, label="Adapter Conditioning Factor (%)")
 
-                with gr.Accordion("Styles", open=False, visible=True):
+                with gr.Accordion("Styles", open=False, visible=True) as menu_styles:
 
                     try:
                         style_names_found = sd_gen.model.STYLE_NAMES
@@ -1195,17 +1209,15 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
 
                     style_button.click(load_json_style_file, [style_json_gui], [style_prompt_gui])                        
 
-## BEGIN MOD
-                with gr.Accordion("Textual inversion", open=False, visible=True):
+                with gr.Accordion("Textual inversion", open=False, visible=True) as menu_ti:
                     active_textual_inversion_gui = gr.Checkbox(value=False, label="Active Textual Inversion in prompt")
                     use_textual_inversion_gui = gr.CheckboxGroup(choices=get_embed_list(get_model_pipeline(model_name_gui.value)) if active_textual_inversion_gui.value else [], value=None, label="Use Textual Invertion in prompt")
                     def update_textual_inversion_gui(active_textual_inversion_gui, model_name_gui):
                         return gr.update(choices=get_embed_list(get_model_pipeline(model_name_gui)) if active_textual_inversion_gui else [])
                     active_textual_inversion_gui.change(update_textual_inversion_gui, [active_textual_inversion_gui, model_name_gui], [use_textual_inversion_gui])
                     model_name_gui.change(update_textual_inversion_gui, [active_textual_inversion_gui, model_name_gui], [use_textual_inversion_gui])
-## END MOD
 
-                with gr.Accordion("Detailfix", open=False, visible=True):
+                with gr.Accordion("Detailfix", open=False, visible=True) as menu_detail:
 
                     # Adetailer Inpaint Only
                     adetailer_inpaint_only_gui = gr.Checkbox(label="Inpaint only", value=True)
@@ -1243,7 +1255,7 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
                         mask_blur_b_gui = gr.Number(label="Mask blur:", value=4, minimum=1)
                         mask_padding_b_gui = gr.Number(label="Mask padding:", value=32, minimum=1)
 
-                with gr.Accordion("Other settings", open=False, visible=True):
+                with gr.Accordion("Other settings", open=False, visible=True) as menu_other:
                     image_previews_gui = gr.Checkbox(value=True, label="Image Previews")
                     hires_before_adetailer_gui = gr.Checkbox(value=False, label="Hires Before Adetailer")
                     hires_after_adetailer_gui = gr.Checkbox(value=True, label="Hires After Adetailer")
@@ -1262,7 +1274,6 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
                     retain_hires_model_previous_load_gui = gr.Checkbox(value=False, label="Retain Hires Model Previous Load")
                     xformers_memory_efficient_attention_gui = gr.Checkbox(value=False, label="Xformers Memory Efficient Attention")
 
-## BEGIN MOD
                 with gr.Accordion("Examples and help", open=True, visible=True):
                     gr.Examples(
                         examples=[
@@ -1596,6 +1607,12 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
                 btn_send.click(send_img, [img_source, img_result], [image_control, image_mask_gui])
 
 ## BEGIN MOD
+        interface_mode_gui.change(
+            change_interface_mode,
+            [interface_mode_gui],
+            [menu_model, menu_from_image, menu_negative, menu_gen, menu_hires, menu_lora, menu_ipa,\
+              menu_i2i, menu_t2i, menu_styles, menu_ti, menu_detail, menu_other, task_gui, quick_speed_gui]
+        )
         model_name_gui.change(get_t2i_model_info, [model_name_gui], [model_info_gui])
 
         quick_model_type_gui.change(
@@ -1653,6 +1670,26 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
         lora3_copy_button.click(apply_lora_prompt, [prompt_gui, lora3_trigger_gui], [prompt_gui])
         lora4_copy_button.click(apply_lora_prompt, [prompt_gui, lora4_trigger_gui], [prompt_gui])
         lora5_copy_button.click(apply_lora_prompt, [prompt_gui, lora5_trigger_gui], [prompt_gui])
+        
+        search_civitai_button_lora.click(
+            search_civitai_lora,
+            [search_civitai_query_lora, search_civitai_basemodel_lora],
+            [search_civitai_result_lora, search_civitai_desc_lora],
+            queue=True,
+            show_progress="minimal",
+        )
+        search_civitai_query_lora.submit(
+            search_civitai_lora,
+            [search_civitai_query_lora, search_civitai_basemodel_lora],
+            [search_civitai_result_lora, search_civitai_desc_lora],
+            queue=True,
+            show_progress="minimal",
+        )
+        search_civitai_result_lora.change(
+            select_civitai_lora,
+            [search_civitai_result_lora],
+            [text_lora, search_civitai_desc_lora],
+        )
 
         use_textual_inversion_gui.change(set_textual_inversion_prompt, [use_textual_inversion_gui, prompt_gui, neg_prompt_gui, prompt_syntax_gui], [prompt_gui, neg_prompt_gui])
 
@@ -1699,6 +1736,8 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
             outputs=[prompt_gui],
         )
 
+        random_character_gui.click(select_random_character, [series_dbt, character_dbt], [series_dbt, character_dbt])
+
         generate_db_random_button.click(
             parse_upsampling_output(v2b.on_generate),
             inputs=[
@@ -1848,6 +1887,7 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
                 with gr.Group():
                     input_character = gr.Textbox(label="Character tags", placeholder="hatsune miku")
                     input_copyright = gr.Textbox(label="Copyright tags", placeholder="vocaloid")
+                    pick_random_character = gr.Button(value="Pick a random character 🎲", size="sm")
                     input_general = gr.TextArea(label="General tags", lines=4, placeholder="1girl, ...", value="")
                     input_tags_to_copy = gr.Textbox(value="", visible=False)
                     copy_input_btn = gr.Button(value="Copy to clipboard", size="sm", interactive=False)
@@ -1913,6 +1953,8 @@ with gr.Blocks(theme="NoCrypt/miku", elem_id="main", css=CSS) as app:
             gradio_copy_text, inputs=[input_tags_to_copy], js=COPY_ACTION_JS,
         )
 
+        pick_random_character.click(select_random_character, [input_copyright, input_character], [input_copyright, input_character])
+
         generate_btn.click(
             parse_upsampling_output(v2.on_generate),
             inputs=[

modutils.py

@@ -7,6 +7,34 @@ from PIL import Image
 HF_LORA_PRIVATE_REPOS = ['John6666/loratest1', 'John6666/loratest3', 'John6666/loratest4', 'John6666/loratest6', 'John6666/loratest10', 'John6666/loratest']
 HF_LORA_ESSENTIAL_PRIVATE_REPO = 'John6666/loratest1'
 directory_loras = 'loras'
+CIVITAI_API_KEY = os.environ.get("CIVITAI_API_KEY")
+
+
+def get_user_agent():
+    return 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:127.0) Gecko/20100101 Firefox/127.0'
+
+
+def change_interface_mode(mode: str):
+    if mode == "Fast":
+        return gr.update(open=False), gr.update(visible=True), gr.update(open=False), gr.update(open=False),\
+        gr.update(visible=True), gr.update(open=False), gr.update(visible=True), gr.update(visible=True),\
+        gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True),\
+        gr.update(visible=True), gr.update(visible=True), gr.update(value="Fast")
+    elif mode == "Simple": # t2i mode
+        return gr.update(open=True), gr.update(visible=True), gr.update(open=True), gr.update(open=True),\
+        gr.update(visible=True), gr.update(open=False), gr.update(visible=False), gr.update(visible=False),\
+        gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False),\
+        gr.update(visible=False), gr.update(visible=False), gr.update(value="Standard")
+    elif mode == "LoRA": # t2i LoRA  mode
+        return gr.update(open=True), gr.update(visible=True), gr.update(open=True), gr.update(open=False),\
+        gr.update(visible=True), gr.update(open=True), gr.update(visible=True), gr.update(visible=True),\
+        gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True),\
+        gr.update(visible=True), gr.update(visible=False), gr.update(value="Standard")
+    else: # Standard
+        return gr.update(open=False), gr.update(visible=True), gr.update(open=False), gr.update(open=False),\
+        gr.update(visible=True), gr.update(open=False), gr.update(visible=True), gr.update(visible=True),\
+        gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True),\
+        gr.update(visible=True), gr.update(visible=True), gr.update(value="Standard")
 
 
 def get_model_list(directory_path):
@@ -294,9 +322,7 @@ def get_civitai_info(path):
     if path in set(civitai_not_exists_list): return ["", "", "", "", ""]
     from pathlib import Path
     if not Path(path).exists(): return None
-    from fake_useragent import UserAgent
-    ua = UserAgent()
-    user_agent = ua.random
+    user_agent = get_user_agent()
     headers = {'User-Agent': user_agent, 'content-type': 'application/json'}
     base_url = 'https://civitai.com/api/v1/model-versions/by-hash/'
     params = {}
@@ -442,6 +468,60 @@ def move_file_lora(filepaths):
     ),
 
 
+def search_lora_on_civitai(query: str, allow_model: list[str]):
+    import requests
+    if not query: return None
+    user_agent = get_user_agent()
+    headers = {'User-Agent': user_agent, 'content-type': 'application/json'}
+    base_url = 'https://civitai.com/api/v1/models'
+    params = {'query': query, 'types': ['LORA'], 'sort': 'Highest Rated', 'period': 'AllTime',
+              'nsfw': 'true'}
+    r = requests.get(base_url, params=params, headers=headers, timeout=(3.0, 30))
+    if not r.ok: return None
+    json = r.json()
+    if not 'items' in json: return None
+    items = []
+    for j in json['items']:
+        for model in j['modelVersions']:
+            item = {}
+            if not model['baseModel'] in set(allow_model): continue
+            item['name'] = j['name']
+            item['creator'] = j['creator']['username']
+            item['tags'] = j['tags']
+            item['model_name'] = model['name']
+            item['base_model'] = model['baseModel']
+            item['dl_url'] = model['downloadUrl']
+            item['md'] = f'<img src="{model["images"][0]["url"]}" alt="thumbnail" width="150" height="240"><br>[LoRA Model URL](https://civitai.com/models/{j["id"]})'
+            items.append(item)
+    return items
+
+
+civitai_lora_last_results = {}
+
+
+def search_civitai_lora(query, base_model):
+    global civitai_lora_last_results
+    items = search_lora_on_civitai(query, base_model)
+    if not items: return gr.update(choices=[("", "")], value="", visible=False), gr.update(value="", visible=False)
+    civitai_lora_last_results = {}
+    choices = []
+    for item in items:
+        base_model_name = "Pony🐴" if item['base_model'] == "Pony" else item['base_model']
+        name = f"{item['name']} (for {base_model_name} / By {item['creator']} / Tags={', '.join(item['tags'])})"
+        value = item['dl_url']
+        choices.append((name, value))
+        civitai_lora_last_results[value] = item['md']
+    if not choices: return gr.update(choices=[("", "")], value="", visible=False), gr.update(value="", visible=False)
+    md = civitai_lora_last_results.get(choices[0][1], "None")
+    return gr.update(choices=choices, value=choices[0][1], visible=True), gr.update(value=md, visible=True)
+
+
+def select_civitai_lora(search_result):
+    if not "http" in search_result: return gr.update(value=""), gr.update(value="None", visible=True)
+    md = civitai_lora_last_results.get(search_result, "None")
+    return gr.update(value=search_result), gr.update(value=md, visible=True)
+
+
 quality_prompt_list = [
     {
         "name": "None",

tagger.py

@@ -81,6 +81,14 @@ def character_list_to_series_list(character_list):
     return output_series_tag
 
 
+def select_random_character(series: str, character: str):
+    from random import randrange
+    character_list = list(anime_series_dict.keys())
+    character = character_list[randrange(len(character_list) - 1)]
+    series = anime_series_dict.get(character.split(",")[0].strip(), "")
+    return series, character
+
+
 def danbooru_to_e621(dtag, e621_dict):
     def d_to_e(match, e621_dict):
         dtag = match.group(0)