Update app.py

app.py

@@ -1,8 +1,31 @@
+#app.py
 import gradio as gr
 import info_blog
 import travel_blog
 import product_blog
 
+# 래퍼 함수들 - 5개 반환값을 4개로 조정
+def generate_info_blog_wrapper(query, style):
+    try:
+        html_result, ref1, ref2, ref3, char_count = info_blog.generate_blog_post(query, style)
+        return html_result, ref1, ref2, ref3
+    except Exception as e:
+        return f"<p>오류 발생: {str(e)}</p>", "오류", "오류", "오류"
+
+def generate_travel_blog_wrapper(query, style):
+    try:
+        html_result, ref1, ref2, ref3, char_count = travel_blog.generate_blog_post(query, style)
+        return html_result, ref1, ref2, ref3
+    except Exception as e:
+        return f"<p>오류 발생: {str(e)}</p>", "오류", "오류", "오류"
+
+def generate_product_blog_wrapper(query, style):
+    try:
+        html_result, ref1, ref2, ref3, char_count = product_blog.generate_blog_post(query, style)
+        return html_result, ref1, ref2, ref3
+    except Exception as e:
+        return f"<p>오류 발생: {str(e)}</p>", "오류", "오류", "오류"
+
 # Gradio 인터페이스 설계
 def interface():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -35,8 +58,9 @@ def interface():
                 reference_2 = gr.Textbox(label="참고글 2")
                 reference_3 = gr.Textbox(label="참고글 3")
             
+            # 수정된 버튼 클릭 이벤트
             generate_button_1.click(
-                info_blog.generate_blog_post, 
+                generate_info_blog_wrapper, 
                 inputs=[query_1, style_1], 
                 outputs=[blog_post_1, reference_1, reference_2, reference_3]
             )
@@ -67,8 +91,9 @@ def interface():
                 reference_5 = gr.Textbox(label="참고글 2")
                 reference_6 = gr.Textbox(label="참고글 3")
             
+            # 수정된 버튼 클릭 이벤트
             generate_button_2.click(
-                travel_blog.generate_blog_post, 
+                generate_travel_blog_wrapper, 
                 inputs=[query_2, style_2], 
                 outputs=[blog_post_2, reference_4, reference_5, reference_6]
             )
@@ -99,8 +124,9 @@ def interface():
                 reference_8 = gr.Textbox(label="참고글 2")
                 reference_9 = gr.Textbox(label="참고글 3")
             
+            # 수정된 버튼 클릭 이벤트
             generate_button_3.click(
-                product_blog.generate_blog_post, 
+                generate_product_blog_wrapper, 
                 inputs=[query_3, style_3], 
                 outputs=[blog_post_3, reference_7, reference_8, reference_9]
             )