update

app.py

@@ -1,22 +1,53 @@
 import os
 import gradio as gr
 from huggingface_hub import create_discussion
+from huggingface_hub import hf_hub_download
+from Crypto.PublicKey import RSA
+from Crypto.Random import get_random_bytes
+from Crypto.Cipher import AES, PKCS1_OAEP
+import base64
 
-token = "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"#os.environ["CLAIM"]
+token = os.environ["CLAIM"]
 repo = "Team8/dataset"
 
 # <p style="text-align: center; color: black;"> 🎨 <span style='color: #6d071a;'>FindMy</span>Art</p>
-description = """# <p style="text-align: center; color: black;"> 🎨 <span style='color: #6d071a;'>FindMy</span>Art</p>
+description = """# <p style="text-align: center; color: black;"> 🎨 <span style='color: #6d071a;'>RemoveMy</span>Art</p>
 <span>This is a space to opt-out your images from datasets. After you check that your image is in the dataset, 
 fill an explanation for why you want it to be removed and specify if you want to encrypt the issue that will be opened on the dataset,
  by checking the Encrypt box.</span>"""
 
+
+def encrypt_text(data):
+    loc = hf_hub_download(repo_id="Team8/dataset", repo_type="dataset", filename="receiver.pem")
+    data = data.encode("utf-8")
+    file_out = open("encrypted_data.bin", "wb")
+
+    recipient_key = RSA.import_key(open(loc).read())
+    session_key = get_random_bytes(16)
+
+    # Encrypt the session key with the public RSA key
+    cipher_rsa = PKCS1_OAEP.new(recipient_key)
+    enc_session_key = cipher_rsa.encrypt(session_key)
+
+    # Encrypt the data with the AES session key
+    cipher_aes = AES.new(session_key, AES.MODE_EAX)
+    ciphertext, tag = cipher_aes.encrypt_and_digest(data)
+    [ file_out.write(x) for x in (enc_session_key, cipher_aes.nonce, tag, ciphertext) ]
+    file_out.close()
+    with open("encrypted_data.bin", "rb") as image_file:
+        encoded_string = base64.b64encode(image_file.read())
+    return encoded_string
+
 def open_issue(explanation, email, encrypt):
+    if encrypt:
+        data = explanation + "\n User email: " + email
+        explanation = f"Encrypted\n{encrypt_text(data)}"
+
     create_discussion(
         repo_id=repo,
         repo_type="dataset",
         title="[OPT-OUT REQUEST] Remove image from the dataset",
-        description=explanation + "\n User email: " + email + "\n Encrypt: " + str(encrypt),
+        description=explanation,
         token = token,
     )
     # to do add issue id to the link