Update Image Augmentation.py

Image Augmentation.py

@@ -84,6 +84,8 @@ if uploaded_file:
     blur = st.checkbox("Blur (5x5 Kernel)")
     translate = st.checkbox("Translate (50 pixels right and down)")
 
+    transformed_image = None  # Initialize transformed_image
+
     if st.button("Apply Transformations"):
         transformed_image = image.copy()
 
@@ -98,14 +100,13 @@ if uploaded_file:
 
         st.image(cv2.cvtColor(transformed_image, cv2.COLOR_BGR2RGB), caption="Transformed Image", use_column_width=True)
 
-     # Provide download options after transformations
+    # Provide download options after transformations
     if transformed_image is not None:
         st.markdown("### Download Transformed Image:")
         download_format = st.radio("Select Download Format:", ["JPG", "PNG", "PDF"], key="download_format")
 
-        if download_format == "JPG":
-            st.markdown(download_image(transformed_image, download_format), unsafe_allow_html=True)
-        elif download_format == "PNG":
+        if download_format == "JPG" or download_format == "PNG":
             st.markdown(download_image(transformed_image, download_format), unsafe_allow_html=True)
         elif download_format == "PDF":
-            st.markdown(download_pdf(transformed_image, download_format), unsafe_allow_html=True)
+            st.markdown(download_pdf(transformed_image), unsafe_allow_html=True)
+