update add function that cleans up files

app.py

@@ -1,15 +1,28 @@
 from zipfile import ZipFile
 import numpy as np
-from matplotlib import pyplot as plt
 
 import os, shutil
 
 import gradio as gr
-from PIL import Image, ImageOps
+from PIL import Image
 import sys, os
 from rembg import remove
 
 
+def complete():
+    return "complete"
+
+
+def clean(text_output):
+    if text_output=="complete":
+        print("clean up")
+        os.remove("objects.zip")
+        return ""
+    else:
+        pass
+    
+    
+
 def run_rembg(img):
     output = remove(img)
     output_pil = Image.fromarray(output)
@@ -58,16 +71,11 @@ def load_zip(inputs):
 
         image_name = image_name.replace("jpg", "png")
         cropped_image.save(f"objects/{image_name}")
-        # plt.imsave(f"objects/{image_name}", img_array)
     
     shutil.make_archive("objects", "zip", "objects")
-    # ここはresults
-    with ZipFile("results.zip", "w") as zip_file:
-        zip_file.write("objects.zip")
-
     shutil.rmtree("objects")
-    # ここはresults
-    return "results.zip"
+    
+    return "objects.zip", complete()
 
 
 
@@ -99,14 +107,21 @@ if __name__=="__main__":
             with gr.Row():
                 image_input = gr.File(file_count="multiple")
                 image_output = gr.File()
+                text_output = gr.Textbox(visible=False)
             
             btn = gr.Button("Run!")
             
             btn.click(
                 fn=load_zip,
                 inputs=image_input,
-                outputs=image_output
+                outputs=[image_output, text_output]
                 )
+            text_output.change(
+                fn=clean,
+                inputs=text_output,
+                outputs=text_output
+
+            )
 
 
         with gr.Tab("Image"):