Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,6 +7,7 @@ from dataclasses import dataclass, asdict
 from typing import List, Optional, Dict, Any
 
 import streamlit as st
+import pandas as pd  # for Excel/CSV import
 
 # ---- Crypto deps (install: pip install cryptography argon2-cffi==23.1.0) ----
 from cryptography.hazmat.primitives.ciphers.aead import AESGCM
@@ -263,6 +264,43 @@ def do_lock():
 
 
 # Sidebar: Create / Open / Lock / Export
+# -------------------------- Import helpers --------------------------
+
+def _read_tabular(file) -> pd.DataFrame:
+    name = (file.name or "").lower()
+    if name.endswith(".csv"):
+        return pd.read_csv(file)
+    # Excel requires openpyxl (installed implicitly by pandas if not, pip install openpyxl)
+    return pd.read_excel(file)
+
+
+def _standardize_columns(df: pd.DataFrame) -> pd.DataFrame:
+    # Map common column names to our schema
+    aliases = {
+        "name": ["name", "title", "site", "account"],
+        "username": ["username", "user", "login", "email"],
+        "url": ["url", "link", "website"],
+        "password": ["password", "pass", "pwd"],
+        "note": ["note", "notes", "remark", "remarks"],
+        "tags": ["tags", "label", "labels", "category", "categories"],
+        "type": ["type", "item_type"],
+    }
+    colmap: Dict[str, str] = {}
+    lower_cols = {c.lower().strip(): c for c in df.columns}
+    for target, alias_list in aliases.items():
+        for a in alias_list:
+            if a in lower_cols:
+                colmap[target] = lower_cols[a]
+                break
+    # Ensure all expected columns exist; create if missing
+    for col in ["name", "username", "url", "password", "note", "tags", "type"]:
+        if col not in colmap:
+            df[col] = ""
+            colmap[col] = col
+    return df[[colmap[c] for c in ["name", "username", "url", "password", "note", "tags", "type"]]]\
+             .rename(columns={colmap[c]: c for c in colmap})
+
+
 with st.sidebar:
     st.title("🔐 SmartPass")
     st.caption("Local, file-based, zero-knowledge vault. No servers.")
@@ -330,15 +368,48 @@ with st.sidebar:
 
     st.divider()
     st.subheader("Export")
-    if st.session_state.vault is not None:
-        export_json = json.dumps(st.session_state.vault, separators=(",", ":"))
-        st.download_button(
-            label="Download vault JSON",
-            data=export_json,
-            file_name="vault.smartpass.json",
-            mime="application/json",
-            use_container_width=True,
-        )
+if st.session_state.vault is not None:
+    export_json = json.dumps(st.session_state.vault, separators=(",", ":"))
+    st.download_button(
+        label="Download vault JSON",
+        data=export_json,
+        file_name="vault.smartpass.json",
+        mime="application/json",
+        use_container_width=True,
+    )
+
+st.divider()
+st.subheader("Import from Excel/CSV")
+imp = st.file_uploader("Upload creds file (.xlsx/.xls/.csv)", type=["xlsx","xls","csv"], accept_multiple_files=False, key="importer")
+if imp is not None and st.session_state.unlocked:
+    try:
+        df = _read_tabular(imp)
+        df = _standardize_columns(df)
+        # Default type to 'login' if empty
+        df['type'] = df['type'].fillna('').astype(str).str.lower().replace({'': 'login'})
+        added = 0
+        items_local: List[VaultItem] = [VaultItem(**it) for it in st.session_state.vault.get("items", [])]
+        for _, row in df.iterrows():
+            payload = {
+                "type": (row.get('type') or 'login') if (row.get('type') in ['login','note']) else 'login',
+                "name": str(row.get('name') or '').strip(),
+                "username": str(row.get('username') or '').strip(),
+                "url": str(row.get('url') or '').strip(),
+                "password": str(row.get('password') or '').strip(),
+                "note": str(row.get('note') or '').strip(),
+                "tags": [t.strip() for t in str(row.get('tags') or '').split(',') if t.strip()],
+            }
+            # Skip empty rows
+            if not any(payload.values()):
+                continue
+            items_local.append(encrypt_item(st.session_state.keys["data_key"], payload))
+            added += 1
+        st.session_state.vault["items"] = [it.to_dict() for it in items_local]
+        re_hmac(st.session_state.vault, st.session_state.keys["mac_key"])
+        st.success(f"Imported {added} item(s).")
+        st.experimental_rerun()
+    except Exception as e:
+        st.error(f"Import failed: {e}")
 
 # -------------------------- Main Area --------------------------
 st.title(APP_TITLE)
@@ -494,4 +565,4 @@ with st.expander("Security Notes & Tips"):
 - **Clipboard caution:** This demo does not auto-copy passwords to avoid OS clipboard risks.
 - **KDF tuning:** If your device is slow, reduce memory/iterations in the sidebar's Advanced KDF.
         """
-    )
+    )