Update

app.py

@@ -156,8 +156,14 @@ def main(image, lineart):
 
 
 # Gradioインターフェース用のメイン関数
-def gradio_interface(image):
-    image_path = 'temp_input_image.png'
+def gradio_interface(image_path):
+
+    # 画像を読み込む
+    image = Image.open(image_path).convert('RGBA')
+
+    #元画像の名前を取得
+    image_name = os.path.basename(image_path)
+
     # 元のアルファチャンネルを保持
     alpha = image.getchannel('A') if image.mode == 'RGBA' else None
 
@@ -173,15 +179,20 @@ def gradio_interface(image):
     # アルファチャンネルを再適用
     if alpha:
         replace_color_image.putalpha(alpha)
+    # image_nameをreplace_color_image_pathに反映
+    replace_color_image_path = image_name + 'replace_color.png'
 
-    return replace_color_image
+    # 透過pngで保存
+    replace_color_image.save(replace_color_image_path)
+
+    return replace_color_image_path
 
 
 # Gradioアプリを設定し、起動する
 iface = gr.Interface(
     fn=gradio_interface,
-    inputs=gr.Image(type='pil', label="Original Image"),
-    outputs=gr.Image(type='pil', label="Processed Image", format="png"), 
+    inputs=gr.Image(type='filepath', label="Original Image"),
+    outputs=gr.Image(type='filepath', label="Processed Image"), 
     title="LineArt-Removar",
     description="画像をアップロードして線画を除去します。"
 )