app

app.py

@@ -346,29 +346,32 @@ if __name__ == "__main__":
         b_decrypt_result = gr.Button("🔍 Decrypt scan result")
 
         scan_result = gr.Textbox(label="Scan Result:")
-
+        
+        eval_key_input = gr.Textbox(value=eval_key)
         # Button for key generation
-        b_gen_key.click(fn=keygen, inputs=[eval_key], outputs=[evaluation_key])
+        b_gen_key.click(fn=keygen, inputs=[eval_key_input], outputs=[evaluation_key])
 
+        encodings_input = gr.Textbox(value=encodings)
         # Button to extract vector
         b_extract.click(
             fn=encode_quantize,
-            inputs=[file_input, eval_key, encodings],
+            inputs=[file_input, eval_key, encodings_input],
             outputs=[extracted_vector],
         )
 
         # Button to encrypt file
         b_encrypt_file.click(
             fn=encrypt_encoded_quantize,
-            inputs=[encodings],
+            inputs=[extracted_vector],
             outputs=[encrypted_file],
         )
 
+        eval_key_part_input = gr.Textbox(value=eval_key[1])
         # Button to run FHE-based malware scan
-        b_run_fhe_scan.click(fn=run_fhe, inputs=[eval_key[1]], outputs=[encrypted_scan_result])
+        b_run_fhe_scan.click(fn=run_fhe, inputs=[eval_key_part_input], outputs=[encrypted_scan_result])
 
         # Button to decrypt the scan result
-        b_decrypt_result.click(fn=decrypt_prediction(), inputs=[eval_key[1]], outputs=[scan_result])
+        b_decrypt_result.click(fn=decrypt_prediction(), inputs=[eval_key_part_input], outputs=[scan_result])
 
         gr.Markdown(
             "ClairVault is built using advanced Fully Homomorphic Encryption techniques to ensure your data remains private and secure throughout the entire malware scanning process."