app

app.py

@@ -47,6 +47,8 @@ N_USER_KEY_STORED = 20
 eval_key = []
 #Encodings vector
 encodings = []
+#User ID
+user_id = []
 
 def clean_tmp_directory():
     # Allow 20 user keys to be stored.
@@ -87,6 +89,7 @@ def keygen(eval_key):
     numpy.save(f"tmp/tmp_evaluation_key_{user_id}.npy", evaluation_key)
 
     eval_key = [list(evaluation_key)[:ENCRYPTED_DATA_BROWSER_LIMIT], user_id]
+    user_id = eval_key[1]
 
 
 def encode_quantize(test_file, eval_key, encodings):
@@ -366,12 +369,12 @@ if __name__ == "__main__":
             outputs=[encrypted_file],
         )
 
-        eval_key_part_input = gr.Textbox(value=eval_key[1])
+        user_id_input = gr.Textbox(value=user_id)
         # Button to run FHE-based malware scan
-        b_run_fhe_scan.click(fn=run_fhe, inputs=[eval_key_part_input], outputs=[encrypted_scan_result])
+        b_run_fhe_scan.click(fn=run_fhe, inputs=[user_id_input], outputs=[encrypted_scan_result])
 
         # Button to decrypt the scan result
-        b_decrypt_result.click(fn=decrypt_prediction(), inputs=[eval_key_part_input], outputs=[scan_result])
+        b_decrypt_result.click(fn=decrypt_prediction(), inputs=[user_id_input], outputs=[scan_result])
 
         gr.Markdown(
             "ClairVault is built using advanced Fully Homomorphic Encryption techniques to ensure your data remains private and secure throughout the entire malware scanning process."