(ref_image, edit_image_dict, preprocess = None)
| 327 | return image |
| 328 | |
| 329 | def preprocess_input(ref_image, edit_image_dict, preprocess = None): |
| 330 | err_msg = "" |
| 331 | is_suc = True |
| 332 | if ref_image is not None: |
| 333 | ref_image = pillow_convert(ref_image, "RGB") |
| 334 | |
| 335 | if edit_image_dict is None: |
| 336 | edit_image = None |
| 337 | edit_mask = None |
| 338 | else: |
| 339 | edit_image = edit_image_dict["background"] |
| 340 | edit_mask = np.array(edit_image_dict["layers"][0])[:, :, 3] |
| 341 | if np.sum(np.array(edit_image)) < 1: |
| 342 | edit_image = None |
| 343 | edit_mask = None |
| 344 | elif np.sum(np.array(edit_mask)) < 1: |
| 345 | edit_image = pillow_convert(edit_image, "RGB") |
| 346 | w, h = edit_image.size |
| 347 | edit_mask = Image.new("L", (w, h), 255) |
| 348 | else: |
| 349 | edit_image = pillow_convert(edit_image, "RGB") |
| 350 | edit_mask = Image.fromarray(edit_mask).convert('L') |
| 351 | if ref_image is None and edit_image is None: |
| 352 | err_msg = "Please provide the reference image or edited image." |
| 353 | return None, None, None, False, err_msg |
| 354 | return edit_image, edit_mask, ref_image, is_suc, err_msg |
| 355 | |
| 356 | def run_chat( |
| 357 | prompt, |
nothing calls this directly
no outgoing calls
no test coverage detected