Update Image Augmentation.py

Image Augmentation.py

@@ -98,12 +98,14 @@ if uploaded_file:
 
         st.image(cv2.cvtColor(transformed_image, cv2.COLOR_BGR2RGB), caption="Transformed Image", use_column_width=True)
 
-    st.markdown("### Download Transformed Image:")
-    download_format = st.radio("Select Download Format:", ["JPG", "PNG", "PDF"])
-
-    if download_format == "JPG":
-        st.markdown(download_image(transformed_image, download_format), unsafe_allow_html=True)
-    elif 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), unsafe_allow_html=True)
+     # 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":
+            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)