Update app.py

app.py

@@ -3,25 +3,16 @@ import openai
 import os
 import requests
 from typing import Optional
-from gradio_client import Client
 
 #############################
-
 # OpenAI API 클라이언트 설정
+#############################
+
 openai.api_key = os.getenv("OPENAI_API_KEY")
 if not openai.api_key:
     raise ValueError("OpenAI API 토큰(OPENAI_API_KEY)이 설정되지 않았습니다.")
 
-def call_openai_api(
-    content: str,
-    system_message: str,
-    max_tokens: int,
-    temperature: float,
-    top_p: float
-) -> str:
-    """
-    OpenAI의 GPT-4o-mini 모델을 이용해 한 번의 질문(content)에 대한 답변을 반환하는 함수.
-    """
+def call_openai_api(content: str, system_message: str, max_tokens: int, temperature: float, top_p: float) -> str:
     try:
         response = openai.ChatCompletion.create(
             model="gpt-4o-mini",
@@ -33,114 +24,69 @@ def call_openai_api(
             temperature=temperature,
             top_p=top_p,
         )
-        assistant_message = response.choices[0].message['content']
-        return assistant_message
+        return response.choices[0].message['content']
     except Exception as e:
-        return f"오류가 발생했습니다: {str(e)}"
+        return f"오류 발생: {str(e)}"
 
 #############################
-# 고급 설정 (OpenAI) - 코드에서만 정의 (UI에 노출 금지)
+# API 요청 함수
 #############################
 
-OPENAI_SYSTEM_MESSAGE = """반드시 한글로 답변할 것.
-너는 최고의 비서이다.
-내가 요구하는 것들을 최대한 자세하고 정확하게 답변하라.
-##[기본규칙]
-1. 반드시 한국어(한글)로 작성하라.
-2. 너는 가장 주목받는 마케터이며 블로그 마케팅 전문가이다.
-3. 특히 너는 '정보성(Informative)' 전문 블로그 마케팅 전문가이다.
-4. 정보 제공에 초점을 맞추어 작성한다.
-##[텍스트 작성 규칙]
-1. 소주제를 5개로 구분하여 2000자 이상되도록 작성하라.
-2. 전체 맥락을 이해하고 문장의 일관성을 유지하라.
-3. 절대로 참고글을 한문장 이상 그대로 출력하지 말 것.
-4. 주제와 상황에 맞는 적절한 어휘를 선택하라.
-5. 한글 어휘의 난이도는 쉽게 작성하라.
-6. 절대 문장의 끝에 '답니다'를 사용하지 말 것.
-###[정보성 블로그 작성 규칙]
-1. 독자가 얻고자 하는 유용한 정보와 흥미로운 정보를 제공하도록 작성하라.
-2. 독자의 공감을 이끌어내고 궁금증을 해결하도록 작성하라.
-3. 독자의 관심사를 충족시키도록 작성하라.
-4. 독자에게 이득이 되는 정보를 작성하라.
-##[제외 규칙]
-1. 반드시 비속어 및 욕설(expletive, abusive language, slang)은 제외하라.
-2. 반드시 참고글의 링크(URL)는 제외하라.
-3. 참고글에서 '링크를 확인해주세요'와 같은 링크 이동의 문구는 제외하라.
-4. 참고글에 있는 작성자, 화자, 유튜버, 기자의 이름, 애칭, 닉네임은 반드시 제외하라.
-5. 반드시 문장의 끝부분이 어색한 한국어 표현은 제외하라('예요', '답니다', '해요', '해주죠', '됐죠', '됐어요', '고요' 등.)
-"""
-
-OPENAI_MAX_TOKENS = 4000
-OPENAI_TEMPERATURE = 0.7
-OPENAI_TOP_P = 0.95
+def fetch_ref(url: str) -> str:
+    if not url:
+        return "URL을 입력해주세요."
+    try:
+        response = requests.post("https://kims12-blog.hf.space/predict", json={"url": url})
+        response.raise_for_status()
+        return response.json()  # JSON 결과 반환
+    except Exception as e:
+        return f"API 호출 오류: {str(e)}"
 
 #############################
-# API 클라이언트 설정 (허깅페이스 스페이스)
+# 블로그 생성기
 #############################
-client = Client("Kims12/blog")
+
+def generate_blog(tone_value: str, ref1_value: str, ref2_value: str, ref3_value: str) -> str:
+    tone_messages = {
+        "친근하게": "친근하고 편안한 대화 스타일로 작성합니다.",
+        "일반적인": "중립적이고 깔끔한 스타일로 작성합니다.",
+        "전문적인": "전문적이고 신뢰감을 주는 스타일로 작성합니다.",
+    }
+    tone_system_message = f"{tone_messages.get(tone_value, '')}\n"
+
+    # OpenAI API 호출
+    prompt = f"말투: {tone_value}\n참조글1: {ref1_value}\n참조글2: {ref2_value}\n참조글3: {ref3_value}\n"
+    return call_openai_api(
+        content=prompt,
+        system_message=tone_system_message,
+        max_tokens=4000,
+        temperature=0.7,
+        top_p=0.95
+    )
 
 #############################
-# UI - 블로그 생성기
+# UI 구성
 #############################
+
 with gr.Blocks() as demo:
     gr.Markdown("# 블로그 생성기")
 
-    # 말투바꾸기 (라디오 버튼)
-    tone_radio = gr.Radio(
-        label="말투바꾸기",
-        choices=["친근하게", "일반적인", "전문적인"],
-        value="일반적인"  # 기본 선택
-    )
-
-    # 참조글 입력 (3개)과 API 실행 버튼
-    with gr.Row():
-        with gr.Column():
-            ref1 = gr.Textbox(label="참조글 1")
-            fetch_button1 = gr.Button("참조글 1 API 실행")
-            fetch_button1.click(
-                fn=lambda url: client.predict(url=url, api_name="/predict") if url else "URL을 입력해주세요.",
-                inputs=ref1,
-                outputs=ref1
-            )
-        with gr.Column():
-            ref2 = gr.Textbox(label="참조글 2")
-            fetch_button2 = gr.Button("참조글 2 API 실행")
-            fetch_button2.click(
-                fn=lambda url: client.predict(url=url, api_name="/predict") if url else "URL을 입력해주세요.",
-                inputs=ref2,
-                outputs=ref2
-            )
-        with gr.Column():
-            ref3 = gr.Textbox(label="참조글 3")
-            fetch_button3 = gr.Button("참조글 3 API 실행")
-            fetch_button3.click(
-                fn=lambda url: client.predict(url=url, api_name="/predict") if url else "URL을 입력해주세요.",
-                inputs=ref3,
-                outputs=ref3
-            )
+    tone_radio = gr.Radio(["친근하게", "일반적인", "전문적인"], label="말투 선택", value="일반적인")
+    ref1 = gr.Textbox(label="참조글 1 URL")
+    ref2 = gr.Textbox(label="참조글 2 URL")
+    ref3 = gr.Textbox(label="참조글 3 URL")
+    output_box = gr.Textbox(label="생성된 블로그", lines=20, interactive=False)
 
-    output_box = gr.Textbox(label="결과", lines=20, interactive=False)
+    # API 호출 버튼
+    fetch_button1 = gr.Button("참조글 1 API 실행")
+    fetch_button2 = gr.Button("참조글 2 API 실행")
+    fetch_button3 = gr.Button("참조글 3 API 실행")
 
-    # 블로그 생성 함수
-    def generate_blog(tone_value: str, ref1_value: str, ref2_value: str, ref3_value: str) -> str:
-        # 프롬프트 생성
-        question = (
-            f"말투: {tone_value}\n"
-            f"참조글1: {ref1_value}\n"
-            f"참조글2: {ref2_value}\n"
-            f"참조글3: {ref3_value}\n"
-        )
-
-        # 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
+    fetch_button1.click(fn=fetch_ref, inputs=ref1, outputs=ref1)
+    fetch_button2.click(fn=fetch_ref, inputs=ref2, outputs=ref2)
+    fetch_button3.click(fn=fetch_ref, inputs=ref3, outputs=ref3)
 
+    # 블로그 생성 버튼
     generate_button = gr.Button("생성하기")
     generate_button.click(
         fn=generate_blog,