Update pages/3ML Project Life Cycle.py

pages/3ML Project Life Cycle.py

@@ -1298,9 +1298,72 @@ cv2.destroyAllWindows()
 
     st.markdown("<hr>", unsafe_allow_html=True)
 
+    
+    st.title("Interactive Image Augmentation Tool")
+
+    # Image upload
+    uploaded_file = st.file_uploader("Upload an Image", type=["jpg", "png", "jpeg"])
+    if uploaded_file is not None:
+        # Load the image
+        file_bytes = np.asarray(bytearray(uploaded_file.read()), dtype=np.uint8)
+        img = cv2.imdecode(file_bytes, 1)
+        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+        st.image(img, caption="Uploaded Image", use_column_width=True)
+
+        # Choose operation
+        st.markdown("### Choose Transformation")
+        operation = st.selectbox(
+            "Select the transformation you want to perform:",
+            ["None", "Cropping", "Scaling", "Translation", "Shearing", "Rotation"]
+        )
 
-
-
+        if operation == "Cropping":
+            st.markdown("### Cropping Parameters")
+            x1 = st.number_input("X1 (Start Column)", min_value=0, value=0, step=1)
+            y1 = st.number_input("Y1 (Start Row)", min_value=0, value=0, step=1)
+            x2 = st.number_input("X2 (End Column)", min_value=x1 + 1, value=img.shape[1], step=1)
+            y2 = st.number_input("Y2 (End Row)", min_value=y1 + 1, value=img.shape[0], step=1)
+            if st.button("Apply Cropping"):
+                cropped_img = img[y1:y2, x1:x2]
+                st.image(cropped_img, caption="Cropped Image", use_column_width=True)
+
+        elif operation == "Scaling":
+            st.markdown("### Scaling Parameters")
+            scale_x = st.slider("Scale X", 0.1, 3.0, 1.0, step=0.1)
+            scale_y = st.slider("Scale Y", 0.1, 3.0, 1.0, step=0.1)
+            if st.button("Apply Scaling"):
+                scaled_img = cv2.resize(img, None, fx=scale_x, fy=scale_y, interpolation=cv2.INTER_LINEAR)
+                st.image(scaled_img, caption="Scaled Image", use_column_width=True)
+
+        elif operation == "Translation":
+            st.markdown("### Translation Parameters")
+            tx = st.slider("Shift X (pixels)", -img.shape[1], img.shape[1], 0)
+            ty = st.slider("Shift Y (pixels)", -img.shape[0], img.shape[0], 0)
+            if st.button("Apply Translation"):
+                t_mat = np.float32([[1, 0, tx], [0, 1, ty]])
+                translated_img = cv2.warpAffine(img, t_mat, (img.shape[1], img.shape[0]))
+                st.image(translated_img, caption="Translated Image", use_column_width=True)
+
+        elif operation == "Shearing":
+            st.markdown("### Shearing Parameters")
+            sh_x = st.slider("Shear X", -2.0, 2.0, 0.0, step=0.1)
+            sh_y = st.slider("Shear Y", -2.0, 2.0, 0.0, step=0.1)
+            if st.button("Apply Shearing"):
+                shear_mat = np.float32([[1, sh_x, 0], [sh_y, 1, 0]])
+                sheared_img = cv2.warpAffine(img, shear_mat, (img.shape[1], img.shape[0]))
+                st.image(sheared_img, caption="Sheared Image", use_column_width=True)
+
+        elif operation == "Rotation":
+            st.markdown("### Rotation Parameters")
+            angle = st.slider("Rotation Angle (degrees)", -360, 360, 0)
+            if st.button("Apply Rotation"):
+                center = (img.shape[1] // 2, img.shape[0] // 2)
+                rot_mat = cv2.getRotationMatrix2D(center, angle, 1.0)
+                rotated_img = cv2.warpAffine(img, rot_mat, (img.shape[1], img.shape[0]))
+                st.image(rotated_img, caption="Rotated Image", use_column_width=True)
+
+    else:
+        st.warning("Please upload an image file to continue.")
 
 
 
@@ -1320,11 +1383,6 @@ cv2.destroyAllWindows()
 
 
 
-
-
-
-
-
 #------------------------------------------------------ Projects ------------------------------------------
 elif st.session_state.page == "projects":