-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.py
47 lines (41 loc) · 1.28 KB
/
server.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
from flask import Flask
from flask import request
import base64
import ImageSteganography as stego
import cipher_key as text_cipher
import PIL
from flask_ngrok import run_with_ngrok
app = Flask(__name__)
run_with_ngrok(app)
img = None
@app.route('/encode', methods = ['POST'])
def encode():
try:
if request.method == 'POST':
text = request.form.get('text')
text_to_encode, key = text_cipher.encode_text(str(text).upper())
print(text_to_encode)
imageStr = stego.encode_image(text_to_encode).decode('utf-8')
with open("file.txt","w") as t:
t.write(imageStr)
return {"key" : key, "image":imageStr}
except:
return "Error",500
@app.route('/decode', methods = ['POST'])
def decode():
try:
try:
if request.method == 'POST':
imageStr = request.form.get('image')
text = stego.decode_image(imageStr)
try:
print(text)
plain_text = text_cipher.originalText(text, request.form.get('key'))
return {"text":plain_text}
except:
return {"text":"Key wrong"}
except:
return {"text":"Key Wrong"}
except:
return "Error",500
app.run()