Update app.py

app.py

@@ -28,13 +28,13 @@ import gc
 snapshot_download(repo_id="LiuZichen/MagicQuill-models", repo_type="model", local_dir="models")
 # HF_TOKEN = os.environ.get("HF_TOKEN")
 # The client has been made public. Welcome to duplicate our repo.
-_HELPER_SPACE = "LiuZichen/MagicQuillHelper"
-_client = None
-def client():
-    global _client
-    if _client is None:
-        _client = Client(_HELPER_SPACE)
-    return _client
+# _HELPER_SPACE = "LiuZichen/MagicQuillHelper"
+# _client = None
+# def client():
+#     global _client
+#     if _client is None:
+#         _client = Client(_HELPER_SPACE)
+#     return _client
 scribbleColorEditModel = ScribbleColorEditModel()
 
 def tensor_to_numpy(tensor):
@@ -108,49 +108,49 @@ def prepare_images_and_masks(total_mask, original_image, add_color_image, add_ed
     return add_color_image_tensor, original_image_tensor, total_mask, add_edge_mask, remove_edge_mask
 
 def guess_prompt_handler(original_image, add_color_image, add_edge_image):
-    original_image_tensor = load_and_preprocess_image(original_image)
+    # original_image_tensor = load_and_preprocess_image(original_image)
     
-    if add_color_image:
-        add_color_image_tensor = load_and_preprocess_image(add_color_image)
-    else:
-        add_color_image_tensor = original_image_tensor
+    # if add_color_image:
+    #     add_color_image_tensor = load_and_preprocess_image(add_color_image)
+    # else:
+    #     add_color_image_tensor = original_image_tensor
 
-    width, height = original_image_tensor.shape[1], original_image_tensor.shape[2]
-    add_edge_mask = create_alpha_mask(add_edge_image) if add_edge_image else torch.zeros((1, height, width), dtype=torch.float32, device="cpu")
+    # width, height = original_image_tensor.shape[1], original_image_tensor.shape[2]
+    # add_edge_mask = create_alpha_mask(add_edge_image) if add_edge_image else torch.zeros((1, height, width), dtype=torch.float32, device="cpu")
     
-    original_image_numpy = tensor_to_numpy(original_image_tensor.squeeze(0))
-    add_color_image_numpy = tensor_to_numpy(add_color_image_tensor.squeeze(0))
-    add_edge_mask_numpy = tensor_to_numpy(add_edge_mask.squeeze(0).unsqueeze(-1))
-    original_image_numpy = cv2.cvtColor(original_image_numpy, cv2.COLOR_RGB2BGR)
-    add_color_image_numpy = cv2.cvtColor(add_color_image_numpy, cv2.COLOR_RGB2BGR)
+    # original_image_numpy = tensor_to_numpy(original_image_tensor.squeeze(0))
+    # add_color_image_numpy = tensor_to_numpy(add_color_image_tensor.squeeze(0))
+    # add_edge_mask_numpy = tensor_to_numpy(add_edge_mask.squeeze(0).unsqueeze(-1))
+    # original_image_numpy = cv2.cvtColor(original_image_numpy, cv2.COLOR_RGB2BGR)
+    # add_color_image_numpy = cv2.cvtColor(add_color_image_numpy, cv2.COLOR_RGB2BGR)
 
-    original_image_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png", mode='w+b')
-    add_color_image_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png", mode='w+b')
-    add_edge_mask_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png", mode='w+b')
+    # original_image_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png", mode='w+b')
+    # add_color_image_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png", mode='w+b')
+    # add_edge_mask_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png", mode='w+b')
     
-    cv2.imwrite(original_image_file.name, original_image_numpy)
-    cv2.imwrite(add_color_image_file.name, add_color_image_numpy)
-    cv2.imwrite(add_edge_mask_file.name, add_edge_mask_numpy)
+    # cv2.imwrite(original_image_file.name, original_image_numpy)
+    # cv2.imwrite(add_color_image_file.name, add_color_image_numpy)
+    # cv2.imwrite(add_edge_mask_file.name, add_edge_mask_numpy)
 
-    original_image_file.close()
-    add_color_image_file.close()
-    add_edge_mask_file.close()
+    # original_image_file.close()
+    # add_color_image_file.close()
+    # add_edge_mask_file.close()
     
-    res = client().predict(
-        handle_file(original_image_file.name),
-        handle_file(add_color_image_file.name),
-        handle_file(add_edge_mask_file.name),
-        api_name="/guess_prompt"
-    )
+    # res = client().predict(
+    #     handle_file(original_image_file.name),
+    #     handle_file(add_color_image_file.name),
+    #     handle_file(add_edge_mask_file.name),
+    #     api_name="/guess_prompt"
+    # )
     
-    if original_image_file and os.path.exists(original_image_file.name):
-        os.remove(original_image_file.name)
-    if add_color_image_file and os.path.exists(add_color_image_file.name):
-        os.remove(add_color_image_file.name)
-    if add_edge_mask_file and os.path.exists(add_edge_mask_file.name):
-        os.remove(add_edge_mask_file.name)
+    # if original_image_file and os.path.exists(original_image_file.name):
+    #     os.remove(original_image_file.name)
+    # if add_color_image_file and os.path.exists(add_color_image_file.name):
+    #     os.remove(add_color_image_file.name)
+    # if add_edge_mask_file and os.path.exists(add_edge_mask_file.name):
+    #     os.remove(add_edge_mask_file.name)
 
-    return res
+    return ""
 
 def generate(ckpt_name, total_mask, original_image, add_color_image, add_edge_image, remove_edge_image, positive_prompt, negative_prompt, grow_size, stroke_as_edge, fine_edge, edge_strength, color_strength, inpaint_strength, seed, steps, cfg, sampler_name, scheduler):
     add_color_image, original_image, total_mask, add_edge_mask, remove_edge_mask = prepare_images_and_masks(total_mask, original_image, add_color_image, add_edge_image, remove_edge_image)