Update app.py

app.py

@@ -2,11 +2,12 @@ import streamlit as st
 import requests
 import matplotlib.pyplot as plt
 from transformers import pipeline
-import openai
+from openai import OpenAI  # 1.0.0+에서 새로 변경된 임포트 방식
 import os
 
 # OpenAI API 키 설정
-openai.api_key = os.getenv("OPENAI_API_KEY", "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")  # 환경 변수 또는 직접 키 입력
+# 환경변수에서 API 키를 가져오기 위해 os.getenv 사용
+openai.api_key = os.getenv("OPENAI_API_KEY")  # Hugging Face Secrets에 API 키를 설정한 경우
 
 # 네이버 뉴스 API를 통해 실제 뉴스 기사 가져오기
 def fetch_naver_news(query, display=5):
@@ -38,18 +39,19 @@ def load_sentiment_model():
     classifier = pipeline("text-classification", model="bucketresearch/politicalBiasBERT")
     return classifier
 
-# GPT-4를 이용해 반대 관점 기사 생성 (최신 OpenAI API 방식)
+# GPT-4를 이용해 반대 관점 기사 생성
 def generate_article_gpt4(prompt):
     try:
-        client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+        # OpenAI 클라이언트 인스턴스 생성
+        client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
-        # ChatGPT API 호출
+        # GPT-4로 반대 관점 기사 생성
         response = client.chat.completions.create(
-            model="gpt-4",  # GPT-4 모델 사용
             messages=[
                 {"role": "system", "content": "You are a helpful assistant that generates articles."},
                 {"role": "user", "content": prompt}
             ],
+            model="gpt-4",
             max_tokens=512,
             temperature=0.7
         )