Update app.py

app.py

@@ -38,8 +38,7 @@ def inference(prompt, guidance, steps, image_size="Square", seed=0, img=None, st
 
   generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
 
-  if(not disable_auto_prompt_correction):
-    prompt,neg_prompt=auto_prompt_correction(prompt,neg_prompt,cool_japan_type)
+  prompt,neg_prompt=auto_prompt_correction(prompt,neg_prompt,cool_japan_type,disable_auto_prompt_correction)
 
   if(image_size=="Portrait"):
       height=768
@@ -60,7 +59,7 @@ def inference(prompt, guidance, steps, image_size="Square", seed=0, img=None, st
       return txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator), None
   except Exception as e:
     return None, error_str(e)
-def auto_prompt_correction(prompt_ui,neg_prompt_ui,cool_japan_type_ui):
+def auto_prompt_correction(prompt_ui,neg_prompt_ui,cool_japan_type_ui,disable_auto_prompt_correction):
     # auto prompt correction
     cool_japan_type=str(cool_japan_type_ui)
     if(cool_japan_type=="Manga"):
@@ -74,6 +73,10 @@ def auto_prompt_correction(prompt_ui,neg_prompt_ui,cool_japan_type_ui):
     neg_prompt=str(neg_prompt_ui)
     prompt=prompt.lower()
     neg_prompt=neg_prompt.lower()
+    if(disable_auto_prompt_correction):
+        prompt=f"{cool_japan_type}, {prompt}"
+        return prompt, neg_prompt
+
     if(prompt=="" and neg_prompt==""):
         #prefix=["masterpiece","evangelion, mika pikazo", "konosuba, mika pikazo","steins; gate, ilya kuvshinov",
         #        "ghibli, shinkai makoto", "evangelion, madoka magica"]