Upload 4 files

app.py

@@ -437,6 +437,7 @@ from modutils import (
     preset_quality,
     preset_sampler_setting,
     set_quick_presets,
+    insert_model_recom_prompt,
 )
 def description_ui():
     gr.Markdown(
@@ -610,6 +611,9 @@ class GuiSD:
         vae_msg = f"VAE: {vae_model}" if vae_model else ""
         msg_lora = []
 
+## BEGIN MOD
+        prompt, neg_prompt = insert_model_recom_prompt(prompt, neg_prompt, model_name)
+## END MOD
         
         if model_name in model_list:
             model_is_xl = "xl" in model_name.lower()

modutils.py

@@ -78,7 +78,7 @@ def download_private_repo(repo_id, dir_path, is_replace):
     if not hf_read_token: return
     try:
         snapshot_download(repo_id=repo_id, local_dir=dir_path, allow_patterns=['*.ckpt', '*.pt', '*.pth', '*.safetensors', '*.bin'], use_auth_token=hf_read_token)
-    except (EnvironmentError, OSError, ValueError, HTTPError, Timeout) as e:
+    except Exception as e:
         return
     else:
         if is_replace:
@@ -99,7 +99,7 @@ def get_private_model_list(repo_id, dir_path):
     if not hf_read_token: return []
     try:
         files = api.list_repo_files(repo_id, token=hf_read_token)
-    except (EnvironmentError, OSError, ValueError, HTTPError, Timeout) as e:
+    except Exception as e:
         return []
     else:
         model_list = []
@@ -135,7 +135,7 @@ def download_private_file(repo_id, path, is_replace):
     dirname = file.parent.name
     try:
         hf_hub_download(repo_id=repo_id, filename=filename, local_dir=dirname, use_auth_token=hf_read_token)
-    except (EnvironmentError, OSError, ValueError, HTTPError, Timeout) as e:
+    except Exception as e:
         return
     else:
         if is_replace:
@@ -154,7 +154,7 @@ def get_model_id_list():
     try:
         models_vp = api.list_models(author="votepurchase", cardData=True, sort="likes")
         models_john = api.list_models(author="John6666", cardData=True, sort="last_modified")
-    except (EnvironmentError, OSError, ValueError, HTTPError, Timeout) as e:
+    except Exception as e:
         return model_ids
     else:
         for model in models_vp:
@@ -203,7 +203,7 @@ def get_tupled_model_list(model_list):
         try:
             if not api.repo_exists(repo_id): continue
             model = api.model_info(repo_id=repo_id)
-        except (EnvironmentError, OSError, ValueError, HTTPError, Timeout) as e:
+        except Exception as e:
             continue
         else:
             if model.private or model.gated: continue
@@ -859,7 +859,7 @@ def get_model_pipeline(repo_id: str):
     try:
         if " " in repo_id or not api.repo_exists(repo_id): return default
         model = api.model_info(repo_id=repo_id)
-    except (EnvironmentError, OSError, ValueError, HTTPError, Timeout) as e:
+    except Exception as e:
         return default
     else:
         if model.private or model.gated: return default
@@ -871,3 +871,47 @@ def get_model_pipeline(repo_id: str):
             return "StableDiffusionPipeline"
         else:
             return default
+
+
+def load_model_prompt_dict():
+    import json
+    dict = {}
+    with open('model_dict.json', encoding='utf-8') as f:
+        dict = json.load(f)
+    return dict
+
+
+model_prompt_dict = load_model_prompt_dict()
+
+
+def insert_model_recom_prompt(prompt: str = "", neg_prompt: str = "", model_name: str = "None"):
+    def to_list(s):
+        return [x.strip() for x in s.split(",") if not s == ""]
+    
+    def list_sub(a, b):
+        return [e for e in a if e not in b]
+    
+    def list_uniq(l):
+        return sorted(set(l), key=l.index)
+
+    if not model_name or not model_name in model_prompt_dict.keys(): return prompt, neg_prompt
+    animagine_ps = to_list("anime artwork, anime style, key visual, vibrant, studio anime, highly detailed, masterpiece, best quality, very aesthetic, absurdres")
+    animagine_nps = to_list("lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract]")
+    pony_ps = to_list("source_anime, score_9, score_8_up, score_7_up, masterpiece, best quality, very aesthetic, absurdres")
+    pony_nps = to_list("source_pony, source_furry, source_cartoon, score_6, score_5, score_4, busty, ugly face, mutated hands, low res, blurry face, black and white, the simpsons, overwatch, apex legends")
+    other_ps = to_list("anime artwork, anime style, key visual, vibrant, studio anime, highly detailed, cinematic photo, 35mm photograph, film, bokeh, professional, 4k, highly detailed")
+    other_nps = to_list("photo, deformed, black and white, realism, disfigured, low contrast, drawing, painting, crayon, sketch, graphite, impressionist, noisy, blurry, soft, deformed, ugly")
+    prompts = to_list(prompt)
+    neg_prompts = to_list(neg_prompt)
+    prompts = list_sub(prompts, animagine_ps + pony_ps + other_ps)
+    neg_prompts = list_sub(neg_prompts, animagine_nps + pony_nps + other_nps)
+    last_empty_p = [""] if not prompts and type != "None" else []
+    last_empty_np = [""] if not neg_prompts and type != "None" else []
+    ps = to_list(model_prompt_dict[model_name]["prompt"])
+    nps = to_list(model_prompt_dict[model_name]["negative_prompt"])
+    prompts = prompts + ps
+    neg_prompts = neg_prompts + nps
+    prompt = ", ".join(list_uniq(prompts) + last_empty_p)
+    neg_prompt = ", ".join(list_uniq(neg_prompts) + last_empty_np)
+    return prompt, neg_prompt
+