Update app.py

app.py

@@ -41,20 +41,17 @@ PATCHES_DIR = "patches"
 PRED_PATCHES_DIR = "pred_patches"
 CSV_LOG_PATH = "image_log.csv"
 
-
-def split(image, destination = PATCHES_DIR, patch_size = 650):
+def split(image, destination=PATCHES_DIR, patch_size=650):
     img = cv2.imread(image)
-    h,w,_ = img.shape
+    h, w, _ = img.shape
     for y in range(0, h, patch_size):
         for x in range(0, w, patch_size):
             patch = img[y:y+patch_size, x:x+patch_size]
-
-            
             patch_filename = f"patch_{y}_{x}.png" 
             patch_path = os.path.join(destination, patch_filename)
             cv2.imwrite(patch_path, patch)
 
-def merge(patch_folder , dest_image = 'out.png', image_shape = None):
+def merge(patch_folder, dest_image='out.png', image_shape=None):
     merged = np.zeros(image_shape[:-1] + (3,), dtype=np.uint8)
     for filename in os.listdir(patch_folder):
         if filename.endswith(".png"):
@@ -273,7 +270,7 @@ def upload_page():
         else:
             st.success('Image analyzed')
             st.session_state.page = 'result'
-            st.rerun()
+            st.experimental_rerun()
 
 def result_page():
     st.title('Analysis Result')
@@ -283,9 +280,7 @@ def result_page():
         if st.button('Back to Upload'):
             st.session_state.page = 'upload'
             st.session_state.file_uploaded = False
-            st.session_state.filename = None
-            st.session_state.mask_filename = None
-            st.rerun()
+            st.experimental_rerun()
         return
 
     col1, col2 = st.columns(2)
@@ -328,12 +323,9 @@ def result_page():
         st.error("Image or mask file not found for overlay.")
 
     if st.button('Back to Upload'):
-        
         st.session_state.page = 'upload'
         st.session_state.file_uploaded = False
-        st.session_state.filename = None
-        st.session_state.mask_filename = None
-        st.rerun()
+        st.experimental_rerun()
 
 def main():
     st.title('Building area estimation')