Update app.py

app.py

@@ -66,6 +66,7 @@ def translation(img):
     tm=np.array([[1,0,tx],[0,1,ty]],dtype=np.float32)
     trans_img=cv2.warpAffine(img,tm,dsize=(img.shape[1],img.shape[0]),borderMode=cv2.BORDER_REFLECT)
     img_trans_rgb = cv2.cvtColor(trans_img, cv2.COLOR_BGR2RGB)
+    save_image(img_trans_rgb)
     st.image(img_trans_rgb, caption="Translation Applied", use_container_width=True)
 
 def rotation(img):
@@ -105,6 +106,15 @@ def cropping(img):
     img_crop_rgb=cv2.cvtColor(img_crop,cv2.COLOR_BGR2RGB)
     st.image(img_crop_rgb, caption="Shearing Applied", use_container_width=True)
 
+def save_image(image, folder_name="output_images"):
+    if not os.path.exists(folder_name):
+        os.makedirs(folder_name)
+    
+    image_name = f"image_{len(os.listdir(folder_name)) + 1}.jpg"
+    image_path = os.path.join(folder_name, image_name)
+    cv2.imwrite(image_path, image)
+    st.success(f"Image saved to {image_path}")
+
 
 if uploaded_file is not None:
     # Converting the image into array