Update app.py

app.py

@@ -1,77 +1,53 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
+import openai
 import os
 from typing import Optional
 
 #############################
-# [기본코드] - Cohere 관련 부분만 남김
-#############################
-
-# Cohere Command R+ 모델 ID 정의
-COHERE_MODEL = "CohereForAI/c4ai-command-r-plus-08-2024"
-
-def get_client(model_name: str):
-    """
-    모델 이름에 맞춰 InferenceClient 생성.
-    토큰은 환경 변수에서 가져옴.
-    """
-    hf_token = os.getenv("HF_TOKEN")
-    if not hf_token:
-        raise ValueError("HuggingFace API 토큰(HF_TOKEN)이 설정되지 않았습니다.")
 
-    if model_name == "Cohere Command R+":
-        model_id = COHERE_MODEL
-    else:
-        raise ValueError("유효하지 않은 모델 이름입니다.")
-    return InferenceClient(model_id, token=hf_token)
+# OpenAI API 클라이언트 설정
+openai.api_key = os.getenv("OPENAI_API_KEY")
+if not openai.api_key:
+    raise ValueError("OpenAI API 토큰(OPENAI_API_KEY)이 설정되지 않았습니다.")
 
-def respond_cohere_qna(
-    question: str,
+def call_openai_api(
+    content: str,
     system_message: str,
     max_tokens: int,
     temperature: float,
     top_p: float
-):
+) -> str:
     """
-    Cohere Command R+ 모델을 이용해 한 번의 질문(question)에 대한 답변을 반환하는 함수.
+    OpenAI의 GPT-4o-mini 모델을 이용해 한 번의 질문(content)에 대한 답변을 반환하는 함수.
     """
-    model_name = "Cohere Command R+"
-    try:
-        client = get_client(model_name)
-    except ValueError as e:
-        return f"오류: {str(e)}"
-
-    messages = [
-        {"role": "system", "content": system_message},
-        {"role": "user", "content": question}
-    ]
-
     try:
-        response_full = client.chat_completion(
-            messages,
+        response = openai.ChatCompletion.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": system_message},
+                {"role": "user", "content": content},
+            ],
             max_tokens=max_tokens,
             temperature=temperature,
             top_p=top_p,
         )
-        assistant_message = response_full.choices[0].message.content
+        assistant_message = response.choices[0].message['content']
         return assistant_message
     except Exception as e:
         return f"오류가 발생했습니다: {str(e)}"
 
 #############################
-# 고급 설정 (Cohere) - 코드에서만 정의 (UI에 노출 금지)
+# 고급 설정 (OpenAI) - 코드에서만 정의 (UI에 노출 금지)
 #############################
 
-COHERE_SYSTEM_MESSAGE = """반드시 한글로 답변할 것.
+OPENAI_SYSTEM_MESSAGE = """반드시 한글로 답변할 것.
 너는 최고의 비서이다.
 내가 요구하는 것들을 최대한 자세하고 정확하게 답변하라.
-
 ##[기본규칙]
 1. 반드시 한국어(한글)로 작성하라.
 2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
 3. 특히 너는 '정보성(Informative)' 전문 블로그 마케팅 전문가이다.
 4. 정보 제공에 초점을 맞추어 작성한다.
-
 ##[텍스트 작성 규칙]
 1. 소주제를 5개로 구분하여 2000자 이상되도록 작성하라.
 2. 전체 맥락을 이해하고 문장의 일관성을 유지하라.
@@ -79,13 +55,11 @@ COHERE_SYSTEM_MESSAGE = """반드시 한글로 답변할 것.
 4. 주제와 상황에 맞는 적절한 어휘를 선택하라.
 5. 한글 어휘의 난이도는 쉽게 작성하라.
 6. 절대 문장의 끝에 '답니다'를 사용하지 말 것.
-
 ###[정보성 블로그 작성 규칙]
 1. 독자가 얻고자 하는 유용한 정보와 흥미로운 정보를 제공하도록 작성하라.
 2. 독자의 공감을 이끌어내고 궁금증을 해결하도록 작성하라.
 3. 독자의 관심사를 충족시키도록 작성하라.
 4. 독자에게 이득이 되는 정보를 작성하라.
-
 ##[제외 규칙]
 1. 반드시 비속어 및 욕설(expletive, abusive language, slang)은 제외하라.
 2. 반드시 참고글의 링크(URL)는 제외하라.
@@ -94,9 +68,9 @@ COHERE_SYSTEM_MESSAGE = """반드시 한글로 답변할 것.
 5. 반드시 문장의 끝부분이 어색한 한국어 표현은 제외하라('예요', '답니다', '해요', '해주죠', '됐죠', '됐어요', '고요' 등.)
 """
 
-COHERE_MAX_TOKENS = 4000
-COHERE_TEMPERATURE = 0.7
-COHERE_TOP_P = 0.95
+OPENAI_MAX_TOKENS = 4000
+OPENAI_TEMPERATURE = 0.7
+OPENAI_TOP_P = 0.95
 
 #############################
 # UI - 블로그 생성기
@@ -116,25 +90,24 @@ with gr.Blocks() as demo:
     ref2 = gr.Textbox(label="참조글 2")
     ref3 = gr.Textbox(label="참조글 3")
 
-    output_box = gr.Textbox(label="결과", lines=8, interactive=False)
+    output_box = gr.Textbox(label="결과", lines=20, interactive=False)
 
-    def generate_blog(tone_value, ref1_value, ref2_value, ref3_value):
-        # 프롬프트: “~~”
+    def generate_blog(tone_value: str, ref1_value: str, ref2_value: str, ref3_value: str) -> str:
+        # 프롬프트 생성
         question = (
-            f"~~\n"
             f"말투: {tone_value}\n"
             f"참조글1: {ref1_value}\n"
             f"참조글2: {ref2_value}\n"
             f"참조글3: {ref3_value}\n"
         )
 
-        # Cohere Command R+ 모델 호출
-        response = respond_cohere_qna(
-            question=question,
-            system_message=COHERE_SYSTEM_MESSAGE,
-            max_tokens=COHERE_MAX_TOKENS,
-            temperature=COHERE_TEMPERATURE,
-            top_p=COHERE_TOP_P
+        # OpenAI GPT-4o-mini 모델 호출
+        response = call_openai_api(
+            content=question,
+            system_message=OPENAI_SYSTEM_MESSAGE,
+            max_tokens=OPENAI_MAX_TOKENS,
+            temperature=OPENAI_TEMPERATURE,
+            top_p=OPENAI_TOP_P
         )
         return response