Update app.py

app.py

@@ -41,17 +41,19 @@ def load_sentiment_model():
 # GPT-4를 이용해 반대 관점 기사 생성
 def generate_article_gpt4(prompt):
     try:
-        response = openai.Completion.create(
-            engine="text-davinci-004",  # GPT-4 엔진
-            prompt=prompt,
-            max_tokens=300,
+        response = openai.ChatCompletion.create(
+            model="gpt-4",  # GPT-4 모델 사용
+            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()
+        # 응답에서 텍스트만 추출
+        return response['choices'][0]['message']['content']
     except Exception as e:
-        st.error(f"Error generating text: {e}")
-        return "GPT-4 텍스트 생성 실패"
-
+        return f"Error generating text: {e}"
 
 # 정치 성향 분석
 def analyze_article_sentiment(text, classifier):