Update src/streamlit_app.py

src/streamlit_app.py

@@ -11,7 +11,7 @@ from collections import Counter
 import json
 import os
 from datetime import datetime, timedelta
-import openai  # 구 버전 방식 사용
+from openai import OpenAI  # 새로운 import 방식
 from dotenv import load_dotenv
 import traceback
 import plotly.graph_objects as go
@@ -50,14 +50,12 @@ load_dotenv()  # .env 파일에서 로드 시도
 # 1. 환경 변수에서 API 키 확인
 if os.environ.get('OPENAI_API_KEY'):
     st.session_state.openai_api_key = os.environ.get('OPENAI_API_KEY')
-    openai.api_key = st.session_state.openai_api_key
 
 # 2. Streamlit secrets에서 API 키 확인 (try-except로 오류 방지)
 if not st.session_state.openai_api_key:
     try:
         if 'OPENAI_API_KEY' in st.secrets:
             st.session_state.openai_api_key = st.secrets['OPENAI_API_KEY']
-            openai.api_key = st.session_state.openai_api_key
     except Exception as e:
         pass  # secrets 파일이 없어도 오류 발생하지 않음
 
@@ -91,8 +89,12 @@ with st.sidebar:
     api_key = st.text_input("OpenAI API 키 입력", type="password")
     if api_key:
         st.session_state.openai_api_key = api_key
-        openai.api_key = api_key
-        st.success("API 키가 설정되었습니다!")
+
+# OpenAI 클라이언트 초기화를 위한 함수
+def init_openai_client():
+    if st.session_state.openai_api_key:
+        return OpenAI(api_key=st.session_state.openai_api_key)
+    return None
 
 # 저장된 기사를 불러오는 함수
 def load_saved_articles():
@@ -335,36 +337,39 @@ def analyze_news_content(news_df):
         results['top_keywords'] = []
     return results
 
-# OpenAI API를 이용한 새 기사 생성 (구 버전 방식)
+# OpenAI API를 이용한 새 기사 생성 (새로운 버전 방식)
 def generate_article(original_content, prompt_text):
     try:
         if not st.session_state.openai_api_key:
             return "OpenAI API 키가 설정되지 않았습니다."
             
-        response = openai.ChatCompletion.create(
-            model="gpt-4.1-mini",
+        client = init_openai_client()
+        response = client.chat.completions.create(
+            model="gpt-4",  # 또는 사용 가능한 적절한 모델
             messages=[
                 {"role": "system", "content": "당신은 전문적인 뉴스 기자입니다. 주어진 내용을 바탕으로 새로운 기사를 작성해주세요."},
                 {"role": "user", "content": f"다음 내용을 바탕으로 {prompt_text}\n\n{original_content[:1000]}"}
             ],
             max_tokens=2000
         )
-        return response.choices[0].message['content']
+        return response.choices[0].message.content
     except Exception as e:
         return f"기사 생성 오류: {str(e)}"
 
-# OpenAI API를 이용한 이미지 생성 (구 버전 방식)
+# OpenAI API를 이용한 이미지 생성 (새로운 버전 방식)
 def generate_image(prompt):
     try:
         if not st.session_state.openai_api_key:
             return "OpenAI API 키가 설정되지 않았습니다."
             
-        response = openai.Image.create(
+        client = init_openai_client()
+        response = client.images.generate(
+            model="dall-e-3",  # 또는 "dall-e-2"
             prompt=prompt,
             n=1,
             size="1024x1024"
         )
-        return response['data'][0]['url']
+        return response.data[0].url
     except Exception as e:
         return f"이미지 생성 오류: {str(e)}"
 
@@ -645,9 +650,9 @@ elif menu == "기사 분석하기":
                     if st.session_state.openai_api_key:
                         with st.spinner("기사의 감정을 분석 중입니다..."):
                             try:
-                                # 감정 분석 프롬프트 설정 (구 버전 방식)
-                                response = openai.ChatCompletion.create(
-                                    model="gpt-4.1-mini",
+                                client = init_openai_client()
+                                response = 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}, ...]}"},
                                         {"role": "user", "content": f"다음 뉴스 기사를 분석해 주세요:\n\n제목: {selected_article['title']}\n\n내용: {selected_article['content'][:1500]}"}
@@ -655,8 +660,8 @@ elif menu == "기사 분석하기":
                                     max_tokens=800
                                 )
                                 
-                                # JSON 파싱 (구 버전 방식)
-                                analysis_result = json.loads(response.choices[0].message['content'])
+                                # JSON 파싱
+                                analysis_result = json.loads(response.choices[0].message.content)
                                 
                                 # 결과 시각화
                                 st.subheader("감정 분석 결과")
@@ -885,8 +890,6 @@ elif menu == "새 기사 생성하기":
                 else:
                     st.warning("OpenAI API 키를 사이드바에서 설정해주세요.")
 
-
-
 elif menu == "뉴스 기사 예약하기":
     st.header("뉴스 기사 예약하기")