Add image encryption and decryption functionality

app.py

@@ -0,0 +1,118 @@
+import streamlit as st
+from image_cipher import ImageCipher
+from PIL import Image
+import tempfile
+import os
+
+
+def save_uploaded_file_to_temp(uploaded_file):
+    temp_dir = tempfile.mkdtemp()
+    temp_file_path = os.path.join(temp_dir, uploaded_file.name)
+    with open(temp_file_path, "wb") as f:
+        f.write(uploaded_file.getbuffer())
+    return temp_file_path
+
+
+def convert_to_png(image_path):
+    img = Image.open(image_path)
+    png_image_path = f"{os.path.splitext(image_path)[0]}.png"
+    img.save(png_image_path, "PNG")
+    return png_image_path
+
+
+def encrypt_image(image_path, message, encrypt):
+    cipher = ImageCipher()
+    encoded_image_path = cipher.encode(image_path, message, encrypt=encrypt)
+    return encoded_image_path, cipher.key
+
+
+def decrypt_image(image_path, key):
+    cipher = ImageCipher()
+    decoded_message = cipher.decode(image_path, key)
+    return decoded_message
+
+
+def main():
+    st.sidebar.image("image1.png", use_column_width=True)
+    with open("styles.css") as f:
+        st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
+    st.sidebar.markdown(
+        """<a href='https://github.com/samadpls/ImageCipher'>\
+        <img src='https://img.shields.io/github/stars/samadpls/ImageCipher?\
+            color=red&label=star%20me&logoColor=red&style=social'></a>""",
+        unsafe_allow_html=True,
+    )
+    st.title("Image Cipher")
+    app_mode = st.sidebar.radio(
+        "Choose the app mode", ["Encrypt", "Decrypt", "About Us"]
+    )
+
+    if app_mode == "Encrypt":
+        st.header("Encrypt Image")
+        uploaded_image = st.file_uploader(
+            "Choose an image...", type=["png"]
+        )
+        message = st.text_area("Enter your message")
+        use_encryption = st.checkbox("Use Encryption")
+
+        if st.button("Encrypt"):
+            if uploaded_image and message:
+                temp_image_path = save_uploaded_file_to_temp(uploaded_image)
+                if uploaded_image.type in ["image/jpeg", "image/jpg"]:
+                    temp_image_path = convert_to_png(temp_image_path)
+                encoded_image_path, key = encrypt_image(
+                    temp_image_path, message, use_encryption
+                )
+                if use_encryption:
+                    st.markdown("### Encryption Key:")
+                    st.markdown(f"```\n{key.decode()}\n```")
+                with open(encoded_image_path, "rb") as file:
+                    st.download_button(
+                        "Download Encrypted Image",
+                        file,
+                        file_name="encrypted_image.png",
+                    )
+                st.image(encoded_image_path, caption="Encrypted Image")
+            else:
+                st.error(
+                    "Please upload an image and enter a message\
+                         to encrypt."
+                )
+
+    elif app_mode == "Decrypt":
+        st.header("Decrypt Image")
+        uploaded_image = st.file_uploader(
+            "Choose an image...", type=["png"]
+        )
+        key = st.text_input("Enter key (if any)")
+
+        if st.button("Decrypt"):
+            try:
+                if uploaded_image:
+                    temp_image_path = save_uploaded_file_to_temp(uploaded_image)
+                    if uploaded_image.type in ["image/jpeg", "image/jpg"]:
+                        temp_image_path = convert_to_png(temp_image_path)
+                    decrypted_message = decrypt_image(temp_image_path, key)
+                    st.text_area(
+                        "Decrypted Message",
+                        value=decrypted_message,
+                        height=200
+                    )
+                else:
+                    st.error("Please upload an image to decrypt.")
+            except Exception as e:
+                st.error(e)
+
+    elif app_mode == "About Us":
+        st.header("About Us")
+        st.markdown(
+            "This project is developed by <ul> <li>(Abdul Samad)\
+                    [https://github.com/samadpls/] </li>\
+                     <li> [Maira Usman](https://github.com/Myrausman/)</li> \
+                    </ul>",
+            unsafe_allow_html=True,
+        )
+
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,6 @@
+Pillow
+cryptography
+pytest
+flake8
+streamlit
+imagecipher

styles.css

@@ -0,0 +1,97 @@
+/* Main container */
+body {
+  background-color: #181d27;
+  color: #ffbf00;
+}
+
+/* Sidebar */
+.css-1d391kg {
+  background-color: #4f535a;
+}
+
+.css-1lcbmhc, .css-1d2azem {
+  color: #ffbf00;
+}
+
+.css-1v8qudd {
+  background-color: #2c374d;
+  color: #ffbf00;
+}
+
+/* Header */
+.css-1siy2j7 {
+  color: #ffbf00;
+}
+
+/* Radio buttons and text input */
+.css-1ex6qaa {
+  background-color: #2c374d;
+  color: #ffbf00;
+}
+
+.css-1ex6qaa input[type="radio"] {
+  background-color: #2c374d;
+  color: #ffbf00;
+}
+
+.css-1ex6qaa input[type="text"] {
+  background-color: #2c374d;
+  color: #ffbf00;
+}
+
+.css-1ex6qaa textarea {
+  background-color: #2c374d;
+  color: #ffbf00;
+}
+
+/* Buttons */
+.css-1df45cr, .css-1f1iwnp, .css-1ktajzc {
+  background-color: #152544;
+  color: #ffbf00;
+}
+
+.css-1df45cr:hover, .css-1f1iwnp:hover, .css-1ktajzc:hover {
+  background-color: #152544;
+  color: #ffffff;
+}
+
+/* Download button */
+.css-10trblm {
+  background-color: #152544;
+  color: #ffbf00;
+}
+
+.css-10trblm:hover {
+  background-color: #152544;
+  color: #ffffff;
+}
+
+/* Image */
+.css-11v1u3b img {
+  border: 2px solid #ffbf00;
+}
+
+/* Text */
+.css-qrbaxs p {
+  color: #ffbf00;
+}
+
+/* Error messages */
+.css-19hj99j {
+  background-color: #2c374d;
+  color: #ffbf00;
+}
+
+/* Markdown */
+.css-1uix8eu h1, .css-1uix8eu h2, .css-1uix8eu h3, .css-1uix8eu h4, .css-1uix8eu h5, .css-1uix8eu h6, .css-1uix8eu p {
+  color: #ffbf00;
+}
+
+/* Links */
+a {
+  color: #ffbf00;
+}
+
+a:hover {
+  color: #ffffff;
+}