[feat] make a preprocess an option

app.py

@@ -33,7 +33,7 @@ def clear_uploaded_images():
         if os.path.isfile(file_path):
             os.remove(file_path)
 
-def reflection_removal(input_image):
+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)."]
     
@@ -52,7 +52,7 @@ def reflection_removal(input_image):
         "--norm", "batch", "--epoch", "310",
         "--num_test", str(count_files(UPLOAD_DIR)),
         "--gpu_ids", "-1",
-        "--preprocess", "scale_width_and_crop",
+        "--preprocess", preprocess_type
     ]
     subprocess.run(cmd, check=True)
     
@@ -80,16 +80,16 @@ sample_images = [
     if file.endswith((".jpg", ".jpeg", ".png"))
 ]
 
-iface = gr.Interface(
-    fn=reflection_removal,
-    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"))
-    ],
-    title="Reflection Remover with Pix2Pix",
-    description="Upload images to remove reflections using a Pix2Pix model. You can also try the sample images below."
-)
+def gradio_interface():
+    preprocess_options = ["resize_and_crop", "crop", "scale_width", "scale_width_and_crop", "none"]
+    gr.Interface(
+        fn=reflection_removal,
+        inputs=[
+            gr.inputs.Image(type="file", label="Input Image"),
+            gr.inputs.Dropdown(preprocess_options, label="Preprocess Type")
+        ],
+        outputs=gr.outputs.Image(type="pil", label="Output Image")
+    ).launch()
 
 if __name__ == "__main__":
-    iface.launch()
+    gradio_interface()