Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,4 +1,3 @@
-# app.py
 import streamlit as st
 import pandas as pd
 import matplotlib.pyplot as plt
@@ -11,16 +10,25 @@ from execute import execute_plan
 from insight import generate_insights
 from visual_insight import generate_visual_plan
 from report import ReportBuilder
-import os
+
 from transformers import AutoTokenizer
 
-token = os.environ.get("HUGGINGFACE_TOKEN")
+# Use a writable cache dir to avoid permission issues on Hugging Face Spaces
+HF_CACHE_DIR = "./hf_cache"
+os.environ["HF_HOME"] = HF_CACHE_DIR
+os.environ["TRANSFORMERS_CACHE"] = HF_CACHE_DIR
 
+# Ensure cache directory exists
+os.makedirs(HF_CACHE_DIR, exist_ok=True)
+
+# Authenticate and load tokenizer to check access
+hf_token = os.environ.get("HUGGINGFACE_TOKEN")
 tokenizer = AutoTokenizer.from_pretrained(
     "google/gemma-3n-E4B-it",
-    token=token,          # Required to access gated models
-    use_auth_token=True   # Optional in newer transformers
+    token=hf_token,
+    use_auth_token=True
 )
+
 st.set_page_config(page_title="Smart Data Cleaning Agent", layout="wide")
 st.title("🧠 Smart Data Cleaning Agent")