Update app.py

app.py

@@ -10,7 +10,7 @@ def hex_to_rgb(hex_code):
     # Convert the hex code to RGB
     return tuple(int(hex_code[i:i+2], 16) for i in (0, 2, 4))
 
-def resize_and_overlay_image(input_image, reduction_percentage, shift_pixels, shift_pixels_ud, background_color, crop_top, crop_bottom):
+def resize_and_overlay_image(input_image, reduction_percentage, shift_pixels, shift_pixels_ud, background_color, crop_top_percentage, crop_bottom_percentage):
     # Check if the input image is empty
     if input_image.size == 0:
         return None
@@ -24,6 +24,10 @@ def resize_and_overlay_image(input_image, reduction_percentage, shift_pixels, sh
     # Get the image dimensions
     height, width = img.shape[:2]
 
+    # Calculate the number of pixels to crop from the top and bottom as a percentage of the image height
+    crop_top = int(height * crop_top_percentage / 100)
+    crop_bottom = int(height * crop_bottom_percentage / 100)
+
     # Crop the top and/or bottom of the image
     img = img[crop_top:height - crop_bottom, :]
 
@@ -59,13 +63,13 @@ iface = gr.Interface(
         gr.inputs.Slider(minimum=-150, maximum=150, step=10, default=0, label="Shift Pixels Left / Right"),
         gr.inputs.Slider(minimum=-150, maximum=250, step=10, default=0, label="Shift Pixels Up / Down"),
         gr.inputs.Textbox(default="#ffffff", label="Background Color (Hex Code)"),
-        gr.inputs.Slider(minimum=0, maximum=100, step=10, default=0, label="Crop Top (pixels)"),
-        gr.inputs.Slider(minimum=0, maximum=100, step=10, default=0, label="Crop Bottom (pixels)")
+        gr.inputs.Slider(minimum=0, maximum=100, step=10, default=0, label="Crop Top (%)"),
+        gr.inputs.Slider(minimum=0, maximum=100, step=10, default=0, label="Crop Bottom (%)")
     ],
     outputs=[gr.outputs.Image(type="numpy", label="Result"), gr.outputs.Text(label="Image Info")],
     title="Image Resizer",
-    description="Reduce the size of an image, overlay it on a colored background, shift it, and crop the top and/or bottom."
+    description="Reduce the size of an image, overlay it on a colored background, shift it, and crop the top and/or bottom as a percentage."
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()