Update src/streamlit_app.py

src/streamlit_app.py

@@ -8,7 +8,7 @@ st.title("🔎 Simple Password Lookup")
 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
+# Hugging Face Spaces mounts /data as persistent storage; locally fall back to ./creds.xlsx
 PERSIST_FILE = Path("/data/creds.xlsx") if Path("/data").exists() else Path("creds.xlsx")
 
 ALIASES = {
@@ -22,34 +22,38 @@ EXPECTED = ["name", "username", "url", "password", "note"]
 
 def standardize_columns(df: pd.DataFrame) -> pd.DataFrame:
     """Map common header aliases -> EXPECTED, create missing columns, clean strings."""
+    # Normalize headers
     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 columns 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})
+    # Clean to strings
     for c in EXPECTED:
-        out[c] = out[c].astype(str).fillna("")
-    # 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)
+        out[c] = out[c].astype(str).fillna("").replace("nan", "")
+
+    # ✅ Keep rows that have at least one non-empty (after strip) among key fields
+    key_cols = ["name", "username", "url", "password"]
+    mask = out[key_cols].applymap(lambda x: str(x).strip() != "").any(axis=1)
+    out = out[mask].reset_index(drop=True)
+    return out
 
 def read_any(file) -> pd.DataFrame:
-    """Read CSV or Excel from an uploaded file-like."""
+    """Read CSV or Excel from an uploaded file-like object."""
     name = (getattr(file, "name", "") or "").lower()
     if name.endswith(".csv"):
         return pd.read_csv(file)
+    # Default to Excel
     return pd.read_excel(file)
 
 # ---------------- Load or Upload once ----------------
@@ -72,6 +76,7 @@ if st.session_state.creds is None:
     try:
         df = standardize_columns(read_any(up))
         st.session_state.creds = df
+        # Try to persist for future sessions
         try:
             PERSIST_FILE.parent.mkdir(parents=True, exist_ok=True)
             df.to_excel(PERSIST_FILE, index=False)
@@ -82,7 +87,7 @@ if st.session_state.creds is None:
         st.error(f"Failed to read file: {e}")
         st.stop()
 else:
-    # Optional: allow replacing the saved file if needed
+    # Optional: allow replacing the saved file
     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:
@@ -105,10 +110,11 @@ q = st.text_input("Search by platform/site/username", placeholder="e.g., netflix
 if q.strip():
     Q = q.lower().strip()
     df = st.session_state.creds
+    # Robust contains with na=False to avoid NaN issues
     mask = (
-        df["name"].str.lower().str.contains(Q)
-        | df["username"].str.lower().str.contains(Q)
-        | df["url"].str.lower().str.contains(Q)
+        df["name"].str.lower().str.contains(Q, na=False)
+        | df["username"].str.lower().str.contains(Q, na=False)
+        | df["url"].str.lower().str.contains(Q, na=False)
     )
     results = df[mask]
     if results.empty:
@@ -116,12 +122,17 @@ if q.strip():
     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"]).strip()
             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 row.get("note", "").strip():
-                    st.caption("Note: " + row["note"])
+                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"]))
 else:
     st.caption("Type a keyword above to search.")