Update src/streamlit_app.py

src/streamlit_app.py

@@ -26,8 +26,12 @@ from huggingface_hub import InferenceClient
 # ---------- CONFIGURATION ----------
 st.set_page_config(page_title="Data Analysis App", layout="wide")
 
-# Load HF token
-HF_TOKEN = st.secrets.get("HF_TOKEN", os.getenv("HF_TOKEN"))
+# ✅ Safe HF_TOKEN loader (works locally + on Spaces)
+try:
+    HF_TOKEN = st.secrets["HF_TOKEN"]
+except Exception:
+    HF_TOKEN = os.getenv("HF_TOKEN")
+
 if not HF_TOKEN:
     st.warning("⚠️ HF_TOKEN not found. Please add it to your Hugging Face Space secrets or environment.")
 else:
@@ -229,7 +233,7 @@ if uploaded_file:
         second_col = st.selectbox("Second column", options=[c for c in cleaned_df.columns if c != viz_col])
 
     if st.button("Show Visualization"):
-        fig, ax = plt.subplots(figsize=(8,5))
+        fig, ax = plt.subplots(figsize=(8, 5))
         try:
             if viz_type == 'Histogram':
                 sns.histplot(cleaned_df[viz_col], kde=True, ax=ax)