update(8)

src/.streamlit/config.toml

@@ -1,22 +1,15 @@
 [general]
 browser.gatherUsageStats = false
 
-[server]
-headless = true
-port = 8501
-enableCORS = false
-enableXsrfProtection = false
-fileWatcherType = "poll"
-
-[client]
-showErrorDetails = true
-
-[runner]
-fastReruns = true
-
 [logger]
 level = "info"
 
 [cache]
 storage = "filesystem"
-path = "/tmp/streamlit-cache"
+path = "/tmp/streamlit/cache"
+
+[server]
+headless = true
+enableCORS = false
+enableXsrfProtection = false
+fileWatcherType = "poll"

src/app.py

@@ -1,7 +1,19 @@
 import os
 
-os.environ["STREAMLIT_CACHE_DIR"] = "/tmp/streamlit-cache"
-os.makedirs("/tmp/streamlit-cache", exist_ok=True)
+os.environ["STREAMLIT_HOME"] = "/tmp/streamlit"
+os.environ["STREAMLIT_CACHE_DIR"] = "/tmp/streamlit/cache"
+os.environ["STREAMLIT_CONFIG_DIR"] = "/tmp/streamlit"
+os.environ["HF_HOME"] = "/tmp/huggingface"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
+os.environ["SENTENCE_TRANSFORMERS_HOME"] = "/tmp/huggingface/sentence_transformers"
+
+
+os.makedirs("/tmp/streamlit", exist_ok=True)
+os.makedirs("/tmp/huggingface", exist_ok=True)
+os.makedirs("/tmp/streamlit/cache", exist_ok=True)
+
+DATA_DIR = "/tmp/data"
+os.makedirs(DATA_DIR, exist_ok=True)
 import streamlit as st
 
 from services.queryService import QService
@@ -25,11 +37,6 @@ llm = LLMClient().client
 
 @st.cache_resource
 def load_model():
-    os.environ["HF_HOME"] = "/tmp/huggingface"
-    os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
-    os.environ["SENTENCE_TRANSFORMERS_HOME"] = "/tmp/huggingface/sentence_transformers"
-
-    os.makedirs("/tmp/huggingface", exist_ok=True)
     return SentenceTransformer(
         "sentence-transformers/all-MiniLM-L6-v2", cache_folder="/tmp/huggingface"
     )
@@ -39,7 +46,7 @@ embedder = load_model()
 lead_scorer = lead_scoring.LeadScoring(llm, embedder)
 
 st.set_page_config(page_title="Caprae Capital Lead Generation Tool", layout="wide")
-main_lead_info_file = "src/data/all_cleaned_companies.json"
+main_lead_info_file = os.path.join(DATA_DIR, "all_cleaned_companies.json")
 # st.title("Lead Management Dashboard")
 # This is the navigation section
 if "page" not in st.session_state: