update

app.py

@@ -15,6 +15,22 @@ import pickle
 import numpy as np
 import pdb
 
+import json
+
+def load_data_from_file(filename):
+    """
+    Load data from a JSON file in the current directory.
+
+    Parameters:
+        filename (str): The name of the file to load.
+
+    Returns:
+        data: The data loaded from the file.
+    """
+    with open(filename, 'r') as file:
+        data = json.load(file)
+    return data
+
 # This repository's directory
 REPO_DIR = Path(__file__).parent
 subprocess.Popen(["uvicorn", "server:app"], cwd=REPO_DIR)
@@ -302,16 +318,22 @@ if __name__ == "__main__":
             interactive=False,
         )
 
+
+        #load 
+        malware_data = load_data_from_file('malware_features.txt')
+        print(data)
+
         gr.Markdown("# 📤 Step 2: Upload a file for scanning")
         gr.Markdown("## Client side")
         gr.Markdown(
             "Upload a file you want to scan for malware. ClairVault will encrypt it locally before sending it to the cloud."
         )
-        file_input = gr.File(label="Upload a file:", file_count = "single", value="./smallexe64.exe") 
+        file_input = gr.File(label="Upload a harmless exe file or use default:", file_count = "single", value="./harmless.exe") 
 
         gr.Markdown("# 📥 Step 3: Extract executable file features")
 
-        b_extract = gr.Button("📥 Extract features and save")
+        b_extract = gr.Button("📥 Extract features from harmless file and save")
+        c_extract = gr.Button("Load features from malware file and save")
 
         extracted_vector = gr.JSON(
             label="Extracted vector:",
@@ -355,12 +377,25 @@ if __name__ == "__main__":
 
         encodings_input = gr.Textbox(value=encodings, visible=False)
         # Button to extract vector
+
+
         b_extract.click(
             fn=encode_quantize,
             inputs=[file_input, eval_key_input, encodings_input],
             outputs=[extracted_vector],
         )
 
+
+
+        
+        c_extract.click(
+            fn=encrypt_encoded_quantize,
+            inputs =[malware_data,  user_id_input, eval_key_input],
+            outputs=[encrypted_file],
+        )
+
+        
+
         # Button to encrypt file
         b_encrypt_file.click(
             fn=encrypt_encoded_quantize,