Update app.py

app.py

@@ -277,6 +277,40 @@ try:
         
         print("=== End Custom ZNormalization ===\n")
 
+    # Add PadOrCrop debug prints
+    from fastMONAI.vision_augmentation import PadOrCrop
+    original_padorcrop_encodes = PadOrCrop.encodes
+
+    def patched_padorcrop_encodes(self, o):
+        """Patched version of PadOrCrop.encodes with debug prints."""
+        print("\n=== PadOrCrop Debug ===")
+        print(f"Input type: {type(o)}")
+        if hasattr(o, 'shape'): print(f"Input shape: {o.shape}")
+        
+        # Safely get attributes with fallbacks
+        target_shape = getattr(self, 'size', o.shape[1:])  # Default to input shape
+        padding_mode = getattr(self, 'padding_mode', 'constant')
+        mask_name = getattr(self, 'mask_name', None)
+        
+        print(f"Target shape: {target_shape}")
+        print(f"Padding mode: {padding_mode}")
+        print(f"Mask name: {mask_name}")
+        
+        try:
+            result = original_padorcrop_encodes(self, o)
+            print(f"Output type: {type(result)}")
+            if hasattr(result, 'shape'): print(f"Output shape: {result.shape}")
+            return result
+        except Exception as e:
+            print(f"Error in PadOrCrop: {str(e)}")
+            print("Fallback - returning input as-is")
+            return o
+        
+        print("=== End PadOrCrop Debug ===\n")
+
+    PadOrCrop.encodes = patched_padorcrop_encodes
+    print("✅ Applied PadOrCrop patch with debug prints")
+
     # Replace the encodes method
     ZNormalization.encodes = patched_encodes
     print("✅ Applied ZNormalization patch to handle missing attributes in encodes method")