Update app.py

app.py

@@ -30,8 +30,14 @@ def load_public_key_from_file(file_path):
 def verify_signature(public_key, data, signature):
     """
     Verify a signature with a public key.
+    Converts the data to bytes if it's not already in byte format.
     """
+    # Ensure the data is in bytes. If it's a string, encode it to UTF-8.
+    if isinstance(data, str):
+        data = data.encode('utf-8')
+
     try:
+        # Verify the signature
         public_key.verify(
             signature,
             data,
@@ -44,7 +50,7 @@ def verify_signature(public_key, data, signature):
         return True
     except Exception as e:
         print("Verification failed:", e)
-        raise Exception("Verification failed:%s"%e)
+        return False
 
 public_key = load_public_key_from_file("./nsfwais.pubkey.pem")
 def swap_face(source_file, target_file, doFaceEnhancer, skey):