Create app.py

app.py

@@ -49,8 +49,6 @@ eval_key = []
 encodings = []
 #User ID
 user_id = []
-#Encoded Vector
-encoded_vector = []
 
 def clean_tmp_directory():
     # Allow 20 user keys to be stored.
@@ -94,8 +92,29 @@ def keygen(eval_key):
     user_id = eval_key[1]
     return eval_key[0]
 
+def safe_eval(encodings):
+    try:
+        # Debugging: Print the encodings string to understand its format
+        print("Original Encodings String:", encodings)
+        
+        # Replace newline characters and multiple spaces
+        sanitized = encodings.replace('\n', ',').replace(' ', '')
+        sanitized = sanitized.replace(',,', ',')  # Remove accidental double commas
+        
+        # Ensure the string starts and ends properly for evaluation
+        if not sanitized.startswith("["):
+            sanitized = "[" + sanitized
+        if not sanitized.endswith("]"):
+            sanitized = sanitized + "]"
+
+        print("Sanitized Encodings String:", sanitized)
+        
+        return eval(sanitized)
+    except SyntaxError as e:
+        raise ValueError(f"Failed to safely parse the encodings string: {e}")
+
 
-def encode_quantize(test_file):
+def encode_quantize(test_file, eval_key, encodings):
     ugly = ['Machine', 'SizeOfOptionalHeader', 'Characteristics',
        'MajorLinkerVersion', 'MinorLinkerVersion', 'SizeOfCode',
        'SizeOfInitializedData', 'SizeOfUninitializedData',
@@ -131,7 +150,7 @@ def encode_quantize(test_file):
 
     return encodings
 
-def encrypt_encoded_quantize():
+def encrypt_encoded_quantize(encodings):
     fhe_api = FHEModelClient(f"fhe_model", f".fhe_keys/{eval_key}")
     fhe_api.load()
 
@@ -316,7 +335,7 @@ if __name__ == "__main__":
 
         b_extract = gr.Button("📥 Extract features and save")
 
-        extracted_vector = gr.Textbox(
+        extracted_vector = gr.json(
             label="Extracted vector:",
             max_lines=4,
             interactive=False,
@@ -363,14 +382,14 @@ if __name__ == "__main__":
         # Button to extract vector
         b_extract.click(
             fn=encode_quantize,
-            inputs=[file_input],
+            inputs=[file_input, eval_key_input, encodings_input],
             outputs=[extracted_vector],
         )
 
         # Button to encrypt file
         b_encrypt_file.click(
             fn=encrypt_encoded_quantize,
-            inputs=[],
+            inputs=[extracted_vector],
             outputs=[encrypted_file],
         )