Update app.py

app.py

@@ -35,14 +35,6 @@ def handle_file_upload(uploaded_file):
     except Exception as e:
         raise ValueError(f"File upload error: {e}")
 
-def create_download_link(data, filename):
-    try:
-        b64 = base64.b64encode(data).decode('utf-8')
-        href = f'<a href="data:application/octet-stream;base64,{b64}" download="{filename}">Download {filename}</a>'
-        return href
-    except Exception as e:
-        raise ValueError(f"Error creating download link: {e}")
-
 st.title("File Encryption and Decryption")
 
 tabs = st.tabs(["Encrypt", "Decrypt"])
@@ -55,9 +47,14 @@ with tabs[0]:
         if uploaded_file and encryption_key:
             try:
                 file_content = handle_file_upload(uploaded_file)
-                encrypted_content = encrypt_content(file_content, encryption_key)
+                encrypted_content = encrypt_content(file_content, encryption_key).encode('utf-8')
                 filename = f"encrypted_{uploaded_file.name}"
-                st.markdown(create_download_link(encrypted_content, filename), unsafe_allow_html=True)
+                st.download_button(
+                    label="Download Encrypted File",
+                    data=encrypted_content,
+                    file_name=filename,
+                    mime="application/octet-stream"
+                )
             except Exception as e:
                 st.error(f"An error occurred: {e}")
         else:
@@ -73,7 +70,12 @@ with tabs[1]:
                 file_content = handle_file_upload(uploaded_file)
                 decrypted_content = decrypt_content(file_content, decryption_key)
                 filename = f"decrypted_{uploaded_file.name}"
-                st.markdown(create_download_link(decrypted_content, filename), unsafe_allow_html=True)
+                st.download_button(
+                    label="Download Decrypted File",
+                    data=decrypted_content,
+                    file_name=filename,
+                    mime="application/octet-stream"
+                )
             except Exception as e:
                 st.error(f"An error occurred: {e}")
         else: