Upload 3 files

.gitattributes

@@ -131,3 +131,4 @@ Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0217_testv2.exe filte
 Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0218_testv1.exe filter=lfs diff=lfs merge=lfs -text
 Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0218_testv2.exe filter=lfs diff=lfs merge=lfs -text
 Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0220_testv1.exe filter=lfs diff=lfs merge=lfs -text
+Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0222_testv2.exe filter=lfs diff=lfs merge=lfs -text

Danbooru Prompt Selector/TEST2024/NAIA_0222_testv2.exe

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9baaef49ccf109a0801e668887076115b221036392d30f2af5d8bb210e43f6c7
+size 888974587

Danbooru Prompt Selector/TEST2024/NAIA_generation.py

@@ -4,10 +4,15 @@ from PIL import Image, ImageOps, ImageTk
 from datetime import datetime
 import time
 from pathlib import Path
-import io, re
+import io, re, base64
 import zipfile
 
-BASE_URL="https://api.novelai.net"
+BASE_URL="https://image.novelai.net"
+
+def image_to_base64(image):
+    image_bytesIO = io.BytesIO()
+    image.save(image_bytesIO, format="PNG")
+    return base64.b64encode(image_bytesIO.getvalue()).decode()
 
 def make_turbo_prompt(gen_request):
     lines = gen_request['prompt']
@@ -264,18 +269,18 @@ def generate(gen_request):
     params = {
         "legacy": False,
         "quality_toggle": True if gen_request["quality_toggle"] == 1 else False,
-        "width": gen_request["width"],
-        "height": gen_request["height"],
+        "width": int(gen_request["width"]),
+        "height": int(gen_request["height"]),
         "n_samples": 1,
-        "seed": gen_request["seed"],
+        "seed": int(gen_request["seed"]),
         "extra_noise_seed": random.randint(0,9999999999),
         "sampler": gen_request["sampler"],
         "steps": 28 if gen_request["type"]!="upper" else gen_request["steps"],
         "scale": gen_request["scale"],
         "uncond_scale": gen_request["uncond_scale"],
         "negative_prompt": ', '.join([keyword.strip() for keyword in gen_request["negative"].split(',') if not keyword.strip().startswith('#')]),
-        "sm" : gen_request["sema"],
-        "sm_dyn" : gen_request["sema_dyn"],
+        "sm" : True if gen_request["sema"] == 1 else False,
+        "sm_dyn" : True if gen_request["sema_dyn"] == 1 else False,
         "decrisper": False,
         "controlnet_strength": 1.0,
         "add_original_image": False,
@@ -283,6 +288,11 @@ def generate(gen_request):
         "noise_schedule": "native"
         }
     
+    if "image" not in gen_request and "reference_image" in gen_request:
+        params["reference_image"] = gen_request["reference_image"]
+        params["reference_information_extracted"] = gen_request["reference_information_extracted"]
+        params["reference_strength"] = gen_request["reference_strength"]
+    
     if gen_request["type"]=="inpaint":
         if "mask" in gen_request:
             params["mask"] = gen_request["mask"]