Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,9 +5,23 @@ import pandas as pd
 import seaborn as sns
 import streamlit as st
 from nltk.tokenize import word_tokenize
+import os
 
-nltk.download("punkt")
-nltk.download("punkt_tab")
+# Create a local nltk_data directory in your working directory
+nltk_data_dir = os.path.join(os.path.dirname(__file__), "nltk_data")
+
+# Ensure the path exists
+os.makedirs(nltk_data_dir, exist_ok=True)
+
+# Add this path to nltk's search path
+nltk.data.path.append(nltk_data_dir)
+
+# Download 'punkt' only if not already downloaded
+try:
+    nltk.data.find("tokenizers/punkt")
+except LookupError:
+    nltk.download("punkt", download_dir=nltk_data_dir)
+    nltk.download("punkt_tab", download_dir=nltk_data_dir)
 
 st.title("📊 Bayesian Token Co-occurrence Simulator")