Update

app.py

@@ -153,16 +153,22 @@ def main(image, lineart):
 def gradio_interface(image):
     image_path = 'temp_input_image.png'
     image.save(image_path)
+    
+    # 画像を再読込し、RGBAモードに変換する
     image = Image.open(image_path).convert('RGBA')
-    # アルファ情報を保存する
-    alpha = None
-    if image.mode == 'RGBA':
-        alpha = image.getchannel('A')
+
+    # アルファチャンネルを分離する
+    alpha = image.getchannel('A')
+    
+    # RGBチャンネルのみを処理する
+    rgb_image = image.convert('RGB')
     lineart = process_XDoG(image_path).convert('L')
-    replace_color_image = main(image, lineart)
-    # 保存したアルファ情報を適用して透過する
-    if alpha:
-        replace_color_image.putalpha(alpha)
+    replace_color_image = main(rgb_image, lineart)
+
+    # 処理済みの画像をRGBAに変換し、アルファチャンネルを再適用する
+    replace_color_image = replace_color_image.convert('RGBA')
+    replace_color_image.putalpha(alpha)
+    
     return replace_color_image
 
 # Gradioアプリを設定し、起動する