image1copy.py

import cv2 as cv 
import numpy as np
import png
import imageio 
import time
from tqdm import tqdm 
from tkinter.filedialog import askopenfilename
from tkinter.ttk import *
from tkinter import ttk
from tkinter import *
from tkinter import messagebox
import sys


start = time.time()

image_window = Tk()
image_window.geometry ( '355x200' )
image_window.title ('Image Cryptography')

efile = open('efile.pem', 'r')
e = int(efile.read())
efile.close()
 
dfile = open('dfile.pem', 'r')
d = int(dfile.read())
dfile.close()
   
nfile = open('nfile.pem', 'r')
n = int(nfile.read()) 
nfile.close()


def goback():
	exit()

def powmod(b, e, m):
	b2 = b
	res = 1
	while e:
		if e & 1:
			res = (res * b2) % m   
		b2 = (b2*b2) % m        
		e >>= 1                 
	return res

def filenameerror():
	messagebox.showinfo('Error Occured!', 'Please select a 8bit/16bit Image File')

def enfinish():
	messagebox.showinfo('Done!', 'Encryption is finished and saved as EnImage.png')

def definish():
	messagebox.showinfo('Done!', 'Decryption is finished and saved as DeImage.png')

def grey_encrypt():

	try:

			img = cv.imread(askopenfilename(filetypes = [( "PNG files", "*.png" ),( "JPEG files", "*.jpeg" ),( "JPG files", "*.jpg" ),("BMP files","*.BMP")]), 0)
			#print(type(img))
			img = img.astype(np.uint16)
			a,b = img.shape
			print('\n\nOriginal image: ')
			print(img)
			print((a,b))
			tup = a,b

			for i in tqdm(range(0, tup[0])):
				for j in (range(0, tup[1])):
					x = img[i][j] 
					x = powmod(x,e,n)
					img[i][j] = x

			print('\n\nEncrypted Image:\n\n')
			print(img)
			cv.imwrite('Enimage.png', img)
			end = time.time()
			eTime = end - start
			print(eTime)

			enfinish()
			

	except TypeError:
		filenameerror()

	except AttributeError:
		filenameerror()

def rgb_encrypt():
	
	try:
			
			img = cv.imread(askopenfilename(filetypes = [( "PNG files", "*.png" ),( "JPEG files", "*.jpeg" ),( "JPG files", "*.jpg" ),("BMP files","*.BMP")]))
			#print(type(img))
			img = img.astype(np.uint16)
			a = img.shape
			print('\n\nOriginal image: ')
			print(img)
			print(a)
		
			img= img.tolist()
			start = time.time() #TODO remove
			for i in tqdm(range(len(img))):
				for j in (range(len(img[i]))):
					for k in (range(len(img[i][j]))):
						x = img[i][j][k] 
						x = powmod(x,e,n)
						img[i][j][k] = x

			img = np.array(img).astype(np.uint16)
			imageio.imwrite('Enimage.png', img, format='PNG-FI')

			print('\n\nEncrypted Image:\n\n')
			print(img)
			end = time.time()
			eTime = end - start
			print(eTime)

			enfinish()

		
	except TypeError:
		filenameerror()

	except AttributeError:
		filenameerror()
				
			

def grey_decrypt():

	try: 
				
			img1 = imageio.imread(askopenfilenamefiletypes = [( "PNG files", "*.png" )]())
			print('\n\nReading Encrypted Image again:\n\n')
			print(img1)

			#for g in tqdm(range(100)):
			img1= img1.tolist()
			print('Decrypting....')
			for i1 in tqdm(range(len(img1))):
				for j1 in (range(len(img1[i1]))):
					x1 = img1[i1][j1] 
					x1 = powmod(x1,d,n)
					img1[i1][j1] = x1

			img1 = np.array(img1)#.reshape(184,275)
			print('\n\nDecrypted Image:\n\n')
			print(img1)
			#cv.imshow('DeImage', img1)
			cv.imwrite('Deimage.png', img1)

			end = time.time()
			eTime = end - start
			print(eTime)
			
			definish()
	
	except TypeError:
		filenameerror()

	except AttributeError:
		filenameerror()

def rgb_decrypt():

	try:

			img1 = imageio.imread(askopenfilename(filetypes = [( "PNG files", "*.png" )]), format='PNG-FI')
			print('\n\nReading Encrypted Image again:\n\n')
			print(img1)
			img1= img1.tolist()
			print('Decrypting....')

			start = time.time() #TODO remove
			for i1 in tqdm(range(len(img1))):
				for j1 in (range(len(img1[i1]))):
					for k1 in (range(len(img1[i1][j1]))):
						x1 = img1[i1][j1][k1] 
						x1 = powmod(x1,d,n)
						img1[i1][j1][k1] = x1

			img1 = np.array(img1)
			print('\n\nDecrypted Image:\n\n')
			print(img1)
			cv.imwrite('Deimage.png', img1)

			end = time.time()
			eTime = end - start
			print(eTime)
		
			definish()
	
	except TypeError:
		filenameerror()

	except AttributeError:
		filenameerror()

btn_rgb_decrypt = Button( image_window, text = "RGB Image Decryption", command = rgb_decrypt)
btn_rgb_decrypt.place( x=10, y=100 )
btn_grey_decrypt = Button( image_window, text = "Grey Image Decryption", command = grey_decrypt)
btn_grey_decrypt.place( x=180, y=100 )
#btn_decrypt.bind('<Return>', decrypt)
btn_grey_encrypt = Button( image_window, text = "Grey Image Decryption", command = grey_encrypt)
btn_grey_encrypt.place( x=180, y=50 )
btn_rgb_encrypt = Button( image_window, text = "RGB Image Encryption", command = rgb_encrypt)
btn_rgb_encrypt.place( x=10, y=50 )
#btn_encrypt.bind('<Return>', encrypt)
btn_exit = Button( image_window, text = "Go Back" , command = goback).place( x=140, y=150 )
image_window.mainloop()