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