Update app.py

app.py

@@ -59,14 +59,24 @@ crop=st.checkbox("Crop")
    
 btn=st.button("Submit")
 
-def transalation(img):
+def translation(img):
     tx=30 #np.random.randint(2,60)
     ty=45 #np.random.randint(2,55)
     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]))
+    trans_img=cv2.warpAffine(img,tm,dsize=(img.shape[1],img.shape[0]),borderMode=cv2.BORDER_REFLECT)
     img_rgb = cv2.cvtColor(trans_img, cv2.COLOR_BGR2RGB)
     st.image(img_rgb, caption="Translation Applied", use_container_width=True)
 
+def rotation(img):
+    #center = (width//2, height//2)
+    angle=90
+    r_x=img.shape[0]//2
+    r_y=img.shape[1]//2
+    rm=cv2.getRotationMatrix2D((r_y,r_x),angle,1)
+    r_img=cv2.warpAffine(img,rm,(img.shape[1],img.shape[0]))
+    img_rgb = cv2.cvtColor(r_img, cv2.COLOR_BGR2RGB)
+    st.image(img_rgb, caption="Rotation Applied", use_container_width=True)
+
 
 if uploaded_file is not None:
     # Converting the image into array
@@ -77,7 +87,7 @@ if uploaded_file is not None:
     # img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     transformation = []
     if trans:
-        transalation(img)
+        translation(img)
         transformation.append("Translation")
     if rot:
         transformation.append("Rotation")