Update app.py

app.py

@@ -3,13 +3,10 @@ import requests
 import matplotlib.pyplot as plt
 from transformers import pipeline
 import openai
-import pandas as pd
 import os
 
-OPENAI_API_KEY="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"
-
 # OpenAI API 키 설정
-openai.api_key = os.getenv("OPENAI_API_KEY", "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")  # 환경 변수 또는 직접 키 입력
+openai.api_key = "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"  # 환경 변수 또는 직접 키 입력
 
 # 네이버 뉴스 API를 통해 실제 뉴스 기사 가져오기
 def fetch_naver_news(query, display=5):
@@ -41,19 +38,16 @@ def load_sentiment_model():
     classifier = pipeline("text-classification", model="bucketresearch/politicalBiasBERT")
     return classifier
 
-# GPT-4를 이용해 반대 관점 기사 생성
+# GPT-4를 이용해 반대 관점 기사 생성 (최신 OpenAI API 방식)
 def generate_article_gpt4(prompt):
     try:
-        response = openai.ChatCompletion.create(
-            model="gpt-4",
-            messages=[
-                {"role": "system", "content": "You are a helpful assistant that generates articles."},
-                {"role": "user", "content": prompt}
-            ],
+        response = openai.Completion.create(
+            model="gpt-4",  # GPT-4 모델 사용
+            prompt=prompt,
             max_tokens=512,
             temperature=0.7
         )
-        return response.choices[0].message.content
+        return response['choices'][0]['text'].strip()  # GPT의 응답 텍스트
     except Exception as e:
         return f"Error generating text: {e}"
 
@@ -106,7 +100,6 @@ def analyze_news_political_viewpoint(query):
 
     return results, sentiment_counts
 
-
 # 성향 분포 시각화
 def visualize_sentiment_distribution(sentiment_counts):
     fig, ax = plt.subplots()
@@ -142,4 +135,3 @@ if st.button("분석 시작"):
                 st.write("---")
         else:
             st.error("분석된 뉴스 데이터가 없습니다.")
-