Update app.py

app.py

@@ -129,7 +129,7 @@ def encode_quantize(test_file, eval_key, encodings):
 
     return encodings
 
-def encrypt_encoded_quantize(encodings):
+def encrypt_encoded_quantize(encodings, user_id, eval_key):
     fhe_api = FHEModelClient(f"fhe_model", f".fhe_keys/{eval_key}")
     fhe_api.load()
     
@@ -138,7 +138,7 @@ def encrypt_encoded_quantize(encodings):
     encrypted_quantized_encoding = fhe_api.quantize_encrypt_serialize(encodings)
 
     numpy.save(
-        f"tmp/tmp_encrypted_quantized_encoding_{eval_key[1]}.npy",
+        f"tmp/tmp_encrypted_quantized_encoding_{user_id}.npy",
         encrypted_quantized_encoding,
     )
 
@@ -349,7 +349,7 @@ if __name__ == "__main__":
             "The encrypted scan result is sent back to the client, who can finally decrypt it with their private key. Only the client is aware of the original file content and the scan result."
         )
         b_decrypt_result = gr.Button("🔍 Decrypt scan result")
-
+        user_id_input = gr.Textbox(value=user_id)
         scan_result = gr.Textbox(label="Scan Result:")
         
         eval_key_input = gr.Textbox(value=eval_key)
@@ -367,11 +367,10 @@ if __name__ == "__main__":
         # Button to encrypt file
         b_encrypt_file.click(
             fn=encrypt_encoded_quantize,
-            inputs=[extracted_vector],
+            inputs=[extracted_vector, user_id_input, eval_key_input],
             outputs=[encrypted_file],
         )
 
-        user_id_input = gr.Textbox(value=user_id)
         # Button to run FHE-based malware scan
         b_run_fhe_scan.click(fn=run_fhe, inputs=[user_id_input], outputs=[encrypted_scan_result])