Flask Web Application
This commit is contained in:
cleontay
parent
6f23a081b8
commit
5796b2b685
|
|
@ -0,0 +1,160 @@
|
||||||
|
import cv2
|
||||||
|
import numpy as np
|
||||||
|
|
||||||
|
|
||||||
|
class img_steg:
|
||||||
|
def __init__(self, image_name: str = "image.png", bit_to_hide: int = 2):
|
||||||
|
"""
|
||||||
|
Initialize the class
|
||||||
|
:param image_name: Name of the image to encode or decode
|
||||||
|
:type image_name: str
|
||||||
|
:param bit_to_hide: Bit to hide the data in (1 - LSB to 8 - MSB)
|
||||||
|
:type bit_to_hide: int
|
||||||
|
"""
|
||||||
|
self.image_name = image_name
|
||||||
|
self.bit_to_hide = 8 - bit_to_hide
|
||||||
|
self.delimiter = "\n"
|
||||||
|
|
||||||
|
def to_bin(self, data: str) -> str | list[str]:
|
||||||
|
"""
|
||||||
|
Convert data to binary format as string
|
||||||
|
Args:
|
||||||
|
data: String
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Binary Data: String | List[String]
|
||||||
|
"""
|
||||||
|
if isinstance(data, str):
|
||||||
|
return ''.join([format(ord(i), "08b") for i in data])
|
||||||
|
elif isinstance(data, bytes) or isinstance(data, np.ndarray):
|
||||||
|
return [format(i, "08b") for i in data]
|
||||||
|
elif isinstance(data, int) or isinstance(data, np.unit8):
|
||||||
|
return format(data, "08b")
|
||||||
|
else:
|
||||||
|
raise TypeError("Type not supported")
|
||||||
|
|
||||||
|
def from_bin(self, data: str) -> str:
|
||||||
|
"""
|
||||||
|
Convert binary `data` back to the original format
|
||||||
|
Args:
|
||||||
|
data: String
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Original Data: String
|
||||||
|
"""
|
||||||
|
return ''.join([chr(int(data[i:i + 8], 2)) for i in range(0, len(data), 8)])
|
||||||
|
|
||||||
|
def decode(self) -> str:
|
||||||
|
"""
|
||||||
|
Decode the data hidden in the image
|
||||||
|
Args:
|
||||||
|
None
|
||||||
|
:return: Decoded Data: String
|
||||||
|
"""
|
||||||
|
print("[+] Decoding...")
|
||||||
|
image = cv2.imread(self.image_name) # read image
|
||||||
|
binary_data = ""
|
||||||
|
bit_to_hide = self.bit_to_hide
|
||||||
|
for row in image:
|
||||||
|
for pixel in row:
|
||||||
|
r, g, b = self.to_bin(pixel)
|
||||||
|
binary_data += r[bit_to_hide]
|
||||||
|
binary_data += g[bit_to_hide]
|
||||||
|
binary_data += b[bit_to_hide]
|
||||||
|
# Split by 8 bits
|
||||||
|
all_bytes = [binary_data[i: i + 8] for i in range(0, len(binary_data), 8)]
|
||||||
|
# Convert from bits to characters
|
||||||
|
decoded_data = self.from_bin(''.join(all_bytes))
|
||||||
|
if decoded_data.endswith(self.delimiter):
|
||||||
|
decoded_data = decoded_data[:-4]
|
||||||
|
|
||||||
|
decoded_data = decoded_data.split(self.delimiter)[0]
|
||||||
|
return decoded_data
|
||||||
|
|
||||||
|
def encode(self, secret_data: str = "Hello World") -> np.ndarray:
|
||||||
|
"""
|
||||||
|
Encode the secret data in the image
|
||||||
|
Args:
|
||||||
|
secret_data: str
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Encoded Image: np.ndarray
|
||||||
|
"""
|
||||||
|
image = cv2.imread(self.image_name) # read image
|
||||||
|
n_bytes = image.shape[0] * image.shape[1] * 3 // 8 # Max bytes to encode
|
||||||
|
print("[*] Maximum bytes to encode:", n_bytes)
|
||||||
|
secret_data += self.delimiter # Add delimiter at the end of data
|
||||||
|
if len(secret_data) > n_bytes:
|
||||||
|
raise ValueError("[!] Insufficient bytes, need a bigger image or less data")
|
||||||
|
print("[*] Encoding Data...")
|
||||||
|
|
||||||
|
# Convert bit to hide position
|
||||||
|
bit_to_hide = self.bit_to_hide
|
||||||
|
data_index = 0
|
||||||
|
binary_secret_data = self.to_bin(secret_data) # Convert data to binary
|
||||||
|
data_len = len(binary_secret_data) # size of data to hide
|
||||||
|
for row in image:
|
||||||
|
for pixel in row:
|
||||||
|
r, g, b = self.to_bin(pixel) # Convert RGB Values to binary format
|
||||||
|
if data_index < data_len:
|
||||||
|
r = list(r)
|
||||||
|
r[bit_to_hide] = binary_secret_data[data_index] # hide data into least significant bit of red pixel
|
||||||
|
pixel[0] = int(''.join(r), 2)
|
||||||
|
data_index += 1
|
||||||
|
if data_index < data_len:
|
||||||
|
g = list(g)
|
||||||
|
g[bit_to_hide] = binary_secret_data[data_index]
|
||||||
|
pixel[1] = int(''.join(g), 2)
|
||||||
|
data_index += 1
|
||||||
|
if data_index < data_len:
|
||||||
|
b = list(b)
|
||||||
|
b[bit_to_hide] = binary_secret_data[data_index]
|
||||||
|
pixel[2] = int(''.join(b), 2)
|
||||||
|
data_index += 1
|
||||||
|
if data_index >= data_len:
|
||||||
|
break
|
||||||
|
return image
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
print("Welcome to Image Steganography")
|
||||||
|
print("1. Encode\n2. Decode\n3. Exit")
|
||||||
|
choice = int(input("Enter your choice: "))
|
||||||
|
if choice == 1:
|
||||||
|
image_name = input("Enter name of image to encode: ")
|
||||||
|
secret_data = input("Enter data to encode: ")
|
||||||
|
bit_to_hide = int(input("Enter bit to hide (1 - LSB to 8 - MSB): "))
|
||||||
|
encoded_image = img_steg(image_name, bit_to_hide).encode(secret_data)
|
||||||
|
extension = image_name.split(".")[-1]
|
||||||
|
cv2.imwrite("encoded_image." + extension, encoded_image)
|
||||||
|
print("Image Encoded Successfully")
|
||||||
|
elif choice == 2:
|
||||||
|
image_name = input("Enter name of image to decode: ")
|
||||||
|
bit_to_hide = int(input("Enter bit to hide (1 - LSB to 8 - MSB): "))
|
||||||
|
decoded_data = img_steg(image_name, bit_to_hide).decode()
|
||||||
|
print("Decoded Data:", decoded_data)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
|
|
||||||
|
##############################
|
||||||
|
# _ _
|
||||||
|
# | | | |___ __ _ __ _ ___
|
||||||
|
# | | | / __|/ _` |/ _` |/ _ \
|
||||||
|
# | |_| \__ \ (_| | (_| | __/
|
||||||
|
# \___/|___/\__,_|\__, |\___|
|
||||||
|
# |___/
|
||||||
|
##############################
|
||||||
|
#
|
||||||
|
# Initialise class with image name and bit to hide
|
||||||
|
#
|
||||||
|
# image_name = image name with extension (String)
|
||||||
|
# bit_to_hide = bit to hide the data in (1 - LSB to 8 - MSB) (int)
|
||||||
|
#
|
||||||
|
# To encode:
|
||||||
|
# encoded_image = img_steg(image_name, bit_to_hide).encode(secret_data)
|
||||||
|
#
|
||||||
|
# To decode:
|
||||||
|
# decoded_data = img_steg(image_name, bit_to_hide).decode()
|
||||||
|
# Both are string types
|
||||||
|
|
@ -0,0 +1,49 @@
|
||||||
|
from flask import Flask, render_template, request, redirect
|
||||||
|
app = Flask(__name__, template_folder='views')
|
||||||
|
|
||||||
|
@app.route('/')
|
||||||
|
def modeSelection():
|
||||||
|
return render_template('mode_selection.html')
|
||||||
|
|
||||||
|
@app.route('/encode')
|
||||||
|
def encode():
|
||||||
|
return render_template('encode.html')
|
||||||
|
|
||||||
|
@app.route("/encoding", methods=['POST'])
|
||||||
|
def encoding():
|
||||||
|
file = request.files['origin']
|
||||||
|
payload = request.form['payload']
|
||||||
|
|
||||||
|
if file.filename != "":
|
||||||
|
file.save('upload/' + file.filename)
|
||||||
|
|
||||||
|
# run the encoding function
|
||||||
|
#return encoded file to the result below
|
||||||
|
return redirect("/encode_result")
|
||||||
|
|
||||||
|
@app.route('/encode_result')
|
||||||
|
def encode_result():
|
||||||
|
return render_template("encode_result.html")
|
||||||
|
|
||||||
|
@app.route('/decode')
|
||||||
|
def decode():
|
||||||
|
return render_template('decode.html')
|
||||||
|
|
||||||
|
@app.route("/decoding", methods=['POST'])
|
||||||
|
def decoding():
|
||||||
|
file = request.files['encoded_file']
|
||||||
|
|
||||||
|
if file.filename != "":
|
||||||
|
file.save('upload/' + file.filename)
|
||||||
|
|
||||||
|
# run the decoding function
|
||||||
|
#return decoded payload to the result below
|
||||||
|
return redirect("/decode_result")
|
||||||
|
|
||||||
|
@app.route('/decode_result')
|
||||||
|
def decode_result():
|
||||||
|
return render_template("decode_result.html")
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
app.debug = True
|
||||||
|
app.run(host="localhost", port=8000)
|
||||||
|
|
@ -0,0 +1,86 @@
|
||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="en">
|
||||||
|
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta http-equiv="X-UA-Compatible" content="IE=edge">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>Decode</title>
|
||||||
|
<script src="https://ajax.googleapis.com/ajax/libs/jquery/3.6.4/jquery.min.js"></script>
|
||||||
|
<style>
|
||||||
|
.drop-zone {
|
||||||
|
background-color: #f8f8f8;
|
||||||
|
border: 2px dashed #cccccc;
|
||||||
|
border-radius: 5px;
|
||||||
|
padding: 20px;
|
||||||
|
text-align: center;
|
||||||
|
cursor: pointer;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drop-zone:hover {
|
||||||
|
border-color: #aaaaaa;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drop-zone input {
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drop-zone .upload-icon {
|
||||||
|
font-size: 50px;
|
||||||
|
margin-bottom: 15px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drop-zone .upload-text {
|
||||||
|
font-size: 16px;
|
||||||
|
color: #888888;
|
||||||
|
}
|
||||||
|
|
||||||
|
.form-container {
|
||||||
|
max-width: 400px;
|
||||||
|
margin: 0 auto;
|
||||||
|
padding: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.form-container input[type="file"],
|
||||||
|
.form-container input[type="submit"] {
|
||||||
|
display: inline-block;
|
||||||
|
margin-bottom: 10px;
|
||||||
|
padding: 10px;
|
||||||
|
border-radius: 5px;
|
||||||
|
border: 1px solid #cccccc;
|
||||||
|
}
|
||||||
|
|
||||||
|
.form-container input[type="submit"] {
|
||||||
|
background-color: #4CAF50;
|
||||||
|
color: white;
|
||||||
|
cursor: pointer;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
|
||||||
|
<body>
|
||||||
|
<div class="form-container">
|
||||||
|
<form action="/decoding" method="POST" enctype = "multipart/form-data">
|
||||||
|
<div class="drop-zone" id="dropZone">
|
||||||
|
<div class="upload-icon">⇩</div>
|
||||||
|
<div class="upload-text">Drag and drop an image here or click to browse</div>
|
||||||
|
<input type="file" name="encoded_file" id="fileInput">
|
||||||
|
</div>
|
||||||
|
<div style="width: 100%; text-align: center; padding: 10px 0px">
|
||||||
|
<input type="submit" value="Decode">
|
||||||
|
</div>
|
||||||
|
</form>
|
||||||
|
</div>
|
||||||
|
</body>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
$(document).ready(function () {
|
||||||
|
$('#dropZone').click(function (event) {
|
||||||
|
if (!$(event.target).is('#fileInput')) {
|
||||||
|
$("#fileInput").click()
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
|
||||||
|
</html>
|
||||||
|
|
@ -0,0 +1,88 @@
|
||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="en">
|
||||||
|
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta http-equiv="X-UA-Compatible" content="IE=edge">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>Encode</title>
|
||||||
|
<script src="https://ajax.googleapis.com/ajax/libs/jquery/3.6.4/jquery.min.js"></script>
|
||||||
|
<style>
|
||||||
|
.drop-zone {
|
||||||
|
background-color: #f8f8f8;
|
||||||
|
border: 2px dashed #cccccc;
|
||||||
|
border-radius: 5px;
|
||||||
|
padding: 20px;
|
||||||
|
text-align: center;
|
||||||
|
cursor: pointer;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drop-zone:hover {
|
||||||
|
border-color: #aaaaaa;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drop-zone input {
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drop-zone .upload-icon {
|
||||||
|
font-size: 50px;
|
||||||
|
margin-bottom: 15px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drop-zone .upload-text {
|
||||||
|
font-size: 16px;
|
||||||
|
color: #888888;
|
||||||
|
}
|
||||||
|
|
||||||
|
.form-container {
|
||||||
|
max-width: 400px;
|
||||||
|
margin: 0 auto;
|
||||||
|
padding: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.form-container input[type="text"],
|
||||||
|
.form-container input[type="file"],
|
||||||
|
.form-container input[type="submit"] {
|
||||||
|
display: inline-block;
|
||||||
|
margin-bottom: 10px;
|
||||||
|
padding: 10px;
|
||||||
|
border-radius: 5px;
|
||||||
|
border: 1px solid #cccccc;
|
||||||
|
}
|
||||||
|
|
||||||
|
.form-container input[type="submit"] {
|
||||||
|
background-color: #4CAF50;
|
||||||
|
color: white;
|
||||||
|
cursor: pointer;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
|
||||||
|
<body>
|
||||||
|
<div class="form-container">
|
||||||
|
<form action="/encoding" method="POST" enctype = "multipart/form-data">
|
||||||
|
<div class="drop-zone" id="dropZone">
|
||||||
|
<div class="upload-icon">⇩</div>
|
||||||
|
<div class="upload-text">Drag and drop an image here or click to browse</div>
|
||||||
|
<input type="file" name="origin" id="fileInput">
|
||||||
|
</div>
|
||||||
|
<div style="width: 100%; text-align: center; padding: 10px 0px">
|
||||||
|
<input type="text" name="payload">
|
||||||
|
<input type="submit" value="Encode">
|
||||||
|
</div>
|
||||||
|
</form>
|
||||||
|
</div>
|
||||||
|
</body>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
$(document).ready(function () {
|
||||||
|
$('#dropZone').click(function (event) {
|
||||||
|
if (!$(event.target).is('#fileInput')) {
|
||||||
|
$("#fileInput").click()
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
|
||||||
|
</html>
|
||||||
|
|
@ -0,0 +1,48 @@
|
||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="en">
|
||||||
|
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta http-equiv="X-UA-Compatible" content="IE=edge">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>Mode Selection</title>
|
||||||
|
<style>
|
||||||
|
* {
|
||||||
|
padding: 0px;
|
||||||
|
margin: 0px
|
||||||
|
}
|
||||||
|
|
||||||
|
h1 {
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
section {
|
||||||
|
width: 200px;
|
||||||
|
height: 100px;
|
||||||
|
margin: 100px auto;
|
||||||
|
border: 1px solid black;
|
||||||
|
padding: 50px;
|
||||||
|
}
|
||||||
|
|
||||||
|
section > div{
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-around;
|
||||||
|
margin-top: 50px;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
|
||||||
|
<body>
|
||||||
|
|
||||||
|
<section>
|
||||||
|
<h1>Mode Selection</h1>
|
||||||
|
|
||||||
|
<div>
|
||||||
|
<a href="/encode">Encode</a>
|
||||||
|
<a href="/decode">Decode</a>
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
</body>
|
||||||
|
|
||||||
|
</html>
|
||||||
Loading…
Reference in New Issue