commit

app.py

@@ -92,6 +92,14 @@ def keygen(eval_key):
     user_id = eval_key[1]
     return eval_key[0]
 
+def safe_eval(encodings):
+    try:
+        # Replace newline characters and ensure proper commas are added
+        sanitized = encodings.replace('\n', ',').replace(' ', ',')
+        sanitized = sanitized.replace(',,', ',')  # Handle potential double commas
+        return eval(sanitized)
+    except SyntaxError:
+        raise ValueError("Failed to safely parse the encodings string.")
 
 def encode_quantize(test_file, eval_key, encodings):
     ugly = ['Machine', 'SizeOfOptionalHeader', 'Characteristics',
@@ -133,15 +141,11 @@ def encrypt_encoded_quantize(encodings):
     fhe_api = FHEModelClient(f"fhe_model", f".fhe_keys/{eval_key}")
     fhe_api.load()
 
-    # Ensure encodings is a NumPy array
     if isinstance(encodings, str):
-        # If encodings is a string, try to evaluate it to a list or array
-        encodings = eval(encodings)
+        encodings = safe_eval(encodings)
 
-    # Ensure it's converted to a NumPy array with the correct dtype
     encodings = np.array(encodings, dtype=np.float32)
 
-    # Check for NaN or Inf values
     if np.isnan(encodings).any() or np.isinf(encodings).any():
         raise ValueError("Encodings contain NaN or Inf values, which cannot be processed.")