Update app.py

app.py

@@ -2,26 +2,11 @@ import streamlit as st
 import requests
 import matplotlib.pyplot as plt
 from transformers import pipeline
-from openai import OpenAI  # 1.0.0+에서 새로운 클라이언트 방식
+import openai
 import os
 
-# Hugging Face Secrets에서 환경 변수를 가져옵니다.
-# `OPENAI_API_KEY` 환경 변수에 API 키를 설정해야 합니다.
-api_key = os.getenv("OPENAI_API_KEY")  # Secrets에서 API 키 가져오기
-
-# OpenAI 클라이언트 인스턴스 생성
-client = OpenAI(api_key=api_key)  # API 키를 클라이언트에 전달
-
-# 클라이언트를 사용하여 OpenAI API 호출
-def test_openai_api():
-    try:
-        completion = client.completions.create(
-            model="gpt-4",
-            messages=[{"role": "user", "content": "Hello world"}]
-        )
-        return completion.choices[0].text
-    except Exception as e:
-        return f"Error: {e}"
+# OpenAI API 키 설정 (환경변수에서 가져오는 방법)
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
 # 네이버 뉴스 API를 통해 실제 뉴스 기사 가져오기
 def fetch_naver_news(query, display=5):
@@ -53,21 +38,20 @@ def load_sentiment_model():
     classifier = pipeline("text-classification", model="bucketresearch/politicalBiasBERT")
     return classifier
 
-# GPT-4를 이용해 반대 관점 기사 생성
 # GPT-4를 이용해 반대 관점 기사 생성
 def generate_article_gpt4(prompt):
     try:
-        # OpenAI 클라이언트 인스턴스 생성
-        client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-
-        # GPT-4로 반대 관점 기사 생성
-        response = client.completions.create(
-            model="gpt-4",  # 모델 지정
-            prompt=prompt,  # 프롬프트 지정
+        # GPT-4 모델을 이용해 반대 관점 기사를 생성
+        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  # 출력의 창의성 정도
+            temperature=0.7  # 창의성 정도
         )
-        return response.choices[0].text  # GPT의 응답 텍스트
+        return response['choices'][0]['message']['content']  # GPT의 응답 텍스트 반환
     except Exception as e:
         return f"Error generating text: {e}"