minor2

utils.py

@@ -8,7 +8,7 @@ import math
 import gradio as gr
 import hashlib
 import time
-import PIL
+
 from PIL import Image
 
 jwt_token = ''
@@ -84,11 +84,11 @@ def generate_novelai_image(
 ):
     # Assign a random seed if seed is -1
     if seed == -1:
-        seed = random.randint(0, 2**32 - 1)
+        seed = random.randint(0, 2 ** 32 - 1)
 
     characterPrompts = []
     for i in range(len(char_prompts)):
-        characterPrompts.append({"prompt": char_prompts[i], "uc": char_ucs[i], "center": {'x': round(char_coords_x[i]*0.2 - 0.1, 1), 'y': round(char_coords_y[i]*0.2 - 0.1, 1)}})
+        characterPrompts.append({"prompt": char_prompts[i], "uc": char_ucs[i], "center": {'x': round(char_coords_x[i] * 0.2 - 0.1, 1), 'y': round(char_coords_y[i] * 0.2 - 0.1, 1)}})
     
     # Define the payload
     payload = {
@@ -130,8 +130,8 @@ def generate_novelai_image(
         payload["parameters"]["v4_prompt"] = {"caption": {"base_caption": input_text, "char_captions": []}, "use_coords": not auto_pos, "use_order": True}
         payload["parameters"]["v4_negative_prompt"] = {"caption": {"base_caption": negative_prompt, "char_captions": []}}
         for i in range(len(char_prompts)):
-            payload["parameters"]["v4_prompt"]["caption"]["char_captions"].append({"char_caption": char_prompts[i], "centers": [{'x': round(char_coords_x[i]*0.2 - 0.1, 1), 'y': round(char_coords_y[i]*0.2 - 0.1, 1)}]})
-            payload["parameters"]["v4_negative_prompt"]["caption"]["char_captions"].append({"char_caption": char_ucs[i], "centers": [{'x': round(char_coords_x[i]*0.2 - 0.1, 1), 'y': round(char_coords_y[i]*0.2 - 0.1, 1)}]})
+            payload["parameters"]["v4_prompt"]["caption"]["char_captions"].append({"char_caption": char_prompts[i], "centers": [{'x': round(char_coords_x[i] * 0.2 - 0.1, 1), 'y': round(char_coords_y[i] * 0.2 - 0.1, 1)}]})
+            payload["parameters"]["v4_negative_prompt"]["caption"]["char_captions"].append({"char_caption": char_ucs[i], "centers": [{'x': round(char_coords_x[i] * 0.2 - 0.1, 1), 'y': round(char_coords_y[i] * 0.2 - 0.1, 1)}]})
     if ref_images != None:
         payload['parameters']['reference_image_multiple'] = [image2base64(image[0]) for image in ref_images]
         payload['parameters']['reference_information_extracted_multiple'] = info_extracts
@@ -146,7 +146,7 @@ def generate_novelai_image(
         payload['action'] = "infill"
         payload['model'] = model.replace('-preview', '') + '-inpainting'
         mask = inp_img['layers'][0].resize((width, height))
-        mask = mask.resize((mask.size[0]//8, mask.size[1]//8), resample=Image.NEAREST).resize(mask.size, resample=Image.NEAREST).point(lambda x: 0 if x < 255 else 255)
+        mask = mask.resize((mask.size[0] // 8, mask.size[1] // 8), resample=Image.NEAREST).resize(mask.size, resample=Image.NEAREST).point(lambda x: 0 if x < 255 else 255)
         payload['parameters']['mask'] = image2base64(mask)
         payload['parameters']['image'] = image2base64(inp_img['background'])
         payload['parameters']['extra_noise_seed'] = seed