Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import requests
 import matplotlib.pyplot as plt
 from transformers import pipeline
-from openai import OpenAI
+import openai
 import os
 
 # OpenAI API 키 설정
@@ -41,15 +41,19 @@ def load_sentiment_model():
 # GPT-4를 이용해 반대 관점 기사 생성 (최신 OpenAI API 방식)
 def generate_article_gpt4(prompt):
     try:
-        client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-        
-        response = client.completions.create(
+        client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+
+        # ChatGPT API 호출
+        response = client.chat.completions.create(
             model="gpt-4",  # GPT-4 모델 사용
-            prompt=prompt,
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant that generates articles."},
+                {"role": "user", "content": prompt}
+            ],
             max_tokens=512,
             temperature=0.7
         )
-        return response.choices[0].text.strip()  # GPT의 응답 텍스트
+        return response['choices'][0]['message']['content']  # GPT의 응답 텍스트
     except Exception as e:
         return f"Error generating text: {e}"