final fantasy

utils2.py

@@ -81,8 +81,29 @@ def encrypt(text,key, image):
     skips = 0;
     new_pixels = []
     for rgb in pixels:
-        tcr = []
-        for clr in rgb:
+        if type(rgb) == tuple:
+            tcr = []
+            for clr in rgb:
+                if skips == 0:
+                    skips+=iterateSecret(binary_secret)
+                    gidxlog1+=[gidx]
+                    base_log1+=[p_base]
+                    msglog1+=[binary_secret[gidx]]
+                    arr1+=[skips]
+                    if i < len(binary_msg):
+                        if binary_msg[i]=='0':
+                            if clr%2!=0:
+                                clr-=1
+                        else:
+                            if clr%2==0:
+                                clr+=1
+                        i+=1
+                else:
+                    skips-=1;
+                tcr+=[clr]
+            new_pixels += [tuple(tcr)]
+        elif type(rgb) == int:
+            clr = rgb
             if skips == 0:
                 skips+=iterateSecret(binary_secret)
                 gidxlog1+=[gidx]
@@ -91,16 +112,15 @@ def encrypt(text,key, image):
                 arr1+=[skips]
                 if i < len(binary_msg):
                     if binary_msg[i]=='0':
-                       if clr%2!=0:
-                           clr-=1
+                        if clr%2!=0:
+                            clr-=1
                     else:
                         if clr%2==0:
-                           clr+=1
+                            clr+=1
                     i+=1
             else:
                 skips-=1;
-            tcr+=[clr]
-        new_pixels += [tuple(tcr)]
+            new_pixels+=[clr]
     new_image = Image.new(image.mode, image.size)
     new_image.putdata(new_pixels)
     return new_image
@@ -121,7 +141,27 @@ def decrypt(key,image):
     binary_msg = ""
     flag =0
     for rgb in pixels:
-        for clr in rgb:
+        if type(rgb) == tuple:  
+            for clr in rgb:
+                if skips ==0:
+                    cnt = clr%2;
+                    if cnt ==1:
+                        ones+=1;
+                    else:
+                        ones=0; 
+                    binary_msg += str(cnt);
+                    if ones ==8:
+                        flag =1;
+                        break
+                    skips+=iterateSecret(binary_secret)
+                    gidxlog2+=[gidx]
+                    base_log2+=[p_base]
+                    msglog2+=[binary_secret[gidx]]
+                    arr2+=[skips]
+                else:
+                    skips-=1
+        elif type(rgb) == int:
+            clr = rgb
             if skips ==0:
                 cnt = clr%2;
                 if cnt ==1: