Update app.py

app.py

@@ -54,10 +54,11 @@ def verify_signature(public_key, data, signature):
 
 public_key = load_public_key_from_file("./nsfwais.pubkey.pem")
 def swap_face(source_file, target_file, doFaceEnhancer, skey):
-
-    skey = json.loads(base64.b64decode(skey).decode())
+    skey = json.loads(skey)
     #first validate skey
-    verify_signature(public_key, skey["t"], skey["s"])
+    signature = base64.b64decode(skey["s"])
+    if not verify_signature(public_key, skey["t"], signature):
+        raise Exception("verify authkey failed.")
     timestamp_requested = int(skey["t"])
     timestamp_now = int(datetime.datetime.utcnow().replace(tzinfo=datetime.timezone.utc).timestamp())
     if timestamp_now - timestamp_requested > 20: