[dwbug]

app.py

@@ -39,6 +39,11 @@ def clear_session_files(session_id):
         shutil.rmtree(session_dir)
 
 def reflection_removal(input_image, preprocess_type="resize_and_crop"):
+    if preprocess_type not in ["resize_and_crop", "crop", "scale_width", "scale_width_and_crop", "none"]:
+        return "Invalid preprocessing type selected. Please choose a valid option."
+    
+    print("Preprocessing Type:", preprocess_type)
+    print("Input Image:", input_image)
 
     session_id = generate_session_id()
     session_dir = os.path.join(UPLOAD_DIR, session_id)
@@ -119,12 +124,12 @@ preprocess_options = [
 
 iface = gr.Interface(
     fn=reflection_removal,
-    inputs=[
+    inputs=[ 
         gr.Image(type="filepath", label="Upload Image (JPG/PNG)"),
         gr.Dropdown(choices=preprocess_options, label="Preprocessing Type", value="resize_and_crop")
     ],
     outputs=gr.Image(label="Result after Reflection Removal"),
-    examples=[
+    examples=[ 
         [os.path.join(SAMPLE_DIR, img), "resize_and_crop"]
         for img in sample_images
     ],