Update app.py

app.py

@@ -203,18 +203,18 @@ def upload_page():
             filepath = os.path.join(UPLOAD_DIR, filename)
             converted_filepath = os.path.join(UPLOAD_DIR, converted_filename)
             
-            # Save original file using cv2
-            nparr = np.frombuffer(bytes_data, np.uint8)
-            img = cv2.imdecode(nparr, cv2.IMREAD_UNCHANGED)
-            cv2.imwrite(filepath, img)
+            with open(filepath, "wb") as f:
+                f.write(bytes_data)
             
             if file_extension in ['.tiff', '.tif']:
                 st.info('Processing GeoTIFF image...')
                 rgb_image = read_pansharpened_rgb(filepath)
                 cv2.imwrite(converted_filepath, cv2.cvtColor(rgb_image, cv2.COLOR_RGB2BGR))
                 st.success(f'GeoTIFF converted to 8-bit image and saved as {converted_filename}')
+                img = Image.open(converted_filepath)
             else:
-                cv2.imwrite(converted_filepath, img)
+                img = Image.open(filepath)
+                img.save(converted_filepath)
             
             if os.path.exists(converted_filepath):
                 st.success(f"Image saved successfully: {converted_filepath}")
@@ -223,15 +223,12 @@ def upload_page():
             else:
                 st.error(f"Failed to save image: {converted_filepath}")
             
-            # Display image using PIL (as Streamlit's st.image works better with PIL)
-            pil_img = Image.open(converted_filepath)
-            st.image(pil_img, caption='Uploaded Image', use_column_width=True)
+            st.image(img, caption='Uploaded Image', use_column_width=True)
             st.success(f'Image processed and saved as {converted_filename}')
 
             st.session_state.filename = converted_filename
 
-            # Use cv2 for further processing
-            img_array = cv2.imread(converted_filepath)
+            img_array = np.array(img)
 
             if img_array.shape[0] > 650 or img_array.shape[1] > 650:
                 st.info('Large image detected. Using patch-based processing.')
@@ -240,14 +237,11 @@ def upload_page():
             else:
                 st.info('Small image detected. Processing whole image at once.')
                 with st.spinner('Analyzing image...'):
-                    # Convert to RGB for the model
-                    img_rgb = cv2.cvtColor(img_array, cv2.COLOR_BGR2RGB)
-                    img_pil = Image.fromarray(img_rgb)
-                    img_transformed = transforms(img_pil)
+                    img_transformed = transforms(img)
                     prediction = predict(img_transformed)
                     full_mask = (prediction > 0.5).astype(np.uint8) * 255
 
-            full_mask = refine_mask(full_mask)
+            full_mask = refine_mask(full_mask)#-----------------------------------------------------------------------
             mask_filename = f"mask_{timestamp}.png"
             mask_filepath = os.path.join(MASK_DIR, mask_filename)
             cv2.imwrite(mask_filepath, full_mask)