Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,3 +1,10 @@
+import os
+
+# Set cache dirs to writable local folders
+os.environ["TRANSFORMERS_CACHE"] = "./hf_cache"
+os.environ["HF_HOME"] = "./hf_home"
+os.environ["XDG_CACHE_HOME"] = "./.cache"
+
 import streamlit as st
 import pandas as pd
 import plotly.express as px
@@ -6,15 +13,7 @@ import traceback # Import traceback for detailed error info
 import altair as alt
 # Import functions from main.py
 from main import analyze_text, visualize_results
-import os
-
-# Set huggingface cache to a local path
-os.environ["TRANSFORMERS_CACHE"] = "./hf_cache"
-os.environ["HF_HOME"] = "./hf_home"
-os.environ["XDG_CACHE_HOME"] = "./.cache"
 
-# Optional: prevent streamlit from writing to /
-os.environ["STREAMLIT_HOME"] = "./.streamlit"
 
 st.set_page_config(page_title="Text Analysis Module", layout="centered") # Use wide layout