update UI and limit the image size

app.py

@@ -185,6 +185,54 @@ def initialize_services():
         import traceback
         return f"❌ Failed to initialize services: {str(e)}\n{traceback.format_exc()}"
 
+
+def validate_image_dimensions(image):
+    """Validate image dimensions before GPU allocation.
+    Returns (is_valid, message)
+    """
+    if image is None:
+        return True, ""  # Allow None to pass through
+    
+    # Handle PIL Image or numpy array
+    if isinstance(image, np.ndarray):
+        img_pil = Image.fromarray(image)
+    else:
+        img_pil = image
+    
+    # Check dimensions: width must be >= height (landscape orientation)
+    width, height = img_pil.size
+    if width < height:
+        error_msg = f"❌ Please upload a landscape image (width ≥ height).\nCurrent image: {width}x{height} (portrait orientation)"
+        return False, error_msg
+    
+    return True, ""
+
+
+def validate_and_process_wrapper(image, session_state, progress=gr.Progress()):
+    """Wrapper function to validate image before GPU allocation"""
+    # Skip processing if image is None (intermediate state during replacement)
+    if image is None:
+        return ("Waiting for image upload...", 
+                gr.update(interactive=False),
+                None,
+                False,
+                False,
+                session_state)
+    
+    # Validate image dimensions BEFORE GPU allocation
+    is_valid, error_msg = validate_image_dimensions(image)
+    if not is_valid:
+        return (error_msg,
+                gr.update(interactive=False),
+                None,
+                False,
+                False,
+                session_state)
+    
+    # If validation passes, proceed with GPU-intensive processing
+    return process_image_upload(image, session_state, progress)
+
+
 @spaces.GPU(duration=120)
 def process_image_upload(image, session_state, progress=gr.Progress()):
     """Process uploaded image and run hand reconstruction"""
@@ -205,16 +253,7 @@ def process_image_upload(image, session_state, progress=gr.Progress()):
             time.sleep(2)
     
     if hand_reconstructor is None:
-        return ("Models not initialized. Please wait for initialization to complete.", 
-                gr.update(interactive=False),
-                None,
-                False,
-                False,
-                session_state)
-    
-    # Skip processing if image is None (intermediate state during replacement)
-    if image is None:
-        return ("Waiting for image upload...", 
+        return ("Services not initialized. Please wait for initialization to complete.",
                 gr.update(interactive=False),
                 None,
                 False,
@@ -875,7 +914,7 @@ def create_gradio_interface():
         # Use only change event to handle all image updates (upload, drag-and-drop, example selection)
         # This prevents duplicate processing that occurs when both upload and change events fire
         input_image.change(
-            fn=process_image_upload,
+            fn=validate_and_process_wrapper,
             inputs=[input_image, session_state],
             outputs=[recon_status, generate_btn, hand_data, detected_left, detected_right, session_state],
             show_progress='full'  # Show progress bar for reconstruction