Update app.py

app.py

@@ -60,6 +60,24 @@ crop=st.checkbox("Crop")
    
 btn=st.button("Preview Image")
 
+def save_image_and_show(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)
+
+    # Save the image
+    cv2.imwrite(image_path, image)
+
+    # Show success message
+    st.success(f"Image saved to {image_path} (local path)")
+
+    # Show the image to the user (not the path)
+    st.image(image, caption="Saved Image", channels="BGR")
+
+
+
 
 def translation(img):
     tx=30 #np.random.randint(2,60)
@@ -68,7 +86,7 @@ def translation(img):
     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)
+    #st.image(img_trans_rgb, caption="Translation Applied", use_container_width=True)
 
 def rotation(img):
     #center = (width//2, height//2)
@@ -117,14 +135,6 @@ def cropping(img):
 #     st.success(f"Image saved to {image_path}")
 
 
-def save_image(image, folder_name="output_images"):
-    image_name = f"image_{len(os.listdir(folder_name)) + 1}.jpg"
-    #image_path = os.path.join(folder_name, image_name)
-    path=f"output_images/{image_name}"
-    cv2.imwrite(path, image)
-    
-    st.success(f"Image saved to {path}")
-
 
 if uploaded_file is not None:
     # Converting the image into array