Update main_view.py

main_view.py

@@ -14,7 +14,7 @@ class Main_View:
         st.divider()
 
         image_input = 'test_image.jpg'
-        image_output = 'test_image_pred.png'
+        image_output = 'test_image_pred.PNG'
 
         col1_img, col2_img = st.columns(2,gap='medium')
         with col1_img:
@@ -23,10 +23,11 @@ class Main_View:
             st.image(image_output, caption="Prediction", use_container_width=True)
 
         st.divider()
-        col1_button, col2_button = st.columns(2,gap='medium',vertical_alignment='center')
-        with col1_button:
-            if st.button("Image Upload", key='main_upload', icon = ':material/upload_file:', type = 'secondary'):
-                self.app.change_page("Upload")
-        with col2_button:
-            if st.button("Real time detection", key='main_camera', icon = ':material/photo_camera:', type='primary'):
-                self.app.change_page("Camera")
+        if st.session_state.page == "Main":
+            col1_button, col2_button = st.columns(2,gap='medium',vertical_alignment='center')
+            with col1_button:
+                if st.button("Image Upload", key='main_upload', icon = ':material/upload_file:', type = 'secondary'):
+                    self.app.change_page("Upload")
+            with col2_button:
+                if st.button("Real time detection", key='main_camera', icon = ':material/photo_camera:', type='primary'):
+                    self.app.change_page("Camera")