update app.py

app.py

@@ -615,11 +615,22 @@ def load_kolors_inpainting(inpaint_prompt, input_image, mask_image):
     from gradio_client import Client, handle_file
     import tempfile
     MAX_IMAGE_SIZE = 1024
+
+    def change_RGB_value(image, r0, g0, b0, r1, g1, b1):
+        pixels = image.load()
+        for i in range(image.size[0]):
+            for j in range(image.size[1]):
+                r, g, b = pixels[i, j]
+                if r == r0 and g == g0 and b == b0:
+                    pixels[i, j] = (r1, g1, b1)
+        return image
+
     try:
         # logger.info(f'load_kolors_inpainting_input_image={inpaint_prompt} // {input_image}')
         # logger.info(f'load_kolors_inpainting_mask_image={mask_image}')
 
         job_image = {}
+        job_mask_image = None
         if 'background' in input_image.keys():
             width, height = input_image['background'].size
             if max(width, height) > MAX_IMAGE_SIZE:
@@ -638,31 +649,22 @@ def load_kolors_inpainting(inpaint_prompt, input_image, mask_image):
             img.save(temp_file_path)
             # logger.info(f'load_kolors_inpainting_temp_file_background_={temp_file_path}')
             job_image["background"] = handle_file(temp_file_path)
-        if 'layers' in input_image.keys() and len(input_image['layers']) > 0:
-            _, temp_file_path = tempfile.mkstemp(suffix='.png')
-            img = input_image['layers'][0].convert("RGB").resize((resize_width, resize_height))
-            img.save(temp_file_path)
-            # logger.info(f'load_kolors_inpainting_temp_file_layers_={temp_file_path}')
-            job_image["layers"] = [handle_file(temp_file_path)]
 
-        if 'composite' in input_image.keys():
+        if mask_image is not None:
             _, temp_file_path = tempfile.mkstemp(suffix='.png')
-            img = input_image['composite'].convert("RGB").resize((resize_width, resize_height))
+            img = mask_image.convert("RGB").resize((resize_width, resize_height))
+            # RGB(0,0,0) -->  RGB(230,230,230)           
+            img = change_RGB_value(img, 0, 0, 0, 230, 230, 230)
+            # RGB(255,255,255) --> RGB(170,170,170)
+            img = change_RGB_value(img, 255, 255, 255, 170, 170, 170)
             img.save(temp_file_path)
-            # logger.info(f'load_kolors_inpainting_temp_file_composite_={temp_file_path}')
-            job_image["composite"] = handle_file(temp_file_path)
-
-        _, temp_file_path = tempfile.mkstemp(suffix='.png')
-        img = mask_image.convert("RGB").resize((resize_width, resize_height))
-        img.save(temp_file_path)
-        # logger.info(f'load_kolors_inpainting_temp_file_mask_={temp_file_path}')
-        job_mask_image = handle_file(temp_file_path)
+            # logger.info(f'load_kolors_inpainting_temp_file___mask_={temp_file_path}')
+            job_image["layers"] = [handle_file(temp_file_path)]            
 
         # logger.info(f'load_kolors_inpainting_job_image={job_image}')
         # logger.info(f'load_kolors_inpainting_job_mask_image={job_mask_image}')
 
-        client = Client("Kwai-Kolors/Kolors-Inpainting")
-        
+        client = Client("Kwai-Kolors/Kolors-Inpainting")        
         job = client.submit(
                 prompt=inpaint_prompt,
                 image=job_image,
@@ -670,7 +672,7 @@ def load_kolors_inpainting(inpaint_prompt, input_image, mask_image):
                 negative_prompt="broken fingers, deformed fingers, deformed hands, stumps, blurriness, low quality",
                 seed=0,
                 randomize_seed=True,
-                guidance_scale=6,
+                guidance_scale=6.0,
                 num_inference_steps=25,
                 api_name="/infer"
         )
@@ -879,7 +881,8 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
             image_inpainting = load_kolors_inpainting(inpaint_prompt, input_image, image_mask_for_inpaint)
             if image_inpainting is None:
                 logger.info(f'load_kolors_inpainting_failed_')
-                return None, None, None, None, None, None, None
+                time_cost_str = f"load_kolors_inpainting_task__failed!"
+                return None, None, time_cost_str, gr.update(visible=(time_cost_str !='')), None, None, None
         else:
             # remove from mask
             if mask_source_radio == mask_source_segment:
@@ -916,7 +919,8 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
             image_inpainting = lama_cleaner_process(np.array(image_pil), np.array(mask_pil.convert("L")), cleaner_size_limit)
             if image_inpainting is None:
                 logger.info(f'run_anything_task_failed_')
-                return None, None, None, None, None, None, None
+                time_cost_str = f"run_anything_task[{task_type}]__failed!"
+                return None, None, time_cost_str, gr.update(visible=(time_cost_str !='')), None, None, None
 
             # output_images.append(image_inpainting)
             # run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)