app

app.py

@@ -348,27 +348,27 @@ if __name__ == "__main__":
         scan_result = gr.Textbox(label="Scan Result:")
 
         # Button for key generation
-        b_gen_key.click(keygen(eval_key), inputs=[], outputs=[evaluation_key])
+        b_gen_key.click(fn=keygen, inputs=[eval_key], outputs=[evaluation_key])
 
         # Button to extract vector
         b_extract.click(
-            encode_quantize(file_input, eval_key, encodings),
-            inputs=[],
+            fn=encode_quantize,
+            inputs=[file_input, eval_key, encodings],
             outputs=[extracted_vector],
         )
 
         # Button to encrypt file
         b_encrypt_file.click(
-            encrypt_encoded_quantize(encodings),
-            inputs=[],
+            fn=encrypt_encoded_quantize,
+            inputs=[encodings],
             outputs=[encrypted_file],
         )
 
         # Button to run FHE-based malware scan
-        b_run_fhe_scan.click(run_fhe(eval_key[1]), inputs=[], outputs=[encrypted_scan_result])
+        b_run_fhe_scan.click(fn=run_fhe, inputs=[eval_key[1]], outputs=[encrypted_scan_result])
 
         # Button to decrypt the scan result
-        b_decrypt_result.click(decrypt_prediction(eval_key[1]), inputs=[], outputs=[scan_result])
+        b_decrypt_result.click(fn=decrypt_prediction(), inputs=[eval_key[1]], 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."