Upload 3 files

.gitattributes

@@ -125,3 +125,4 @@ Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0211_testv3.exe filte
 Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0214_testv1.exe filter=lfs diff=lfs merge=lfs -text
 Danbooru[[:space:]]Prompt[[:space:]]Selector/NAIA_0214_testv2.exe filter=lfs diff=lfs merge=lfs -text
 Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0216_testv1.exe filter=lfs diff=lfs merge=lfs -text
+Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0217_testv1.exe filter=lfs diff=lfs merge=lfs -text

Danbooru Prompt Selector/TEST2024/NAIA_0217_testv1.exe

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

Danbooru Prompt Selector/TEST2024/NAIA_generation.py

@@ -283,7 +283,11 @@ def generate(gen_request):
         "noise_schedule": "native"
         }
     
-
+    if gen_request["type"]=="inpaint":
+        if "mask" in gen_request:
+            params["mask"] = gen_request["mask"]
+            params['add_original_image'] = gen_request['add_original_image']
+    
     positive = gen_request["prompt"]
     keywords = [key.strip() for key in positive.split(',')]
 
@@ -313,11 +317,12 @@ def generate(gen_request):
     request_type = "generate"
     if "image" in gen_request:
         params["image"] = gen_request["image"]
-        params["strength"] = gen_request["strength"]
-        params["noise"] = gen_request["noise"]
+        if "strength" in gen_request:
+            params["strength"] = gen_request["strength"]
+            params["noise"] = gen_request["noise"]
         params["sm"] = False
         params["sm_dyn"] = False
-        request_type = "img2img"
+        request_type = "img2img" if "mask" not in gen_request else "infill"
 
     temp_del = []
     for key in keywords:
@@ -357,7 +362,7 @@ def generate(gen_request):
             file.write(error_message)
 
     try:
-        zipped_bytes = generate_image(access_token, positive, "nai-diffusion-3", request_type, params)
+        zipped_bytes = generate_image(access_token, positive, "nai-diffusion-3" if "mask" not in params else "nai-diffusion-3-inpainting", request_type, params)
         if gen_request["png_rule"] == "count":
             additional_folder = "/" + gen_request["start_time"]
         if gen_request["type"] == "turbo":