[feat] optimize file and make output image instead of galery

app.py

@@ -33,16 +33,21 @@ def clear_uploaded_images():
         if os.path.isfile(file_path):
             os.remove(file_path)
 
+def clear_results():
+    for root, _, files in os.walk(RESULTS_DIR):
+        for file in files:
+            file_path = os.path.join(root, file)
+            os.remove(file_path)
+
 def reflection_removal(input_image, preprocess_type="resize_and_crop"):
     if not input_image.lower().endswith((".jpg", ".jpeg", ".png")):
-        return ["File is not supported (only .jpg, .jpeg, .png)."]
+        return "File is not supported (only .jpg, .jpeg, .png)."
     
     file_path = os.path.join(UPLOAD_DIR, os.path.basename(input_image))
     shutil.copy(input_image, file_path)
     
     input_filename = os.path.splitext(os.path.basename(file_path))[0]
     
-    
     cmd = [
         "python", "test.py",
         "--dataroot", UPLOAD_DIR,
@@ -63,11 +68,13 @@ def reflection_removal(input_image, preprocess_type="resize_and_crop"):
                 output_image = Image.open(result_path)
                 
                 clear_uploaded_images()
+                clear_results()
                 
-                return [output_image]
+                return output_image
 
     clear_uploaded_images()
-    return ["No results found."]
+    clear_results()
+    return "No results found."
 
 def use_sample_image(sample_image_name):
     sample_image_path = os.path.join(SAMPLE_DIR, sample_image_name)
@@ -90,7 +97,7 @@ iface = gr.Interface(
         gr.Image(type="filepath", label="Upload Image (JPG/PNG)"),
         gr.Dropdown(choices=preprocess_options, label="Preprocessing Type", value="resize_and_crop")
     ],
-    outputs=gr.Gallery(label="Results after Reflection Removal"),
+    outputs=gr.Image(label="Result after Reflection Removal"),
     examples=[
         [os.path.join("sample_images", img), "resize_and_crop"]
         for img in os.listdir("sample_images") if img.endswith((".jpg", ".jpeg", ".png"))
@@ -99,6 +106,5 @@ iface = gr.Interface(
     description="Upload images to remove reflections using a Pix2Pix model. You can also try the sample images below."
 )
 
-
 if __name__ == "__main__":
     iface.launch()