Update

app.py

@@ -54,7 +54,7 @@ def replace_color(image, color_1, blur_radius=2):
     while np.any(matches):
         new_matches = np.zeros_like(matches)
         match_num = np.sum(matches)
-        for i in range(len(data)):  # Removed tqdm
+        for i in range(len(data)):
             if matches[i]:
                 x, y = divmod(i, original_shape[1])
                 neighbors = [
@@ -166,11 +166,9 @@ class webui:
         #拡張子を取り除いたファイル名を取得
         image_name = os.path.splitext(image_path)[0]
         alpha = image.getchannel('A') if image.mode == 'RGBA' else None
-        image_png_path = f"{image_name}.png"
-        image.save(image_png_path)
-        image = Image.open(image_png_path).convert('RGBA')
+        image = Image.open(image_path).convert('RGBA')
         rgb_image = image.convert('RGB')
-        lineart = process_XDoG(image_png_path).convert('L')
+        lineart = process_XDoG(image_path).convert('L')
         replace_color_image = process_image(rgb_image, lineart).convert('RGBA')
         
         if alpha:
@@ -197,7 +195,7 @@ class webui:
         with self.demo:
             with gr.Row():
                 with gr.Column():
-                    input_image = gr.Image(type='filepath', label="Original Image")
+                    input_image = gr.Image(type='filepath', image_mode="RGBA", label="Original Image（png画像にのみ対応しています）")
                     submit = gr.Button(value="Start")
                 with gr.Row():
                     with gr.Column():