Refactor image processing to save temporary files for original and processed images in PNG and GIF formats

app.py

@@ -194,6 +194,10 @@ def remove_bg_fn(image):
     im = im.convert("RGB")
     origin = im.copy()
     
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_file:
+        origin_filepath = temp_file.name
+        origin.save(origin_filepath, format="PNG")
+    
     if im.format == "GIF":
         frames = []
         for frame in ImageSequence.Iterator(im):
@@ -201,17 +205,22 @@ def remove_bg_fn(image):
             processed_frame = process_image(frame)
             frames.append(processed_frame)
         processed_image = frames[0]
-        processed_image.save(
-            io.BytesIO(),
-            format="GIF",
-            save_all=True,
-            append_images=frames[1:],
-            loop=0,
-        )
+        with tempfile.NamedTemporaryFile(suffix=".gif", delete=False) as temp_file:
+            temp_filepath = temp_file.name
+            processed_image.save(
+                temp_filepath,
+                format="GIF",
+                save_all=True,
+                append_images=frames[1:],
+                loop=0,
+            )
     else:
         processed_image = process_image(im)
+        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_file:
+            temp_filepath = temp_file.name
+            processed_image.save(temp_filepath, format="PNG")
     
-    return (processed_image, origin)
+    return (temp_filepath, origin_filepath)
 
 @spaces.GPU
 def process_image(image):
@@ -254,6 +263,10 @@ def apply_bg_image(image, background_file=None, background_color=None, bg_type="
         input_image = load_img(image, output_type="pil")
         origin = input_image.copy()
         
+        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_file:
+            origin_filepath = temp_file.name
+            origin.save(origin_filepath, format="PNG")
+        
         color_profile = input_image.info.get("icc_profile")
 
         if background_file is not None:
@@ -275,19 +288,23 @@ def apply_bg_image(image, background_file=None, background_color=None, bg_type="
                 frames.append(output_frame)
 
             output_image = frames[0]
-            output_image.save(
-                "output.gif",
-                format="GIF",
-                save_all=True,
-                append_images=frames[1:],
-                loop=0,
-                icc_profile=color_profile,
-            )
+            with tempfile.NamedTemporaryFile(suffix=".gif", delete=False) as temp_file:
+                temp_filepath = temp_file.name
+                output_image.save(
+                    temp_filepath,
+                    format="GIF",
+                    save_all=True,
+                    append_images=frames[1:],
+                    loop=0,
+                    icc_profile=color_profile,
+                )
         else:
             output_image = apply_background(input_image, background_image)
-            output_image.save("output.png", format="PNG", optimize=True, icc_profile=color_profile)
+            with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_file:
+                temp_filepath = temp_file.name
+                output_image.save(temp_filepath, format="PNG", optimize=True, icc_profile=color_profile)
 
-        return (output_image, origin)
+        return (temp_filepath, origin_filepath)
     except Exception as e:
         return str(e)
 
@@ -454,6 +471,7 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
                     gr.update(visible=True),
                     gr.update(visible=False),
                     gr.update(visible=False),
+                    gr.update(visible=False),
                 )
             elif bg_type == "Image":
                 return (
@@ -467,12 +485,14 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
                     gr.update(visible=False),
                     gr.update(visible=False),
                     gr.update(visible=True),
+                    gr.update(visible=True),
                 )
             else:
                 return (
                     gr.update(visible=False),
                     gr.update(visible=False),
                     gr.update(visible=False),
+                    gr.update(visible=False),
                 )
 
         bg_type.change(