Update app.py

app.py

@@ -120,6 +120,7 @@ if st.session_state.screen == 1:
         if st.button(f"Select Image ↑", key="button_1"):
             st.session_state.image = img_1
             st.session_state.screen = 2
+            st.rerun()
     with row2_col2:
         img_2 = Image.open(image_path + "IMG_5357.JPG")
         img_2 = ImageOps.exif_transpose(img_2)
@@ -127,6 +128,7 @@ if st.session_state.screen == 1:
         if st.button(f"Select Image ↑", key="button_2"):
             st.session_state.image = img_2
             st.session_state.screen = 2
+            st.rerun()
     with row2_col3:
         img_3 = Image.open(image_path + "IMG_5368.JPG")
         img_3 = ImageOps.exif_transpose(img_3)
@@ -134,6 +136,7 @@ if st.session_state.screen == 1:
         if st.button(f"Select Image ↑", key="button_3"):
             st.session_state.image = img_3
             st.session_state.screen = 2
+            st.rerun()
     with row2_col4:
         img_4 = Image.open(image_path + "IMG_5369.JPG")
         img_4 = ImageOps.exif_transpose(img_4)
@@ -141,6 +144,7 @@ if st.session_state.screen == 1:
         if st.button(f"Select Image ↑", key="button_4"):
             st.session_state.image = img_4
             st.session_state.screen = 2
+            st.rerun()
     with row3_col1:
         img_5 = Image.open(image_path + "IMG_5382.JPG")
         img_5 = ImageOps.exif_transpose(img_5)
@@ -148,6 +152,7 @@ if st.session_state.screen == 1:
         if st.button(f"Select Image ↑", key="button_5"):
             st.session_state.image = img_5
             st.session_state.screen = 2
+            st.rerun()
     with row3_col2:
         img_6 = Image.open(image_path + "IMG_5383.JPG")
         img_6 = ImageOps.exif_transpose(img_6)
@@ -155,6 +160,7 @@ if st.session_state.screen == 1:
         if st.button(f"Select Image ↑", key="button_6"):
             st.session_state.image = img_6
             st.session_state.screen = 2
+            st.rerun()
     with row3_col3:
         img_7 = Image.open(image_path + "IMG_5408.JPG")
         img_7 = ImageOps.exif_transpose(img_7)
@@ -162,6 +168,7 @@ if st.session_state.screen == 1:
         if st.button(f"Select Image ↑", key="button_7"):
             st.session_state.image = img_7
             st.session_state.screen = 2
+            st.rerun()
     with row3_col4:
         img_8 = Image.open(image_path + "IMG_5409.JPG")
         img_8 = ImageOps.exif_transpose(img_8)
@@ -169,6 +176,7 @@ if st.session_state.screen == 1:
         if st.button(f"Select Image ↑", key="button_8"):
             st.session_state.image = img_8 
             st.session_state.screen = 2
+            st.rerun()
 
 #####
 ###