Update app.py

app.py

@@ -4,6 +4,14 @@ import os
 import zipfile
 
 
+def save_image(image, path):
+    # RGBA画像をRGBに変換して保存
+    if image.mode == 'RGBA':
+        image = image.convert('RGB')
+    
+    # 画像の保存
+    image.save(path)
+
 def process_image(image):
     # 元の画像のサイズ
     original_width, original_height = image.size
@@ -57,8 +65,6 @@ def process_image(image):
     # 出力される4つの画像を返す
     return outputs[0], outputs[1], outputs[2], outputs[3]
 
-
-
 def zip_files(zip_files, zip_path):
     with zipfile.ZipFile(zip_path, 'w') as zipf:
         for file_path in zip_files:
@@ -82,11 +88,11 @@ class webui:
         output3_path = f"{image_name}_3.png"
         output4_path = f"{image_name}_4.png"
         
-        # 画像を保存
-        output1.save(output1_path)
-        output2.save(output2_path)
-        output3.save(output3_path)
-        output4.save(output4_path)
+        # 画像を保存（変換処理あり）
+        save_image(output1, output1_path)
+        save_image(output2, output2_path)
+        save_image(output3, output3_path)
+        save_image(output4, output4_path)
 
         # 保存したファイルをリストに追加
         zip_files_list = [output1_path, output2_path, output3_path, output4_path]
@@ -119,4 +125,4 @@ class webui:
 
 if __name__ == "__main__":
     ui = webui()
-    ui.launch(share=True)
+    ui.launch(share=True)