Update src/streamlit_app.py

src/streamlit_app.py

@@ -24,6 +24,26 @@ TMP_DIR = "/tmp"
 SAVED_ARTICLES_PATH = os.path.join(TMP_DIR, "saved_articles.json")
 SCHEDULED_NEWS_DIR = os.path.join(TMP_DIR, "scheduled_news")
 
+# NLTK 데이터 경로 설정 (현재 디렉토리)
+NLTK_DATA_DIR = "nltk_data"
+
+# NLTK 데이터 경로 추가
+nltk.data.path.insert(0, NLTK_DATA_DIR)
+
+# 필요한 NLTK 데이터 확인
+required_nltk_data = {
+    'punkt': 'tokenizers/punkt',
+    'stopwords': 'corpora/stopwords'
+}
+
+for data_name, data_path in required_nltk_data.items():
+    try:
+        nltk.data.find(data_path)
+    except LookupError:
+        st.error(f"NLTK 데이터 '{data_name}'가 필요합니다. 다음 명령어로 다운로드하세요:")
+        st.code(f"python -c \"import nltk; nltk.download('{data_name}', download_dir='nltk_data')\"")
+        st.stop()
+
 # 워드클라우드 추가
 try:
     from wordcloud import WordCloud
@@ -61,22 +81,6 @@ if st.session_state.openai_api_key is None:
         load_dotenv() # 로컬 .env 파일
         st.session_state.openai_api_key = os.getenv('OPENAI_API_KEY')
 
-# 필요한 NLTK 데이터 다운로드
-try:
-    nltk.data.find('tokenizers/punkt')
-except LookupError:
-    nltk.download('punkt')
-    
-try:
-    nltk.data.find('tokenizers/punkt_tab')
-except LookupError:
-    nltk.download('punkt_tab')
-    
-try:
-    nltk.data.find('corpora/stopwords')
-except LookupError:
-    nltk.download('stopwords')
-
 # OpenAI API 키 설정
 # openai.api_key 설정은 각 API 호출 직전에 st.session_state.openai_api_key 사용하도록 변경하거나,
 # 앱 시작 시점에 한 번 설정합니다. 여기서는 후자를 선택합니다.