diff --git a/Application/lsb_rep.py b/Application/lsb_rep.py index df0fe9c..1c74cca 100644 --- a/Application/lsb_rep.py +++ b/Application/lsb_rep.py @@ -1,5 +1,5 @@ from flask import Flask, render_template, request, redirect, session, send_from_directory -from lib.steganography import img_steg, wav_steg +from lib.steganography import img_steg, wav_steg, txt_steg import cv2 import os import sys @@ -53,6 +53,12 @@ def encoding(): new_wav_file.close() session['wav'] = file.filename session['wav2'] = "encoded_" + file.filename + elif file_extension == ".txt": + encoded_data = txt_steg.txt_steg(WORKING_PATH + file.filename, b2c).encode(payload) + with open(os.path.join(WORKING_PATH, "encoded_" + file.filename), "w") as f: + f.write(encoded_data) + session['txt'] = file.filename + session['txt2'] = "encoded_" + file.filename return redirect("/encode_result") @@ -84,8 +90,12 @@ def decoding(): payload = wav_steg.wav_steg(WORKING_PATH + file.filename, b2c).decode() session["payload"] = payload session["wav"] = file.filename + elif file_extension == ".txt": + payload = txt_steg.txt_steg(WORKING_PATH + file.filename, b2c).decode() + session["payload"] = payload + session["txt"] = file.filename - return redirect("/decode_result") + return redirect("/decode_result") @app.route('/decode_result') def decode_result(): diff --git a/Application/views/decode_result.html b/Application/views/decode_result.html index be29a8c..23b701b 100644 --- a/Application/views/decode_result.html +++ b/Application/views/decode_result.html @@ -43,7 +43,24 @@ $("body").append(html_content) } else if(response.hasOwnProperty("txt")){ - console.log(response) + fetch("upload/" + response["txt"]) + .then(res => res.text()) + .then(data => { + var html_content = `
+
+

Decoded

+ +
+
+

Secret Text:

+

` + response["payload"] + `

+
+
` + $("body").append(html_content) + }) + .catch(error => { + console.error('Error:', error); + }); } }); }) diff --git a/Application/views/encode_result.html b/Application/views/encode_result.html index 0e54869..3f28603 100644 --- a/Application/views/encode_result.html +++ b/Application/views/encode_result.html @@ -49,7 +49,33 @@ $("body").append(html_content) } else if(response.hasOwnProperty("txt")){ - console.log(response) + + fetch("upload/" + response["txt"]) + .then(res => res.text()) + .then(data => { + fetch("upload/" + response["txt2"]) + .then(res2 => res2.text()) + .then(data2 => { + var html_content = `
+
+

Original

+ +
+
+

Encoded

+ +
+ +
` + $("body").append(html_content) + }) + .catch(error2 => { + console.error('Error:', error2); + }); + }) + .catch(error => { + console.error('Error:', error); + }); } }); })