Update app.py

app.py

@@ -62,6 +62,15 @@ crop=st.checkbox("Crop")
 transformation=[]   
 btn=st.button("Preview Image")
 
+def clear_folder(folder_name="output_images"):
+    if os.path.exists(folder_name):
+        for file in os.listdir(folder_name):
+            file_path=os.path.join(folder_name,file)
+            if os.path.isfile(file_path):
+                os.remove(file_path)
+    else:
+        os.makedirs(folder_name)
+
 def save_image_and_show(image, folder_name="output_images"):
     if not os.path.exists(folder_name):
         os.makedirs(folder_name)
@@ -73,7 +82,7 @@ def save_image_and_show(image, folder_name="output_images"):
     cv2.imwrite(image_path, image)
     
 
-    # Show success message
+    # 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")
@@ -102,8 +111,8 @@ def zip_file(folder_name="output_images"):
 
 def translation(img,n=5):
     for i in range(n):
-        tx=np.random.randint(2,50)
-        ty=np.random.randint(1,45)
+        tx=np.random.randint(2,45)
+        ty=np.random.randint(1,40)
         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)
@@ -112,7 +121,6 @@ def translation(img,n=5):
 
 def rotation(img,n=5):
     for i in range(n):
-    #center = (width//2, height//2)
         angle=np.random.choice([90, 180, -90])
         scale=np.round(np.random.uniform(1,1.6),2)
         r_x=img.shape[0]//2
@@ -144,8 +152,8 @@ def shearing(img,n=5):
         sy=np.round(np.random.uniform(1,1.2),2)
         ty=np.random.randint(1,20)
         shm=np.array([[sx,shx,tx],[shy,sy,ty]],dtype=np.float32)
-        img_all=cv2.warpAffine(img,shm,(img.shape[1],img.shape[0]),borderMode=cv2.BORDER_REFLECT)
-        save_image_and_show(img_all)
+        img_shear=cv2.warpAffine(img,shm,(img.shape[1],img.shape[0]),borderMode=cv2.BORDER_REFLECT)
+        save_image_and_show(img_shear)
         #img_shear_rgb=cv2.cvtColor(img_all,cv2.COLOR_BGR2RGB)
         #st.image(img_shear_rgb, caption="Shearing Applied", use_container_width=True)
 
@@ -178,6 +186,7 @@ if uploaded_file is not None:
     img=cv2.imdecode(img_array,cv2.IMREAD_COLOR)
     #  # Convert BGR to RGB as streamlit expects RGB colorspace
     # img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    
     transformation = []
 
     if trans:
@@ -193,27 +202,22 @@ if uploaded_file is not None:
 
 
 
+    if transformation:
+        clear_output_folder()
+        per_trans = max(1, round(image_count / len(transformation)))
 
+        if "Translation" in transformation:
+            translation(img, per_trans)
+        if "Rotation" in transformation:
+            rotation(img, per_trans)
+        if "Scaling" in transformation:
+            scaling(img, per_trans)
+        if "Shearing" in transformation:
+            shearing(img, per_trans)
+        if "Crop" in transformation:
+            cropping(img, per_trans)
 
-
-    
-    if "Translation" in transformation and btn:
-        translation(img,round(image_count/len(transformation)))
-        
-    if "Rotation" in transformation and btn:
-        rotation(img,round(image_count/len(transformation)))
-        #transformation.append("Rotation")
-    if "Scaling" in transformation and btn:
-        scaling(img,round(image_count/len(transformation)))
-        #transformation.append("Scaling")
-    if "Shearing" in transformation and btn:
-        shearing(img,round(image_count/len(transformation)))
-        #transformation.append("Shearing")
-    if "Crop" in transformation:
-        cropping(img,round(image_count/len(transformation)))
-        #transformation.append("Crop")
-
-zip_file()
+        zip_file()
 
     #st.download_button("Download File",img_rgb)