Cowls-Community/Image Steganography.py at main · codemuscles/Cowls-Community · GitHub

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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
import cv2
import numpy as np
import pyfiglet

banner = pyfiglet.figlet_format("STEGBYTE")
print(banner,"\t\t\t\tCreated By Soumalya")


def data2binary(data):
    if type(data) == str:
        return ''.join([format(ord(i),"08b") for i in data])
    elif type(data) == bytes or type(data) == np.ndarray:
        return [format(i,"08b") for i in data]


def hideData(image,secret_data):
    secret_data += "#####"

    data_index = 0
    binary_data = data2binary(secret_data)
    data_length = len(binary_data)

    for values in image:
        for pixel in values:

            r,g,b = data2binary(pixel)

            if data_index < data_length:
                pixel[0] = int(r[:-1] + binary_data[data_index])
                data_index += 1
            if data_index < data_length:
                pixel[1] = int(g[:-1] + binary_data[data_index])
                data_index += 1
            if data_index < data_length:
                pixel[2] = int(b[:-1] + binary_data[data_index])
                data_index += 1
            if data_index >= data_length:
                break

    return image


def encode_text():
    image_name = input("Enter Cover Image Name : ")
    image = cv2.imread(image_name)

    data = input("Enter The Text You Want To Hide : ")
    if data == 0:
        raise ValueError("Data is Empty ... ")

    file_name = input("Enter The Encoded Image Name : ")

    encoded_data = hideData(image,data)
    cv2.imwrite(file_name,encoded_data)

def show_data(image):
    binary_data = ""
    for values in image:
        for pixel in values:
            r,g,b = data2binary(pixel)

            binary_data += r[-1]
            binary_data += g[-1]
            binary_data += b[-1]

    all_bytes = [binary_data[i: i+8] for i in range (0,len(binary_data),8)]

    decoded_data = ""
    for byte in all_bytes:
        decoded_data += chr(int(byte,2))
        if decoded_data[-5:] == "#####":
            break

    return decoded_data[:-5]


def decode_text():
    image_name = input("Enter Image You Want To Extract : ")
    image = cv2.imread(image_name)

    text=show_data(image)
    return text

def stegnography():
    userinput = int(input("\nSTEGBYTE\n\n 1. Encode \n 2. Decode \n\n Enter Here : "))
    if userinput == 1:
        encode_text()
    else:
        final_data=decode_text()
        print("\nDecoded Data : ",final_data)

stegnography()