commit

app.py

@@ -49,6 +49,8 @@ eval_key = []
 encodings = []
 #User ID
 user_id = []
+#Encoded Vector
+encoded_vector = []
 
 def clean_tmp_directory():
     # Allow 20 user keys to be stored.
@@ -150,24 +152,10 @@ def encode_quantize(test_file, eval_key, encodings):
 
     return encodings
 
-def encrypt_encoded_quantize(encodings):
+def encrypt_encoded_quantize():
     fhe_api = FHEModelClient(f"fhe_model", f".fhe_keys/{eval_key}")
     fhe_api.load()
 
-    encodings = encodings[0]
-
-    if isinstance(encodings, str):
-        encodings = safe_eval(encodings)
-
-    encodings = np.array(encodings, dtype=np.float32)
-
-    # Check if the array is empty
-    if encodings.size == 0:
-        raise ValueError("Encodings array is empty, cannot proceed with encryption.")
-
-    if np.isnan(encodings).any() or np.isinf(encodings).any():
-        raise ValueError("Encodings contain NaN or Inf values, which cannot be processed.")
-
     quantized_encodings = fhe_api.model.quantize_input(encodings).astype(numpy.uint8)
     encrypted_quantized_encoding = fhe_api.quantize_encrypt_serialize(encodings)
 
@@ -403,7 +391,7 @@ if __name__ == "__main__":
         # Button to encrypt file
         b_encrypt_file.click(
             fn=encrypt_encoded_quantize,
-            inputs=[extracted_vector],
+            inputs=[],
             outputs=[encrypted_file],
         )