V0 WIP

app.py

@@ -1,7 +1,38 @@
 import gradio as gr
+import stc
+import numpy as np
+import imageio
+from scipy import signal
+import cv2 
+from PIL import Image
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def HILL(input_image, operation, message, key):
+    input_image.seek(0)
+    buffer = input_image.read()
+    I = cv2.imdecode(np.frombuffer(buffer, np.uint8), 1)
+    I = cv2.cvtColor(I,cv2.COLOR_BGR2GRAY)
+    cv2.imwrite('tmp/file.png',I)
+
+    if operation == 'decode': 
+        stc.extract('tmp/file.png', key, 'tmp/output.txt')
+        return 'tmp/output.txt'
+    
+    else: 
+        H = np.array(
+           [[-1,  2, -1],
+            [ 2, -4,  2],
+            [-1,  2, -1]])
+        L1 = np.ones((3, 3)).astype('float32')/(3**2)
+        L2 = np.ones((15, 15)).astype('float32')/(15**2)
+        costs = signal.convolve2d(I, H, mode='same')  
+        costs = abs(costs)
+        costs = signal.convolve2d(costs, L1, mode='same')  
+        costs = 1/costs
+        costs = signal.convolve2d(costs, L2, mode='same')  
+        costs[costs == np.inf] = 1
+        stc.embed('tmp/file.png', costs, message, key, 'tmp/stego.png')
+        return 'tmp/stego.png'
+
+iface = gr.Interface(HILL, ["file", gr.inputs.Radio(["encode", "decode"]), "text", "text"], "file")
+iface.launch(share=False)

example.py

@@ -1,36 +0,0 @@
-#!/usr/bin/env python3
-
-import stc
-import numpy as np
-import imageio
-from scipy import signal
-
-input_image = 'files/1.pgm'
-
-def HILL(input_image):
-    H = np.array(
-       [[-1,  2, -1],
-        [ 2, -4,  2],
-        [-1,  2, -1]])
-    L1 = np.ones((3, 3)).astype('float32')/(3**2)
-    L2 = np.ones((15, 15)).astype('float32')/(15**2)
-    I = imageio.imread(input_image)
-    costs = signal.convolve2d(I, H, mode='same')  
-    costs = abs(costs)
-    costs = signal.convolve2d(costs, L1, mode='same')  
-    costs = 1/costs
-    costs = signal.convolve2d(costs, L2, mode='same')  
-    costs[costs == np.inf] = 1
-    return costs
-
-costs = HILL(input_image)
-print(costs)
-
-stc.embed(input_image, costs, 'files/message.txt', 's3cr3t', 'files/stego.png')
-stc.extract('files/stego.png', 's3cr3t', 'files/output.txt')
-
-print(open('files/output.txt', 'r').read())
-
-
-
-

requirements.txt

@@ -1 +1,2 @@
 pycryptodome
+imageio

stc.py

@@ -12,10 +12,9 @@ from Crypto.Cipher import AES
 from Crypto.Random import get_random_bytes
 from Crypto.Util.Padding import pad, unpad
 
-def prepare_message(filename, password):
+def prepare_message(text, password):
 
-    f = open(filename, 'r')
-    content_data = f.read().encode('utf-8')
+    content_data = text.encode('utf-8')
 
     # Prepare a header with basic data about the message
     content_ver=struct.pack("B", 1) # version 1
@@ -101,15 +100,15 @@ def embed(input_img_path, cost_matrix,  msg_file_path, password, output_img_path
             if cover[idx]==0:
                 costs[3*idx+0] = INF
                 costs[3*idx+1] = 0
-                costs[3*idx+2] = cost_matrix[i, j]
+                costs[3*idx+2] = cost_matrix[j, i]
             elif cover[idx]==255:
-                costs[3*idx+0] = cost_matrix[i, j]
+                costs[3*idx+0] = cost_matrix[j, i]
                 costs[3*idx+1] = 0 
                 costs[3*idx+2] = INF
             else:
-                costs[3*idx+0] = cost_matrix[i, j]
+                costs[3*idx+0] = cost_matrix[j, i]
                 costs[3*idx+1] = 0
-                costs[3*idx+2] = cost_matrix[i, j]
+                costs[3*idx+2] = cost_matrix[j, i]
             idx += 1
 
     # Prepare message