""",
+ 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", "jpg", "jpeg"]
+ )
+ 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)
+ 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", "jpg", "jpeg"]
+ )
+ key = st.text_input("Enter key (if any)")
+
+ if st.button("Decrypt"):
+ try:
+ if uploaded_image:
+ temp_path = save_uploaded_file_to_temp(uploaded_image)
+ decrypted_message = decrypt_image(temp_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 
- ${data.decoded_message}
`; - }; - reader.readAsDataURL(imageFile); - } else { - alert("Please select an image to decrypt."); - } -} diff --git a/docs/server.py b/docs/server.py deleted file mode 100644 index 7f69ff4..0000000 --- a/docs/server.py +++ /dev/null @@ -1,43 +0,0 @@ -from flask import Flask, request, send_file, jsonify -from image_cipher import ImageCipher -from werkzeug.utils import secure_filename -import os - -app = Flask(__name__) -cipher = ImageCipher() - -UPLOAD_FOLDER = 'uploads' -if not os.path.exists(UPLOAD_FOLDER): - os.makedirs(UPLOAD_FOLDER) - -@app.route('/encode', methods=['POST']) -def encode_image(): - if 'image' not in request.files or 'message' not in request.form: - return jsonify({'error': 'No image or message provided'}), 400 - - image = request.files['image'] - message = request.form['message'] - encrypt = request.form.get('encrypt', 'true').lower() == 'true' - - image_path = os.path.join(UPLOAD_FOLDER, secure_filename(image.filename)) - image.save(image_path) - - encoded_image_path = cipher.encode(image_path, message, encrypt) - return send_file(encoded_image_path, as_attachment=True) - -@app.route('/decode', methods=['POST']) -def decode_image(): - if 'image' not in request.files: - return jsonify({'error': 'No image provided'}), 400 - - image = request.files['image'] - key = request.form.get('key', None) - - image_path = os.path.join(UPLOAD_FOLDER, secure_filename(image.filename)) - image.save(image_path) - - message = cipher.decode(image_path, key) - return jsonify({'message': message}) - -if __name__ == '__main__': - app.run(debug=True) diff --git a/requirements.txt b/requirements.txt index 9434775..18771f6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ Pillow==10.4.0 cryptography==42.0.8 pytest==8.2.2 -flake8==7.1.0 \ No newline at end of file +flake8==7.1.0 +streamlit \ No newline at end of file