Update app.py

app.py

@@ -4,9 +4,10 @@ import matplotlib.pyplot as plt
 from transformers import pipeline
 import openai
 import pandas as pd
+import os
 
 # OpenAI API 키 설정
-openai.api_key = "your_openai_api_key"
+openai.api_key = os.getenv("OPENAI_API_KEY", "your_openai_api_key")
 
 # 네이버 뉴스 API를 통해 실제 뉴스 기사 가져오기
 def fetch_naver_news(query, display=5):
@@ -42,7 +43,8 @@ def load_sentiment_model():
 # GPT-4를 이용해 반대 관점 기사 생성
 def generate_article_gpt4(prompt):
     try:
-        response = openai.ChatCompletion.create(
+        client = openai.OpenAI()
+        response = client.chat.completions.create(
             model="gpt-4",
             messages=[
                 {"role": "system", "content": "You are a helpful assistant that generates articles."},
@@ -51,7 +53,7 @@ def generate_article_gpt4(prompt):
             max_tokens=512,
             temperature=0.7
         )
-        return response['choices'][0]['message']['content']
+        return response.choices[0].message.content
     except Exception as e:
         return f"Error generating text: {e}"