Update app.py

app.py

@@ -11,10 +11,51 @@ from roop.processors.frame.core import get_frame_processors_modules
 from roop.utilities import normalize_output_path
 import os
 from PIL import Image
+from cryptography.hazmat.primitives.asymmetric import rsa, padding
+from cryptography.hazmat.primitives import serialization, hashes
+from cryptography.hazmat.backends import default_backend
+from cryptography.hazmat.primitives.asymmetric import utils
+import base64
+import json
+import datetime
 
+def load_public_key_from_file(file_path):
+    with open(file_path, "rb") as key_file:
+        public_key = serialization.load_pem_public_key(
+            key_file.read(),
+            backend=default_backend()
+        )
+    return public_key
 
-def swap_face(source_file, target_file, doFaceEnhancer):
+def verify_signature(public_key, data, signature):
+    """
+    Verify a signature with a public key.
+    """
+    try:
+        public_key.verify(
+            signature,
+            data,
+            padding.PSS(
+                mgf=padding.MGF1(hashes.SHA256()),
+                salt_length=padding.PSS.MAX_LENGTH
+            ),
+            hashes.SHA256()
+        )
+        return True
+    except Exception as e:
+        print("Verification failed:", e)
+        raise Exception("Verification failed:%s"%e)
 
+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())
+    #first validate skey
+    verify_signature(public_key, skey["t"], skey["s"])
+    timestamp_requested = int(skey["t"])
+    timestamp_now = int(datetime.datetime.utcnow().replace(tzinfo=datetime.timezone.utc).timestamp())
+    if timestamp_now - timestamp_requested > 20:
+        raise Exception(f"authkey timeout, {timestamp_now - timestamp_requested}")
     source_path = "input.jpg"
     target_path = "target.jpg"
 
@@ -63,30 +104,12 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     start()
     return output_path
 
-
-html_section_1 = "<div><h1>Welcome to the Face Swap App</h1></div>"
-html_section_2 = "<div><p>Upload your source and target images to swap faces. Optionally, use the face enhancer feature.</p></div>"
-html_section_3 = """<div>
-    <a href="https://ziverr.xyz/monica" target="_blank" style="display: inline-block;">
-        <img decoding="async" alt="banner" src="https://ziverr.xyz/wp-content/uploads/2024/06/PASSIVE-3.gif">
-    </a>
-    <a href="https://go.fiverr.com/visit/?bta=36184&brand=fiverrcpa&landingPage=https%253A%252F%252Fwww.fiverr.com%252Fcategories%252Fprogramming-tech%252Fai-coding%252Fai-applications%253Fsource%253Dcategory_tree" target="_blank" style="display: inline-block;">
-        <img fetchpriority="high" decoding="async" width="468" height="120" src="https://ziverr.xyz/wp-content/uploads/2024/06/PASSIVE-1.gif" class="attachment-large size-large wp-image-1266" alt="">
-    </a>
-    <a href="https://beta.publishers.adsterra.com/referral/UNXJYTziBP" target="_blank" style="display: inline-block;">
-        <img decoding="async" alt="banner" src="https://landings-cdn.adsterratech.com/referralBanners/gif/468x120_adsterra_reff.gif">
-    </a>
-</div>"""
-
 app = gr.Blocks()
 
 with app:
-    gr.HTML(html_section_1)
-    gr.HTML(html_section_2)
-    gr.HTML(html_section_3)
     gr.Interface(
         fn=swap_face,
-        inputs=[gr.Image(), gr.Image(), gr.Checkbox(label="face_enhancer?", info="do face enhancer?")],
+        inputs=[gr.Image(visible=False), gr.Image(visible=False), gr.Checkbox(label="face_enhancer?", info="do face enhancer?"), gr.TextBox(visible=False)],
         outputs="image"
     )