update

app.py

@@ -43,6 +43,8 @@ time.sleep(4)
 ENCRYPTED_DATA_BROWSER_LIMIT = 500
 N_USER_KEY_STORED = 20
 
+#Evaluation Key
+eval_key = []
 
 def clean_tmp_directory():
     # Allow 20 user keys to be stored.
@@ -66,7 +68,7 @@ def clean_tmp_directory():
                 file.unlink()
 
 
-def keygen():
+def keygen(eval_key):
     # Clean tmp directory if needed
     clean_tmp_directory()
 
@@ -82,7 +84,7 @@ def keygen():
 
     numpy.save(f"tmp/tmp_evaluation_key_{user_id}.npy", evaluation_key)
 
-    return [list(evaluation_key)[:ENCRYPTED_DATA_BROWSER_LIMIT], user_id]
+    eval_key = [list(evaluation_key)[:ENCRYPTED_DATA_BROWSER_LIMIT], user_id]
 
 
 def encode_quantize(test_file, eval_key):
@@ -342,15 +344,9 @@ if __name__ == "__main__":
         b_decrypt_result = gr.Button("🔍 Decrypt scan result")
 
         scan_result = gr.Textbox(label="Scan Result:")
-
-        eval_key = keygen()
-        encodings = encode_quantize(file_input, eval_key)
-        encrypted_quantized_encoding = encrypt_encoded_quantize(encodings)
-        encrypted_prediction = run_fhe(eval_key[1])
-        decrypted_encryption = decrypt_prediction(eval_key[1])
-
+        
         # Button for key generation
-        b_gen_key.click(eval_key, inputs=[], outputs=[evaluation_key])
+        b_gen_key.click(keygen, inputs=[eval_key], outputs=[])
 
         # Button to extract vector
         b_extract.click(
@@ -361,16 +357,16 @@ if __name__ == "__main__":
 
         # Button to encrypt file
         b_encrypt_file.click(
-            encrypted_quantized_encoding,
-            inputs=[],
+            encode_quantize,
+            inputs=[file_input, eval_key],
             outputs=[encrypted_file],
         )
 
         # Button to run FHE-based malware scan
-        b_run_fhe_scan.click(encrypted_prediction, inputs=[], outputs=[encrypted_scan_result])
+        b_run_fhe_scan.click(run_fhe, inputs=[eval_key[1]], outputs=[encrypted_scan_result])
 
         # Button to decrypt the scan result
-        b_decrypt_result.click(decrypted_encryption, inputs=[], outputs=[scan_result])
+        b_decrypt_result.click(decrypt_prediction, inputs=[eval_key[1]], 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."