Update app.py

app.py

@@ -44,7 +44,6 @@ def get_category_prompt(category):
 - 바다 환경오염, 식탁으로 되돌아온다. 플라스틱 사용 이대로 괜찮은가요?
 - "맥주 한잔이라도 날마다 술이 땡긴다면? 이것 의심해야"
 """
-
     elif category == "스토리텔링":
         return """
 # 스토리텔링 카피라이팅 생성 규칙
@@ -175,46 +174,41 @@ with gr.Blocks() as iface:
             output_boxes[category] = output_box
 
     ########################################################
-    # 기존 validate_and_generate 함수 이름/출력 형식 유지
-    # → 내부에서 비동기 처리 + Gradio 스트리밍(yield) 사용
+    # (수정) validate_and_generate 함수를 비동기 함수로 바꿔
+    # 3개의 출력에 맞게 매 yield마다 [status, box1, box2]를 반환
     ########################################################
-    def validate_and_generate(topic):
-        async def async_gen():
-            try:
-                # 비동기로 카피라이팅 생성
-                results = {}
-                
-                # 순차가 아닌, '완료된 순서대로' 결과를 yield
-                async for cat, copywriting in generate_copywriting_async(CATEGORIES, topic):
-                    results[cat] = copywriting
-                    # 현재까지 완료된 내용만 업데이트
-                    yield [
-                        gr.update(value=f"진행중: [{cat}] 카피라이팅 완료"),
-                        *[gr.update(value=results.get(c, "")) for c in CATEGORIES]
-                    ]
-
-                # 모든 카테고리 완료 후 최종 메세지
+    async def validate_and_generate(topic):
+        try:
+            results = {}
+            # 완료된 순서대로 스트리밍
+            async for cat, copywriting in generate_copywriting_async(CATEGORIES, topic):
+                results[cat] = copywriting
+                # 현재까지 완료된 copywriting을 반영해서 출력
                 yield [
-                    gr.update(value="카피라이팅 생성이 완료되었습니다."),
+                    gr.update(value=f"진행중: [{cat}] 카피라이팅 완료"),
                     *[gr.update(value=results.get(c, "")) for c in CATEGORIES]
                 ]
 
-            except Exception as e:
-                logger.error(f"Error during copywriting generation: {str(e)}")
-                yield [
-                    gr.update(value=f"오류 발생: {str(e)}"),
-                    *[gr.update(value="") for _ in CATEGORIES]
-                ]
+            # 모든 카테고리가 완료된 뒤 최종 출력
+            yield [
+                gr.update(value="카피라이팅 생성이 완료되었습니다."),
+                *[gr.update(value=results.get(c, "")) for c in CATEGORIES]
+            ]
 
-        # Gradio에서 스트리밍 출력을 위해 async_gen 함수 자체를 반환
-        return async_gen()
+        except Exception as e:
+            logger.error(f"Error during copywriting generation: {str(e)}")
+            yield [
+                gr.update(value=f"오류 발생: {str(e)}"),
+                *[gr.update(value="") for _ in CATEGORIES]
+            ]
 
+    # 비동기 함수를 Gradio 이벤트 핸들러로 등록
     generate_btn.click(
         fn=validate_and_generate,
         inputs=[topic],
         outputs=[status] + [output_boxes[category] for category in CATEGORIES]
     )
 
-# 비동기 처리를 위해 queue()를 사용(Gradio 3.30+ 기준)
-iface.queue()
-iface.launch()
+    # Gradio 3.30+부터는 queue() + async 함수로 스트리밍 가능
+    iface.queue()
+    iface.launch()