[feat] make a preprocess an option

app.py

@@ -80,16 +80,23 @@ sample_images = [
     if file.endswith((".jpg", ".jpeg", ".png"))
 ]
 
-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()
+preprocess_options = [
+    "resize_and_crop", "crop", "scale_width", "scale_width_and_crop", "none"
+]
+
+iface = gr.Interface(
+    fn=reflection_removal,
+    inputs=[
+        gr.Image(type="filepath", label="Upload Image (JPG/PNG)"),
+        gr.Dropdown(choices=preprocess_options, label="Preprocessing Type", default="resize_and_crop")
+    ],
+    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."
+)
 
 if __name__ == "__main__":
-    gradio_interface()
+    iface.launch()