Update app.py

app.py

@@ -1,9 +1,8 @@
-# app.py
 import gradio as gr
 import openai
 import os
 from typing import Optional
-from gradio_client import Client  # gradio_client 추가
+from gradio_client import Client
 
 #############################
 
@@ -75,25 +74,9 @@ OPENAI_TEMPERATURE = 0.7
 OPENAI_TOP_P = 0.95
 
 #############################
-# API 클라이언트 설정
+# API 클라이언트 설정 (허깅페이스 스페이스)
 #############################
-# 외부 Space의 API 클라이언트 설정
-API_CLIENT = Client("Kims12/blog")
-
-def fetch_blog_content(url: str) -> str:
-    """
-    주어진 URL을 이용해 블로그 본문을 가져오는 함수.
-    """
-    if not url.strip():
-        return "URL을 입력해주세요."
-    try:
-        result = API_CLIENT.predict(
-            url=url,
-            api_name="/predict"
-        )
-        return result
-    except Exception as e:
-        return f"API 호출 중 오류가 발생했습니다: {str(e)}"
+client = Client("Kims12/blog")
 
 #############################
 # UI - 블로그 생성기
@@ -108,19 +91,79 @@ with gr.Blocks() as demo:
         value="일반적인"  # 기본 선택
     )
 
-    # 참조글 입력 (3개) 및 API 실행 버튼
+    # 참조글 입력 (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():
-        ref3 = gr.Textbox(label="참조글 3 (URL 입력)")
-        fetch3 = gr.Button("참조글3 API 실행")
+        with gr.Column():
+            ref1 = gr.Textbox(label="참조글 1")
+            fetch_button1 = gr.Button("API 실행1")
+        with gr.Column():
+            ref2 = gr.Textbox(label="참조글 2")
+            fetch_button2 = gr.Button("API 실행2")
+        with gr.Column():
+            ref3 = gr.Textbox(label="참조글 3")
+            fetch_button3 = gr.Button("API 실행3")
 
     output_box = gr.Textbox(label="결과", lines=20, interactive=False)
 
+    # 참조글1 API 실행 함수
+    def fetch_ref1(url: str) -> str:
+        if not url:
+            return "URL을 입력해주세요."
+        try:
+            result = client.predict(
+                url=url,
+                api_name="/predict"
+            )
+            return result
+        except Exception as e:
+            return f"오류가 발생했습니다: {str(e)}"
+
+    # 참조글2 API 실행 함수
+    def fetch_ref2(url: str) -> str:
+        if not url:
+            return "URL을 입력해주세요."
+        try:
+            result = client.predict(
+                url=url,
+                api_name="/predict"
+            )
+            return result
+        except Exception as e:
+            return f"오류가 발생했습니다: {str(e)}"
+
+    # 참조글3 API 실행 함수
+    def fetch_ref3(url: str) -> str:
+        if not url:
+            return "URL을 입력해주세요."
+        try:
+            result = client.predict(
+                url=url,
+                api_name="/predict"
+            )
+            return result
+        except Exception as e:
+            return f"오류가 발생했습니다: {str(e)}"
+
+    # 버튼 클릭 시 해당 참조글 API 실행
+    fetch_button1.click(
+        fn=fetch_ref1,
+        inputs=ref1,
+        outputs=ref1
+    )
+
+    fetch_button2.click(
+        fn=fetch_ref2,
+        inputs=ref2,
+        outputs=ref2
+    )
+
+    fetch_button3.click(
+        fn=fetch_ref3,
+        inputs=ref3,
+        outputs=ref3
+    )
+
+    # 블로그 생성 함수
     def generate_blog(tone_value: str, ref1_value: str, ref2_value: str, ref3_value: str) -> str:
         # 프롬프트 생성
         question = (
@@ -147,35 +190,5 @@ with gr.Blocks() as demo:
         outputs=output_box
     )
 
-    # 참조글1 API 실행
-    def fetch_ref1(url: str) -> str:
-        return fetch_blog_content(url)
-
-    fetch1.click(
-        fn=fetch_ref1,
-        inputs=ref1,
-        outputs=ref1
-    )
-
-    # 참조글2 API 실행
-    def fetch_ref2(url: str) -> str:
-        return fetch_blog_content(url)
-
-    fetch2.click(
-        fn=fetch_ref2,
-        inputs=ref2,
-        outputs=ref2
-    )
-
-    # 참조글3 API 실행
-    def fetch_ref3(url: str) -> str:
-        return fetch_blog_content(url)
-
-    fetch3.click(
-        fn=fetch_ref3,
-        inputs=ref3,
-        outputs=ref3
-    )
-
 if __name__ == "__main__":
     demo.launch()