Update app.py

app.py

@@ -1,8 +1,9 @@
+# app.py
 import gradio as gr
 import openai
 import os
 from typing import Optional
-from gradio_client import Client
+from gradio_client import Client  # gradio_client 추가
 
 #############################
 
@@ -50,7 +51,7 @@ OPENAI_SYSTEM_MESSAGE = """반드시 한글로 답변할 것.
 3. 특히 너는 '정보성(Informative)' 전문 블로그 마케팅 전문가이다.
 4. 정보 제공에 초점을 맞추어 작성한다.
 ##[텍스트 작성 규칙]
-1. 소주제를 5개으로 구분하여 2000자 이상되도록 작성하라.
+1. 소주제를 5개로 구분하여 2000자 이상되도록 작성하라.
 2. 전체 맥락을 이해하고 문장의 일관성을 유지하라.
 3. 절대로 참고글을 한문장 이상 그대로 출력하지 말 것.
 4. 주제와 상황에 맞는 적절한 어휘를 선택하라.
@@ -76,21 +77,17 @@ OPENAI_TOP_P = 0.95
 #############################
 # API 클라이언트 설정
 #############################
-# API 엔드포인트 URL을 환경 변수로 설정하여 보안 유지
-API_ENDPOINT = os.getenv("API_ENDPOINT_URL")
-HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
-
-if not API_ENDPOINT:
-    raise ValueError("API 엔드포인트 URL(API_ENDPOINT_URL)이 설정되지 않았습니다.")
-
-client = Client(API_ENDPOINT, api_key=HUGGINGFACE_TOKEN)
+# 외부 Space의 API 클라이언트 설정
+API_CLIENT = Client("Kims12/blog")
 
 def fetch_blog_content(url: str) -> str:
     """
-    주어진 URL을 사용하여 블로그 본문을 가져오는 API 호출 함수.
+    주어진 URL을 이용해 블로그 본문을 가져오는 함수.
     """
+    if not url.strip():
+        return "URL을 입력해주세요."
     try:
-        result = client.predict(
+        result = API_CLIENT.predict(
             url=url,
             api_name="/predict"
         )
@@ -111,19 +108,16 @@ with gr.Blocks() as demo:
         value="일반적인"  # 기본 선택
     )
 
+    # 참조글 입력 (3개) 및 API 실행 버튼
+    with gr.Row():
+        ref1 = gr.Textbox(label="참조글 1 (URL 입력)")
+        fetch1 = gr.Button("참조글1 API 실행")
+    with gr.Row():
+        ref2 = gr.Textbox(label="참조글 2 (URL 입력)")
+        fetch2 = gr.Button("참조글2 API 실행")
     with gr.Row():
-        # 참조글 1
-        with gr.Column():
-            ref1 = gr.Textbox(label="참조글 1 URL", placeholder="네이버 블로그 URL 입력")
-            fetch_ref1 = gr.Button("참조글1 가져오기")
-        # 참조글 2
-        with gr.Column():
-            ref2 = gr.Textbox(label="참조글 2 URL", placeholder="네이버 블로그 URL 입력")
-            fetch_ref2 = gr.Button("참조글2 가져오기")
-        # 참조글 3
-        with gr.Column():
-            ref3 = gr.Textbox(label="참조글 3 URL", placeholder="네이버 블로그 URL 입력")
-            fetch_ref3 = gr.Button("참조글3 가져오기")
+        ref3 = gr.Textbox(label="참조글 3 (URL 입력)")
+        fetch3 = gr.Button("참조글3 API 실행")
 
     output_box = gr.Textbox(label="결과", lines=20, interactive=False)
 
@@ -146,12 +140,6 @@ with gr.Blocks() as demo:
         )
         return response
 
-    def fetch_ref(ref_url: str) -> str:
-        if ref_url.strip() == "":
-            return "URL을 입력해주세요."
-        return fetch_blog_content(ref_url)
-
-    # 생성하기 버튼
     generate_button = gr.Button("생성하기")
     generate_button.click(
         fn=generate_blog,
@@ -159,23 +147,32 @@ with gr.Blocks() as demo:
         outputs=output_box
     )
 
-    # 참조글1 가져오기 버튼
-    fetch_ref1.click(
-        fn=fetch_ref,
+    # 참조글1 API 실행
+    def fetch_ref1(url: str) -> str:
+        return fetch_blog_content(url)
+
+    fetch1.click(
+        fn=fetch_ref1,
         inputs=ref1,
         outputs=ref1
     )
 
-    # 참조글2 가져오기 버튼
-    fetch_ref2.click(
-        fn=fetch_ref,
+    # 참조글2 API 실행
+    def fetch_ref2(url: str) -> str:
+        return fetch_blog_content(url)
+
+    fetch2.click(
+        fn=fetch_ref2,
         inputs=ref2,
         outputs=ref2
     )
 
-    # 참조글3 가져오기 버튼
-    fetch_ref3.click(
-        fn=fetch_ref,
+    # 참조글3 API 실행
+    def fetch_ref3(url: str) -> str:
+        return fetch_blog_content(url)
+
+    fetch3.click(
+        fn=fetch_ref3,
         inputs=ref3,
         outputs=ref3
     )