[optimization] clear uploads after process is completed

app.py

@@ -23,10 +23,16 @@ model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_FILE, cache_dir=CHE
 expected_model_path = os.path.join(CHECKPOINTS_DIR, MODEL_FILE)
 if not os.path.exists(expected_model_path):
     copyfile(model_path, expected_model_path)
-    
+
 def count_files(directory):
     return sum([len(files) for _, _, files in os.walk(directory)])
 
+def clear_uploaded_images():
+    for filename in os.listdir(UPLOAD_DIR):
+        file_path = os.path.join(UPLOAD_DIR, filename)
+        if os.path.isfile(file_path):
+            os.remove(file_path)
+
 def reflection_removal(input_image):
     if not input_image.lower().endswith((".jpg", ".jpeg", ".png")):
         return ["File is not supported (only .jpg, .jpeg, .png)."]
@@ -35,7 +41,6 @@ def reflection_removal(input_image):
     shutil.copy(input_image, file_path)
     
     input_filename = os.path.splitext(os.path.basename(file_path))[0]
-    print(f"Processing {input_filename}...")
     
     cmd = [
         "python", "test.py",
@@ -49,19 +54,19 @@ def reflection_removal(input_image):
     ]
     subprocess.run(cmd, check=True)
     
-    for root, _, files in os.walk(RESULTS_DIR):
-        for file in files:
-            print(os.path.join(root, file))
-    
     for root, _, files in os.walk(RESULTS_DIR):
         for file in files:
             if file.startswith(input_filename) and file.endswith("_fake.png"):
                 result_path = os.path.join(root, file)
-                return [Image.open(result_path)]
+                output_image = Image.open(result_path)
+                
+                clear_uploaded_images()
+                
+                return [output_image]
 
+    clear_uploaded_images()
     return ["No results found."]
 
-
 def use_sample_image(sample_image_name):
     sample_image_path = os.path.join(SAMPLE_DIR, sample_image_name)
     if not os.path.exists(sample_image_path):
@@ -75,9 +80,7 @@ sample_images = [
 
 iface = gr.Interface(
     fn=reflection_removal,
-    inputs=[
-        gr.Image(type="filepath", label="Upload Image (JPG/PNG)")
-    ],
+    inputs=[gr.Image(type="filepath", label="Upload Image (JPG/PNG)")],
     outputs=gr.Gallery(label="Results after Reflection Removal"),
     examples=[
         os.path.join("sample_images", img) for img in os.listdir("sample_images") if img.endswith((".jpg", ".jpeg", ".png"))