Update app.py

app.py

@@ -59,9 +59,12 @@ scal = st.checkbox("Scaling")
 shear=st.checkbox("Shearing")
 crop=st.checkbox("Crop")
 
+
 transformation=[]   
 btn=st.button("Preview Image")
 
+#sub=st.button("Preview Image")
+
 def clear_folder(folder_name="output_images"):
     if os.path.exists(folder_name):
         for file in os.listdir(folder_name):
@@ -81,11 +84,11 @@ def save_image_and_show(image, folder_name="output_images"):
     # Save the image
     cv2.imwrite(image_path, image)
     
-
-    # Showing the 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")
+    if btn:
+        # Showing the 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")