Add validation to user input

app.py

@@ -735,13 +735,32 @@ def get_sample_images():
     return []
 
 # Gradio interface using Blocks for better layout control
+def validate_image_size(image):
+    """Validate that the image is exactly 130x130 pixels"""
+    if image is None:
+        return False, "No image provided"
+
+    width, height = image.size
+    if width != 130 or height != 130:
+        return False, f"Image must be exactly 130x130 pixels. Your image is {width}x{height} pixels."
+
+    return True, "Valid image size"
+
 def upscale_and_display(image):
     if image is None:
-        return None
+        return None, "Please upload an image or select a sample image."
+
+    # Validate image size
+    is_valid, message = validate_image_size(image)
+    if not is_valid:
+        return None, f"❌ Error: {message}"
 
-    # Get the super-resolution output
-    upscaled = upscale_image(image)
-    return upscaled
+    try:
+        # Get the super-resolution output
+        upscaled = upscale_image(image)
+        return upscaled, "✅ Image successfully enhanced!"
+    except Exception as e:
+        return None, f"❌ Error processing image: {str(e)}"
 
 def select_sample_image(image_path):
     if image_path:
@@ -805,6 +824,7 @@ css = generate_css()
 with gr.Blocks(css=css, title="HAT Super-Resolution for Satellite Images") as iface:
     gr.Markdown("# HAT Super-Resolution for Satellite Images")
     gr.Markdown("Upload a satellite image or select a sample to enhance its resolution by 4x.")
+    gr.Markdown("⚠️ **Important**: Images must be exactly **130x130 pixels** for the model to work properly.")
 
     # Sample images
     sample_images = get_sample_images()
@@ -823,7 +843,7 @@ with gr.Blocks(css=css, title="HAT Super-Resolution for Satellite Images") as if
     with gr.Row():
         input_image = gr.Image(
             type="pil",
-            label="Input Image",
+            label="Input Image (must be 130x130 pixels)",
             elem_classes="image-container",
             sources=["upload"],
             height=500,
@@ -842,12 +862,19 @@ with gr.Blocks(css=css, title="HAT Super-Resolution for Satellite Images") as if
 
     submit_btn = gr.Button("Enhance Image", variant="primary")
 
+    # Status message
+    status_message = gr.Textbox(
+        label="Status",
+        interactive=False,
+        show_label=True
+    )
+
     # Event handlers
     if sample_images:
         for btn, img_path in sample_buttons:
             btn.click(fn=lambda path=img_path: select_sample_image(path), outputs=input_image)
 
-    submit_btn.click(fn=upscale_and_display, inputs=input_image, outputs=output_image)
+    submit_btn.click(fn=upscale_and_display, inputs=input_image, outputs=[output_image, status_message])
 
 if __name__ == "__main__":
     iface.launch()