Update app.py

app.py

@@ -156,6 +156,12 @@ def upload_page():
 
     image = st.file_uploader('Choose a satellite image', type=['jpg', 'png', 'jpeg', 'tiff', 'tif'])
 
+    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.experimental_rerun()
     if image is not None and not st.session_state.file_uploaded:
         try:
             bytes_data = image.getvalue()
@@ -281,7 +287,7 @@ def result_page():
             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)
@@ -323,9 +329,13 @@ def result_page():
     else:
         st.error("Image or mask file not found for overlay.")
 
+    
     if st.button('Back to Upload'):
-        
-        st.rerun()
+        st.session_state.page = 'upload'
+        st.session_state.file_uploaded = False
+        st.session_state.filename = None
+        st.session_state.mask_filename = None
+        st.experimental_rerun()
 
 def main():
     st.title('Building area estimation')