Update pages/Data Collection.py

pages/Data Collection.py

@@ -1853,13 +1853,20 @@ elif st.session_state.current_page == "image_transformations":
             transformed_images.append(transformed_image)
     
         return transformed_images
-    
+        
     # Streamlit App
     st.title("Dynamic Affine Transformation Tool")
     st.write("Select a transformation type to proceed and learn how it works before uploading an image.")
     
-    # Provide guidance based on the selected transformation
-    if transformation:
+    # Transformation Options
+    transformation = st.selectbox(
+        "Step 1: Select a transformation type:",
+        ["Select a Transformation", "Rotation", "Scaling", "Translation", "Shearing", "Cropping"]
+    )
+    
+    # Ensure the user selects a valid transformation
+    if transformation != "Select a Transformation":
+        # Provide guidance based on the selected transformation
         if transformation == "Rotation":
             st.info("Rotation rotates the image around a fixed point. Angles are applied in steps of 10 degrees.")
         elif transformation == "Scaling":
@@ -1870,8 +1877,8 @@ elif st.session_state.current_page == "image_transformations":
             st.info("Shearing skews the image along the x-axis or y-axis, creating a slanted effect.")
         elif transformation == "Cropping":
             st.info("Cropping trims the image edges step by step to focus on a smaller region.")
-
-    if transformation:
+    
+        # Image Uploader (Only appears after selection)
         uploaded_file = st.file_uploader("Step 2: Now, upload an image", type=["jpg", "jpeg", "png"])
     
         if uploaded_file:
@@ -1907,7 +1914,8 @@ elif st.session_state.current_page == "image_transformations":
                     file_name=f"{transformation}_transformed_images.zip",
                     mime="application/zip"
                 )
-
+    else:
+        st.warning("Please select a valid transformation type to proceed.")
 
     col1, col2 = st.columns(2)
     with col1: