Update app.py

app.py

@@ -71,9 +71,11 @@ if st.session_state.screen == 0:
         st.markdown("<p style='text-align: left; color: black;'>This tool allows you to take a photo with your phone of a <i>Where's Waldo?</i> book page. Using the computer vision model YOLOv8-large, this tool find Waldo and his friends Wenda, Odlaw, Wizard, and Woof.</p>", unsafe_allow_html=True)
 
         # proceed button
-        if st.button(f"Proceed", key="Proceed"):
-            st.session_state.screen = 1 
-            st.rerun()
+        _, row1a_col1, _ = st.columns([5,1,5])
+        with row1a_col1:
+            if st.button("Proceed", key="Proceed", use_container_width=True):
+                st.session_state.screen = 1 
+                st.rerun()
         
         # image of Waldo and his friends
         st.image("/home/user/app/home_page_images/Waldo_Friends.jpg", use_column_width=True)
@@ -107,7 +109,6 @@ if st.session_state.screen == 1:
             uploaded_image = Image.open(uploaded_image)
             uploaded_image = ImageOps.exif_transpose(uploaded_image)
             st.session_state.image = uploaded_image
-            uploaded_image = None
             st.session_state.screen = 2
             st.rerun()
 
@@ -119,7 +120,7 @@ if st.session_state.screen == 1:
         img_1 = Image.open(image_path + "IMG_5356.JPG")
         img_1 = ImageOps.exif_transpose(img_1)
         st.image(img_1, use_column_width=True)
-        if st.button(f"Select Image ↑", key="button_1"):
+        if st.button("Select Image ↑", key="button_1", use_container_width=True):
             st.session_state.image = img_1
             st.session_state.screen = 2
             st.rerun()
@@ -127,7 +128,7 @@ if st.session_state.screen == 1:
         img_2 = Image.open(image_path + "IMG_5357.JPG")
         img_2 = ImageOps.exif_transpose(img_2)
         st.image(img_2, use_column_width=True)
-        if st.button(f"Select Image ↑", key="button_2"):
+        if st.button("Select Image ↑", key="button_2", use_container_width=True):
             st.session_state.image = img_2
             st.session_state.screen = 2
             st.rerun()
@@ -135,15 +136,15 @@ if st.session_state.screen == 1:
         img_3 = Image.open(image_path + "IMG_5368.JPG")
         img_3 = ImageOps.exif_transpose(img_3)
         st.image(img_3, use_column_width=True)
-        if st.button(f"Select Image ↑", key="button_3"):
+        if st.button("Select Image ↑", key="button_3", use_container_width=True):
             st.session_state.image = img_3
             st.session_state.screen = 2
             st.rerun()
     with row3_col4:
-        img_4 = Image.open(image_path + "IMG_5369.JPG")
+        img_4 = Image.open(image_path + "IMG_5369.JPG", use_container_width=True)
         img_4 = ImageOps.exif_transpose(img_4)
         st.image(img_4, use_column_width=True)
-        if st.button(f"Select Image ↑", key="button_4"):
+        if st.button("Select Image ↑", key="button_4", use_container_width=True):
             st.session_state.image = img_4
             st.session_state.screen = 2
             st.rerun()
@@ -151,7 +152,7 @@ if st.session_state.screen == 1:
         img_5 = Image.open(image_path + "IMG_5382.JPG")
         img_5 = ImageOps.exif_transpose(img_5)
         st.image(img_5, use_column_width=True)
-        if st.button(f"Select Image ↑", key="button_5"):
+        if st.button("Select Image ↑", key="button_5", use_container_width=True):
             st.session_state.image = img_5
             st.session_state.screen = 2
             st.rerun()
@@ -159,7 +160,7 @@ if st.session_state.screen == 1:
         img_6 = Image.open(image_path + "IMG_5383.JPG")
         img_6 = ImageOps.exif_transpose(img_6)
         st.image(img_6, use_column_width=True)
-        if st.button(f"Select Image ↑", key="button_6"):
+        if st.button("Select Image ↑", key="button_6", use_container_width=True):
             st.session_state.image = img_6
             st.session_state.screen = 2
             st.rerun()
@@ -167,7 +168,7 @@ if st.session_state.screen == 1:
         img_7 = Image.open(image_path + "IMG_5408.JPG")
         img_7 = ImageOps.exif_transpose(img_7)
         st.image(img_7, use_column_width=True)
-        if st.button(f"Select Image ↑", key="button_7"):
+        if st.button("Select Image ↑", key="button_7", use_container_width=True):
             st.session_state.image = img_7
             st.session_state.screen = 2
             st.rerun()
@@ -175,7 +176,7 @@ if st.session_state.screen == 1:
         img_8 = Image.open(image_path + "IMG_5409.JPG")
         img_8 = ImageOps.exif_transpose(img_8)
         st.image(img_8, use_column_width=True)
-        if st.button(f"Select Image ↑", key="button_8"):
+        if st.button("Select Image ↑", key="button_8", use_container_width=True):
             st.session_state.image = img_8 
             st.session_state.screen = 2
             st.rerun()
@@ -194,7 +195,7 @@ if st.session_state.screen == 2:
     
     with row5_col3:
         # navigation
-        row5a_col0, row5a_col1, row5a_col2= st.columns([1,1,1], gap="medium")
+        row5a_col0, row5a_col1, row5a_col2= st.columns([1,1,1], gap="large")
         with row5a_col0:
             if st.button("Back", use_container_width=True):
                 if st.session_state.image_counter > 0: