Update app.py

app.py

@@ -12,7 +12,6 @@ VAULT_FILE = "passwords.json"
 # --- Encryption Logic ---
 
 def get_cipher(master_password):
-    """Derives a cryptographic key from the master password."""
     salt = b'nutech_hassan_salt_123' 
     kdf = PBKDF2HMAC(
         algorithm=hashes.SHA256(),
@@ -30,7 +29,6 @@ def process_vault(action, master_pwd, service=None, secret=None):
     try:
         cipher = get_cipher(master_pwd)
         
-        # Load the vault
         if os.path.exists(VAULT_FILE):
             with open(VAULT_FILE, "r") as f:
                 vault = json.load(f)
@@ -62,8 +60,6 @@ def process_vault(action, master_pwd, service=None, secret=None):
 custom_css = """
 .gradio-container { max-width: 700px; margin: auto; }
 .header-text { text-align: center; margin-bottom: 20px; }
-.header-text h1 { margin: 0; }
-.header-text span { font-size: 0.8em; color: gray; }
 """
 
 with gr.Blocks() as demo:
@@ -93,10 +89,19 @@ with gr.Blocks() as demo:
 
     status_output = gr.Textbox(label="Vault Status", interactive=False)
 
-    # Event Handlers
-    save_btn.click(process_vault, ["Save", master_key, svc, pwd], status_output)
-    get_btn.click(process_vault, ["Retrieve", master_key, search_svc], status_output)
+    # --- FIXED EVENT HANDLERS ---
+    # We use lambda to pass the "Save" and "Retrieve" strings correctly
+    save_btn.click(
+        fn=lambda m, s, p: process_vault("Save", m, s, p),
+        inputs=[master_key, svc, pwd],
+        outputs=status_output
+    )
+
+    get_btn.click(
+        fn=lambda m, s: process_vault("Retrieve", m, s),
+        inputs=[master_key, search_svc],
+        outputs=status_output
+    )
 
 if __name__ == "__main__":
-    # Moved CSS to launch for Gradio 6
     demo.launch(css=custom_css)