Update src/streamlit_app.py

src/streamlit_app.py

@@ -41,21 +41,35 @@ class SchedulerState:
 global_scheduler_state = SchedulerState()
 
 # API 키 관리를 위한 세션 상태 초기화
-if 'openai_api_key' not in st.session_state:
-    st.session_state.openai_api_key = None
+if 'openai_client' not in st.session_state:
+    st.session_state.openai_client = None
 
 # 여러 방법으로 API 키 로드 시도
 load_dotenv()  # .env 파일에서 로드 시도
 
+# OpenAI 클라이언트 초기화를 위한 함수
+def init_openai_client(api_key=None):
+    try:
+        if api_key:
+            client = OpenAI(api_key=api_key)
+            # 간단한 API 키 유효성 검사
+            client.models.list()  # API 키가 유효한지 테스트
+            return client
+        return None
+    except Exception as e:
+        st.error(f"API 키 초기화 오류: {str(e)}")
+        return None
+
 # 1. 환경 변수에서 API 키 확인
-if os.environ.get('OPENAI_API_KEY'):
-    st.session_state.openai_api_key = os.environ.get('OPENAI_API_KEY')
+api_key = os.environ.get('OPENAI_API_KEY')
+if api_key:
+    st.session_state.openai_client = init_openai_client(api_key)
 
-# 2. Streamlit secrets에서 API 키 확인 (try-except로 오류 방지)
-if not st.session_state.openai_api_key:
+# 2. Streamlit secrets에서 API 키 확인
+if not st.session_state.openai_client:
     try:
         if 'OPENAI_API_KEY' in st.secrets:
-            st.session_state.openai_api_key = st.secrets['OPENAI_API_KEY']
+            st.session_state.openai_client = init_openai_client(st.secrets['OPENAI_API_KEY'])
     except Exception as e:
         pass  # secrets 파일이 없어도 오류 발생하지 않음
 
@@ -88,13 +102,12 @@ with st.sidebar:
     st.divider()
     api_key = st.text_input("OpenAI API 키 입력", type="password")
     if api_key:
-        st.session_state.openai_api_key = api_key
-
-# OpenAI 클라이언트 초기화를 위한 함수
-def init_openai_client():
-    if st.session_state.openai_api_key:
-        return OpenAI(api_key=st.session_state.openai_api_key)
-    return None
+        client = init_openai_client(api_key)
+        if client:
+            st.session_state.openai_client = client
+            st.success("API 키가 성공적으로 설정되었습니다!")
+        else:
+            st.error("유효하지 않은 API 키입니다.")
 
 # 저장된 기사를 불러오는 함수
 def load_saved_articles():
@@ -340,11 +353,10 @@ def analyze_news_content(news_df):
 # OpenAI API를 이용한 새 기사 생성 (새로운 버전 방식)
 def generate_article(original_content, prompt_text):
     try:
-        if not st.session_state.openai_api_key:
+        if not st.session_state.openai_client:
             return "OpenAI API 키가 설정되지 않았습니다."
             
-        client = init_openai_client()
-        response = client.chat.completions.create(
+        response = st.session_state.openai_client.chat.completions.create(
             model="gpt-4",  # 또는 사용 가능한 적절한 모델
             messages=[
                 {"role": "system", "content": "당신은 전문적인 뉴스 기자입니다. 주어진 내용을 바탕으로 새로운 기사를 작성해주세요."},
@@ -359,11 +371,10 @@ def generate_article(original_content, prompt_text):
 # OpenAI API를 이용한 이미지 생성 (새로운 버전 방식)
 def generate_image(prompt):
     try:
-        if not st.session_state.openai_api_key:
+        if not st.session_state.openai_client:
             return "OpenAI API 키가 설정되지 않았습니다."
             
-        client = init_openai_client()
-        response = client.images.generate(
+        response = st.session_state.openai_client.images.generate(
             model="dall-e-3",  # 또는 "dall-e-2"
             prompt=prompt,
             n=1,
@@ -647,11 +658,10 @@ elif menu == "기사 분석하기":
 
             elif analysis_type == "감정 분석":
                 if st.button("감정 분석하기"):
-                    if st.session_state.openai_api_key:
+                    if st.session_state.openai_client:
                         with st.spinner("기사의 감정을 분석 중입니다..."):
                             try:
-                                client = init_openai_client()
-                                response = client.chat.completions.create(
+                                response = st.session_state.openai_client.chat.completions.create(
                                     model="gpt-4",  # 또는 사용 가능한 적절한 모델
                                     messages=[
                                         {"role": "system", "content": "당신은 텍스트의 감정과 논조를 분석하는 전문가입니다. 다음 뉴스 기사의 감정과 논조를 분석하고, '긍정적', '부정적', '중립적' 중 하나로 분류해 주세요. 또한 기사에서 드러나는 핵심 감정 키워드를 5개 추출하고, 각 키워드별로 1-10 사이의 강도 점수를 매겨주세요. JSON 형식으로 다음과 같이 응답해주세요: {'sentiment': '긍정적/부정적/중립적', 'reason': '이유 설명...', 'keywords': [{'word': '키워드1', 'score': 8}, {'word': '키워드2', 'score': 7}, ...]}"},
@@ -810,9 +820,9 @@ elif menu == "기사 분석하기":
                                 
                             except Exception as e:
                                 st.error(f"감정 분석 오류: {str(e)}")
-                                st.code(traceback.format_exc())
+                                st.error(traceback.format_exc())
                     else:
-                        st.warning("OpenAI API 키가 설정되어 있지 않습니다. 사이드바에서 API 키를 설정해주세요.")
+                        st.warning("OpenAI API 키를 사이드바에서 설정해주세요.")
 
 elif menu == "새 기사 생성하기":
     st.header("새 기사 생성하기")
@@ -847,7 +857,7 @@ elif menu == "새 기사 생성하기":
             generate_image_too = st.checkbox("기사 생성 후 이미지도 함께 생성하기", value=True)
             
             if st.button("새 기사 생성하기"):
-                if st.session_state.openai_api_key:
+                if st.session_state.openai_client:
                     with st.spinner("기사를 생성 중입니다..."):
                         new_article = generate_article(selected_article['content'], prompt_text)