Update app.py

app.py

@@ -37,15 +37,19 @@ def allowed_file(filename):
 def preprocess_image(image_path):
     """Preprocess image for CNN model (32x32 grayscale)"""
     try:
-        # Read image as grayscale
-        img = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE)
-        
-        if img is None:
+        # Read original image for display
+        img_original = cv2.imread(image_path)
+        if img_original is None:
             raise ValueError("Could not read image")
         
+        # Convert to RGB for display
+        img_display = cv2.cvtColor(img_original, cv2.COLOR_BGR2RGB)
+        
+        # Read as grayscale for model
+        img = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE)
         print(f"Original image shape: {img.shape}")
         
-        # Resize to 32x32
+        # Resize to 32x32 for model
         img_resized = cv2.resize(img, (32, 32))
         
         # Apply histogram equalization
@@ -59,9 +63,6 @@ def preprocess_image(image_path):
         
         print(f"Model input shape: {img_input.shape}")
         
-        # For display - convert to RGB
-        img_display = cv2.cvtColor(img_resized, cv2.COLOR_GRAY2RGB)
-        
         return img_input, img_display
         
     except Exception as e: