update

.gitignore

@@ -1 +1,2 @@
-./outputs
+outputs
+flagged

.pylintrc

@@ -1,2 +1,4 @@
 [pylint.messages_control]
-disable=W0612,W0621,C0411,W0611,C0114,C0116,W1514,C0103,W0718,W0602
+disable=W0612,W0621,C0411,W0611,C0114,C0116,W1514,C0103,W0718,W0602,C0301
+
+# Line too long (C0301)

app.py

@@ -13,11 +13,11 @@ initial_height = 512
 
 params = {
     "enable_SD_api": True,
-    "address": "https://fb9f-2001-d08-e3-41de-e9bf-9904-76d2-5584.ngrok-free.app",
+    "address": "https://253f-2001-d08-e3-5197-8959-3944-7869-3df9.ngrok-free.app",
     "save_img": True,
-    "SD_model": "Unused",  # not actually used
-    "prompt_prefix": "4k, 8k, masterpiece, artstation, in dressed",
-    "negative_prompt": "Naked, Nude, EasyNegative, paintings, sketches, (worst quality:2), (low quality:2), (normal quality:2), lowres, normal quality, ((monochrome)), ((grayscale)), skin spots, acnes, skin blemishes, age spot, glans,extra fingers,fewer fingers,strange fingers,bad hand, paintings, sketches, (worst quality:2), (low quality:2), (normal quality:2), lowres, normal quality, ((monochrome)), ((grayscale)), skin spots, acnes, skin blemishes, age spot, (outdoor:1.6), manboobs, backlight,(ugly:1.331), (duplicate:1.331), (morbid:1.21), (mutilated:1.21), (tranny:1.331), mutated hands, (poorly drawn hands:1.331), blurry, (bad anatomy:1.21), (bad proportions:1.331), extra limbs, (disfigured:1.331), (more than 2 nipples:1.331), (missing arms:1.331), (extra legs:1.331), (fused fingers:1.61051), (too many fingers:1.61051), (unclear eyes:1.331), bad hands, missing fingers, extra digit, (futa:1.1), bad body, NG_DeepNegative_V1_75T,pubic hair, glans, paintings, sketches, (worst quality:2), (low quality:2), (normal quality:2), lowres, normal quality, ((monochrome)), ((grayscale)), skin spots, acnes, skin blemishes, age spot, (outdoor:1.6), manboobs, backlight,(ugly:1.331), (duplicate:1.331), (morbid:1.21), (mutilated:1.21), (tranny:1.331), mutated hands, (poorly drawn hands:1.331), blurry, (bad anatomy:1.21), (bad proportions:1.331), extra limbs, (disfigured:1.331), (more than 2 nipples:1.331), (missing arms:1.331), (extra legs:1.331), (fused fingers:1.61051), (too many fingers:1.61051), (unclear eyes:1.331), bad hands, missing fingers, extra digit, (futa:1.1), bad body, NG_DeepNegative_V1_75T,pubic hair, glans",
+    "SD_model": "Chilloutmix-Ni-prune-fp32-fix.safetensors",
+    "prompt_prefix": "detailed portrait of a asian man in black jacket, Futuristic sci-fi fashion, fine details, realistic shaded, fine-face, pretty face cyberpunk, neotokyo, synthwave, aesthetics, futuristic, low-emission-neon, (blade runner movie scene)",
+    "negative_prompt": "jpeg artifacts, low quality, lowres, 3d, render, doll, plastic, blur, haze, monochrome, b&w, text, (ugly:1.2), unclear eyes, no arms, bad anatomy, cropped, censoring, asymmetric eyes, bad anatomy, bad proportions, cropped, cross-eyed, deformed, extra arms, extra fingers, extra limbs, fused fingers, malformed, mangled hands, misshapen body, missing arms, missing fingers, missing hands, missing legs, poorly drawn, tentacle finger, too many arms, too many fingers, watermark, logo, text, letters, signature, username, words, blurry, cropped",
     "width": initial_width,
     "height": initial_height,
     "restore_faces": False,
@@ -41,9 +41,15 @@ def get_SD_pictures(description):
         "negative_prompt": params["negative_prompt"],
     }
 
-    response = requests.post(
-        url=f'{params["address"]}/sdapi/v1/txt2img', json=payload, timeout=10
-    )
+    try:
+        response = requests.post(
+            url=f'{params["address"]}/sdapi/v1/txt2img', json=payload, timeout=100
+        )
+        response.raise_for_status()  # Raises stored HTTPError, if one occurred
+    except requests.exceptions.HTTPError as http_err:
+        return f"HTTP error occurred: {http_err}"
+    except Exception as err:
+        return f"An error occurred: {err}"
 
     r = response.json()
 
@@ -73,21 +79,38 @@ def get_SD_pictures(description):
     return visible_result
 
 
-def display_image(description: str, kashif: bool, daniel_ho: bool) -> gr.Image:
-    if kashif:
+def display_image(description: str, artist: str) -> gr.Image:
+    if artist == "Kashif":
         description += " <lora:Kashif_v1:0.9>"
-    if daniel_ho:
+    elif artist == "Daniel Ho":
         description += " <lora:DanielHo_v1:0.9>"
-
-    if kashif or daniel_ho:
-        params["width"] = 512
-        params["height"] = 512
+    elif artist == "Adrian":
+        description += " <lora:AdrianSeowV1:0.9>"
+    elif artist == "Tony":
+        description += " <lora:TonyYap4:0.9>"
+    elif artist == "Kevin":
+        description += " <lora:KevinLowV1:0.9>"
+    elif artist == "Aaron":
+        description += " <lora:AaronSiowV1:0.9>"
+
+    if artist == "none":
+        artist = False
+
+    if artist:
+        params["width"] = 768
+        params["height"] = 768
+        params[
+            "prompt_prefix"
+        ] = "detailed portrait of a asian man in black jacket, Futuristic sci-fi fashion, fine details, realistic shaded, fine-face, pretty face cyberpunk, neotokyo, synthwave, aesthetics, futuristic, low-emission-neon, (blade runner movie scene)"
     else:
         params["width"] = initial_width
         params["height"] = initial_height
+        params["prompt_prefix"] = "4k"
 
     visible_result = get_SD_pictures(description)
-    print(description)
+    if "error occurred" in visible_result:
+        return visible_result
+
     image_data = base64.b64decode(visible_result.split(",", 1)[1])
     image = Image.open(io.BytesIO(image_data))
     return image
@@ -95,8 +118,10 @@ def display_image(description: str, kashif: bool, daniel_ho: bool) -> gr.Image:
 
 inputs = [
     gr.inputs.Textbox(lines=2, label="Enter image description"),
-    gr.inputs.Checkbox(label="Kashif"),
-    gr.inputs.Checkbox(label="Daniel Ho"),
+    gr.inputs.Radio(
+        ["Kashif", "Daniel Ho", "Adrian", "Tony", "Kevin", "Aaron", "none"],
+        label="Artist",
+    ),
 ]
 
 outputs = gr.outputs.Image(type="pil", label="Generated Image")