Update src/streamlit_app.py

src/streamlit_app.py

@@ -86,15 +86,15 @@ menu = st.sidebar.radio(
 
 # 저장된 기사를 불러오는 함수
 def load_saved_articles():
-    if os.path.exists('saved_articles/articles.json'):
-        with open('saved_articles/articles.json', 'r', encoding='utf-8') as f:
+    if os.path.exists('/tmp/saved_articles/articles.json'):
+        with open('/tmp/saved_articles/articles.json', 'r', encoding='utf-8') as f:
             return json.load(f)
     return []
 
 # 기사를 저장하는 함수
 def save_articles(articles):
-    os.makedirs('saved_articles', exist_ok=True)
-    with open('saved_articles/articles.json', 'w', encoding='utf-8') as f:
+    os.makedirs('/tmp/saved_articles', exist_ok=True)
+    with open('/tmp/saved_articles/articles.json', 'w', encoding='utf-8') as f:
         json.dump(articles, f, ensure_ascii=False, indent=2)
 
 @st.cache_data
@@ -370,9 +370,9 @@ def perform_news_task(task_type, keyword, num_articles, file_prefix):
             time.sleep(0.5)  # 서버 부하 방지
         
         # 결과 저장
-        os.makedirs('scheduled_news', exist_ok=True)
+        os.makedirs('/tmp/scheduled_news', exist_ok=True)
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename = f"scheduled_news/{file_prefix}_{task_type}_{timestamp}.json"
+        filename = f"/tmp/scheduled_news/{file_prefix}_{task_type}_{timestamp}.json"
         
         with open(filename, 'w', encoding='utf-8') as f:
             json.dump(articles, f, ensure_ascii=False, indent=2)
@@ -1095,13 +1095,13 @@ elif menu == "뉴스 기사 예약하기":
                 )
         
         # 수집된 파일 보기
-        if os.path.exists('scheduled_news'):
-            files = [f for f in os.listdir('scheduled_news') if f.endswith('.json')]
+        if os.path.exists('/tmp/scheduled_news'):
+            files = [f for f in os.listdir('/tmp/scheduled_news') if f.endswith('.json')]
             if files:
                 st.subheader("수집된 파일 열기")
                 selected_file = st.selectbox("파일 선택", files, index=len(files)-1)
                 if selected_file and st.button("파일 내용 보기"):
-                    with open(os.path.join('scheduled_news', selected_file), 'r', encoding='utf-8') as f:
+                    with open(os.path.join('/tmp/scheduled_news', selected_file), 'r', encoding='utf-8') as f:
                         articles = json.load(f)
                     
                     st.write(f"**파일명:** {selected_file}")