Update utils.py

utils.py

@@ -13,26 +13,27 @@ def to_bin(data):
     else:
         raise TypeError("Type not supported.")
 def decode(image_name,txt=None):
-    msg=""
-    BGRimage = cv2.imread(f"{image_name}")
-    image = cv2.cvtColor(BGRimage, cv2.COLOR_BGR2RGB)   
-    binary_data = ""
-    for row in image:
-        for pixel in row:
-            r, g, b = to_bin(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
-    this = decoded_data[:-5].split("#####",1)[0]
-    this = eval(this)
-    enc_in=this
-    return this,msg
+    try:
+        BGRimage = cv2.imread(f"{image_name}")
+        image = cv2.cvtColor(BGRimage, cv2.COLOR_BGR2RGB)   
+        binary_data = ""
+        for row in image:
+            for pixel in row:
+                r, g, b = to_bin(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
+        this = decoded_data[:-5].split("#####",1)[0]
+        #this = eval(this)
+    except Exception as e:
+        this = e
+    return this
 
 def encode(image_name, secret_data,txt=None):
     msg=""