Create app.py

app.py

@@ -0,0 +1,103 @@
+import gradio as gr
+import json
+import os
+import base64
+from cryptography.fernet import Fernet
+from cryptography.hazmat.primitives import hashes
+from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
+
+# File where encrypted data is stored
+VAULT_FILE = "passwords.json"
+
+# --- Encryption Logic ---
+
+def get_cipher(master_password):
+    """Derives a cryptographic key from the master password."""
+    # Use a fixed salt for this project; in production, this should be unique.
+    salt = b'nutech_hassan_salt_123' 
+    kdf = PBKDF2HMAC(
+        algorithm=hashes.SHA256(),
+        length=32,
+        salt=salt,
+        iterations=100000,
+    )
+    key = base64.urlsafe_b64encode(kdf.derive(master_password.encode()))
+    return Fernet(key)
+
+def process_vault(action, master_pwd, service=None, secret=None):
+    if not master_pwd:
+        return "⚠️ Please enter your Master Password first!"
+
+    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)
+        else:
+            vault = {}
+
+        if action == "Save":
+            if not service or not secret:
+                return "⚠️ Service name and Password are required!"
+            encrypted_data = cipher.encrypt(secret.encode()).decode()
+            vault[service] = encrypted_data
+            with open(VAULT_FILE, "w") as f:
+                json.dump(vault, f)
+            return f"✅ Successfully encrypted and saved password for {service}."
+
+        elif action == "Retrieve":
+            if service not in vault:
+                return f"❌ No record found for '{service}'."
+            decrypted_data = cipher.decrypt(vault[service].encode()).decode()
+            return f"🔑 Password for {service}: {decrypted_data}"
+
+    except Exception:
+        return "🚫 Access Denied: Incorrect Master Password or corrupted data."
+
+# --- UI Setup ---
+
+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(css=custom_css) as demo:
+    gr.HTML("""
+        <div class="header-text">
+            <h1>🔐 SecureVault Manager</h1>
+            <span>Developed by Hassan Naseer</span>
+        </div>
+    """)
+
+    with gr.Row():
+        master_key = gr.Textbox(
+            label="Master Password", 
+            type="password", 
+            placeholder="Enter your Master Password to unlock the vault",
+            info="This password is used to generate your unique encryption key."
+        )
+
+    with gr.Tabs():
+        with gr.TabItem("➕ Add New Password"):
+            with gr.Column():
+                svc = gr.Textbox(label="Service Name", placeholder="e.g., GitHub, Facebook")
+                pwd = gr.Textbox(label="Password", type="password", placeholder="Enter password to encrypt")
+                save_btn = gr.Button("Encrypt & Save", variant="primary")
+
+        with gr.TabItem("🔍 Retrieve Password"):
+            with gr.Column():
+                search_svc = gr.Textbox(label="Service Name", placeholder="Enter the name of the service")
+                get_btn = gr.Button("Decrypt & Reveal", variant="secondary")
+
+    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)
+
+if __name__ == "__main__":
+    demo.launch()