Update pages/Data Collection.py

pages/Data Collection.py

@@ -1506,7 +1506,6 @@ elif st.session_state.current_page == "video_processing":
 
 
 elif st.session_state.current_page == "image_transformations":
-    # Content for Image Transformations Page
     st.markdown("""
         <h2 style="color: #BB3385;">Image Augmentation Techniques</h2>
     """, unsafe_allow_html=True)
@@ -1819,41 +1818,33 @@ elif st.session_state.current_page == "image_transformations":
     """, language="python")
 
     col1, col2 = st.columns(2)
+        with col1:
+            st.markdown("""
+                <a href="https://colab.research.google.com/drive/19mQM-7VY0F0y9NyecrhbU5kyOY9j9oIk?usp=sharing" target="_blank">
+                    <button style="background-color: #4CAF50; color: white; padding: 10px; font-size: 16px; border: none; cursor: pointer;">
+                        Colab Notes
+                    </button>
+                </a>
+            """, unsafe_allow_html=True)
+        with col2:
+            if st.button("Main Page"):
+                st.session_state.current_page = "explore_images_video"
     
-    # First row: Colab Notes and Main Page
-    with col1:
-        st.markdown("""
-            <a href="https://colab.research.google.com/drive/19mQM-7VY0F0y9NyecrhbU5kyOY9j9oIk?usp=sharing" target="_blank">
-                <button style="background-color: #4CAF50; color: white; padding: 10px; font-size: 16px; border: none; cursor: pointer;">
-                    Colab Notes
-                </button>
-            </a>
-        """, unsafe_allow_html=True)
-    
-    with col2:
-        if st.button("Main Page"):
-            navigate_to("explore_images_video")  # Main page: Images & Videos
-    
-    # Second row: Previous Page and Next Page
-    col3, col4 = st.columns(2)
-    
-    with col3:
-        if st.button("⬅️ Prev Page"):
-            navigate_to("video_processing")  # Previous page: Video Processing with OpenCV
-    
-    with col4:
-        if st.button("➡️ Next Page"):
-            navigate_to("opencv_projects")  # Next page: OpenCV Projects
+        col3, col4 = st.columns(2)
+        with col3:
+            if st.button("⬅️ Prev Page"):
+                st.session_state.current_page = "video_processing"
+        with col4:
+            if st.button("➡️ Next Page"):
+                st.session_state.current_page = "opencv_projects"
  
 
 
-if st.session_state.current_page == "opencv_projects":
+elif st.session_state.current_page == "opencv_projects":
     st.markdown("""
         <h2 style="color: #BB3385;">OpenCV Projects</h2>
     """, unsafe_allow_html=True)
 
-    
-
     # Project 1: Converting an Image into Tabular Data
     st.markdown("""
         <h3 style="color: #5b2c6f;">Converting an Image into Tabular Data</h3>