Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,11 +1,16 @@
+# simple_lookup_persistent.py
 import streamlit as st
 import pandas as pd
+from pathlib import Path
 
 st.set_page_config(page_title="Simple Password Lookup", page_icon="🔎", layout="centered")
 st.title("🔎 Simple Password Lookup")
-st.caption("Upload your Excel/CSV once → search by platform or username → reveal password.")
+st.caption("Upload your Excel/CSV once → it will be saved for future sessions → search to reveal passwords.")
+
+# ---------------- Config ----------------
+# On Hugging Face Spaces, /data is persistent; locally we fall back to ./creds.xlsx
+PERSIST_FILE = Path("/data/creds.xlsx") if Path("/data").exists() else Path("creds.xlsx")
 
-# ---------------- Helpers ----------------
 ALIASES = {
     "name": ["name", "title", "site", "account", "platform", "service", "app"],
     "username": ["username", "user", "login", "email", "userid", "id"],
@@ -15,51 +20,83 @@ ALIASES = {
 }
 EXPECTED = ["name", "username", "url", "password", "note"]
 
-
 def standardize_columns(df: pd.DataFrame) -> pd.DataFrame:
-    # Normalize input headers to lowercase strings
+    """Map common header aliases -> EXPECTED, create missing columns, clean strings."""
     df.columns = [str(c).strip().lower() for c in df.columns]
     colmap = {}
-    # map aliases
     for target, alias_list in ALIASES.items():
         for c in df.columns:
             if c in alias_list:
                 colmap[target] = c
                 break
-    # ensure all expected exist
     for col in EXPECTED:
         if col not in colmap:
             df[col] = ""
             colmap[col] = col
     out = df[[colmap[c] for c in EXPECTED]].rename(columns={colmap[c]: c for c in colmap})
-    # sanitize to strings
     for c in EXPECTED:
         out[c] = out[c].astype(str).fillna("")
-    # drop fully empty rows
-    mask = (out["name"].str.strip() != "") | (out["username"].str.strip() != "") | (out["url"].str.strip() != "") | (out["password"].str.strip() != "")
+    # keep rows with at least one useful field
+    mask = (
+        out["name"].str.strip() != ""
+        | out["username"].str.strip() != ""
+        | out["url"].str.strip() != ""
+        | out["password"].str.strip() != ""
+    )
     return out[mask].reset_index(drop=True)
 
-
 def read_any(file) -> pd.DataFrame:
-    name = (file.name or "").lower()
+    """Read CSV or Excel from an uploaded file-like."""
+    name = (getattr(file, "name", "") or "").lower()
     if name.endswith(".csv"):
         return pd.read_csv(file)
     return pd.read_excel(file)
 
-# ---------------- Upload once ----------------
+# ---------------- Load or Upload once ----------------
 if "creds" not in st.session_state:
-    st.subheader("1) Upload your Excel/CSV")
+    if PERSIST_FILE.exists():
+        try:
+            st.session_state.creds = standardize_columns(pd.read_excel(PERSIST_FILE))
+            st.info(f"Loaded saved credentials from: {PERSIST_FILE}")
+        except Exception as e:
+            st.error(f"Found {PERSIST_FILE} but failed to read it: {e}")
+            st.session_state.creds = None
+    else:
+        st.session_state.creds = None
+
+if st.session_state.creds is None:
+    st.subheader("1) Upload your Excel/CSV (only once)")
     up = st.file_uploader("Choose file", type=["xlsx", "xls", "csv"], accept_multiple_files=False)
-    if up is None:
+    if not up:
         st.stop()
     try:
-        df = read_any(up)
-        df = standardize_columns(df)
+        df = standardize_columns(read_any(up))
         st.session_state.creds = df
-        st.success(f"Loaded {len(df)} entries.")
+        try:
+            PERSIST_FILE.parent.mkdir(parents=True, exist_ok=True)
+            df.to_excel(PERSIST_FILE, index=False)
+            st.success(f"Saved to {PERSIST_FILE}. You won't need to upload again.")
+        except Exception as e:
+            st.warning(f"Loaded for this session, but could not save for persistence: {e}")
     except Exception as e:
         st.error(f"Failed to read file: {e}")
         st.stop()
+else:
+    # Optional: allow replacing the saved file if needed
+    with st.expander("Replace saved file (optional)"):
+        new_up = st.file_uploader("Upload new Excel/CSV to replace saved file", type=["xlsx", "xls", "csv"], key="replacer")
+        if new_up is not None:
+            try:
+                df_new = standardize_columns(read_any(new_up))
+                st.session_state.creds = df_new
+                try:
+                    PERSIST_FILE.parent.mkdir(parents=True, exist_ok=True)
+                    df_new.to_excel(PERSIST_FILE, index=False)
+                    st.success(f"Replaced saved file at {PERSIST_FILE}.")
+                except Exception as e:
+                    st.warning(f"Replaced in memory only, could not save: {e}")
+            except Exception as e:
+                st.error(f"Failed to read new file: {e}")
 
 # ---------------- Search ----------------
 st.subheader("2) Find your password")
@@ -75,25 +112,25 @@ if q.strip():
     )
     results = df[mask]
     if results.empty:
-        st.warning("No matches. Try another keyword or check your file headers.")
+        st.warning("No matches found.")
     else:
         st.caption(f"Matches: {len(results)} (showing up to 50)")
         for idx, row in results.head(50).iterrows():
-            title = row['name'] or row['username'] or row['url']
+            title = row["name"] or row["username"] or row["url"]
             with st.expander(f"{title} — {row['username']} | {row['url']}"):
                 show = st.checkbox("Show password", key=f"show_{idx}")
                 st.text_input("Password", value=row["password"], type=("default" if show else "password"), key=f"pw_{idx}")
-                if str(row.get("note", "")).strip():
-                    st.caption("Note: " + str(row["note"]))
+                if row.get("note", "").strip():
+                    st.caption("Note: " + row["note"])
 else:
     st.caption("Type a keyword above to search.")
 
-# ---------------- Tips ----------------
-with st.expander("Tips"):
+# ---------------- Footer ----------------
+with st.expander("Notes"):
     st.markdown(
         """
-- Required columns (case-insensitive, aliases ok): name, username, url, password. Optional: note.
-- Data is kept only **in memory** for this session. Re-upload if the app restarts.
-- For large files, use CSV for faster upload.
+- Accepted columns (case-insensitive; aliases auto-mapped): **name**, **username**, **url**, **password**, *(optional)* **note**.
+- Your first upload is saved to a persistent file so you **won’t have to re-upload again**.
+- Use the “Replace saved file” section above if you need to update your credentials later.
         """
     )