Update app.py

app.py

@@ -25,9 +25,9 @@ def get_category_prompt(category):
 # 공포 마케팅 카피라이팅 생성 규칙
 1. 반드시 한글로 출력하라.
 2. 너는 세계 최고의 공포 마케팅 카피라이터이다.
-3. 반드시 카피라이팅은 30자 이내로 작성하고, 20개만 출력하라.(다른 내용 출력 금지)
+3. 반드시 카피라이팅은 30자 이내로 작성하고, 10개만 출력하라.(다른 내용 출력 금지)
 4. 반드시 입력된 주제로만 작성하고 예시를 참고하여 내용을 보강하라.
-5. 반드시 1 부터 20까지 번호(리스트형태)를 같이 출력하라
+5. 반드시 1 부터 10까지 번호(리스트형태)를 같이 출력하라
 6. 사람들에게 두려움을 주면서, 동시에 제품이나 서비스를 사용해야만 하는 이유를 강하게 어필하라.
 7. 사용하지 않을 때의 위험이나 손실을 강조하라.
 8. 감정적인 단어와 표현을 사용하여 독자의 불안을 증폭시키라
@@ -44,15 +44,14 @@ def get_category_prompt(category):
 - 바다 환경오염, 식탁으로 되돌아온다. 플라스틱 사용 이대로 괜찮은가요?
 - "맥주 한잔이라도 날마다 술이 땡긴다면? 이것 의심해야"
 """
-
     elif category == "스토리텔링":
         return """
 # 스토리텔링 카피라이팅 생성 규칙
 1. 반드시 한글로 출력하라.
 2. 너는 세계 최고의 스토리텔링 마케팅 카피라이터이다.
-3. 반드시 카피라이팅은 30자 이내로 작성하고, 20개만 출력하라.(다른 내용 출력 금지)
+3. 반드시 카피라이팅은 30자 이내로 작성하고, 10개만 출력하라.(다른 내용 출력 금지)
 4. 반드시 입력된 주제로만 작성하고 예시를 참고하여 내용을 보강하라.
-5. 반드시 1 부터 20까지 번호(리스트형태)를 같이 출력하라
+5. 반드시 1 부터 10까지 번호(리스트형태)를 같이 출력하라
 6. 제품/서비스와 관련된 짧고 흥미로운 이야기를 만들어라.
 7. 고객이 공감할 수 있는 상황이나 캐릭터를 설정하라.
 8. 이야기를 통해 제품/서비스의 가치를 자연스럽게 전달하라.
@@ -90,15 +89,15 @@ async def call_api_async(session, content, system_message, max_tokens, temperatu
         result = await response.json()
         return result["choices"][0]["message"]["content"]
 
-# 여러 카테고리에 대해 병렬 비동기 호출
+# 여러 카테고리에 대해 병렬 비동기 호출 (두 항목을 동시에 시작)
 async def generate_copywriting_async(categories, topic):
     max_tokens = 1000
     temperature = 0.8
     top_p = 0.95
-
     results = {}
+
     async with aiohttp.ClientSession() as session:
-        # 각 카테고리별로 async 호출 생성
+        # 카테고리별로 동시에 시작할 태스크 생성
         tasks = {}
         for category in categories:
             prompt = get_category_prompt(category)
@@ -106,29 +105,21 @@ async def generate_copywriting_async(categories, topic):
             tasks[category] = asyncio.create_task(
                 call_api_async(session, user_content, prompt, max_tokens, temperature, top_p)
             )
-        
-        # 완료된 태스크부터 결과를 가져와 순차적으로 반환
-        done_count = 0
-        total = len(tasks)
-        while done_count < total:
-            done, pending = await asyncio.wait(tasks.values(), return_when=asyncio.FIRST_COMPLETED)
-            for d in done:
-                finished_category = None
-                # 어느 카테고리 태스크가 끝났는지 식별
-                for cat, task_obj in tasks.items():
-                    if task_obj == d:
-                        finished_category = cat
-                        break
-                if finished_category:
-                    try:
-                        results[finished_category] = d.result()
-                    except Exception as e:
-                        results[finished_category] = f"에러 발생: {str(e)}"
-                    del tasks[finished_category]
-                    done_count += 1
-            
-            # 완료된 카테고리가 있을 때마다 지금까지의 결과를 yield로 반환
-            # (Gradio에서 스트리밍으로 처리하여 중간 결과가 바로바로 표시됨)
+
+        # 모든 태스크가 동시에 시작되며, 먼저 끝난 태스크부터 결과 반환
+        for done_task in asyncio.as_completed(tasks.values()):
+            finished_category = None
+            for cat, t in tasks.items():
+                if t == done_task:
+                    finished_category = cat
+                    break
+
+            try:
+                results[finished_category] = await done_task
+            except Exception as e:
+                results[finished_category] = f"에러 발생: {str(e)}"
+
+            # 현재까지 완료된 결과를 yield
             yield results
 
 # Gradio 인터페이스 부분
@@ -149,34 +140,30 @@ with gr.Blocks() as iface:
 
     # validate_and_generate 함수를 '비동기 + 스트리밍' 형태로 작성
     async def validate_and_generate(topic):
-        # 입력이 없을 경우
         if not topic:
             yield [gr.update(value="주제를 입력하세요")] + [gr.update(value="") for _ in CATEGORIES]
             return
         
-        # 초기 상태 출력
+        # 초기 상태
         yield [gr.update(value="카피라이팅 생성 중...")] + [gr.update(value="대기중...") for _ in CATEGORIES]
         
         # 비동기로 카테고리별 카피라이팅 생성
         async for partial_results in generate_copywriting_async(CATEGORIES, topic):
-            # partial_results 딕셔너리에 지금까지 완료된 카테고리 정보가 들어옴
             updates = []
             for cat in CATEGORIES:
-                # 이미 완료된 카테고리는 결과 표시, 미완료는 "대기중..."
                 if cat in partial_results:
                     updates.append(gr.update(value=partial_results[cat]))
                 else:
                     updates.append(gr.update(value="대기중..."))
             
-            # 상태 메시지: 완료된 카테고리 / 총 카테고리
+            # 진행상황(완료된 카테고리 수 / 전체 카테고리 수)
             yield [gr.update(value=f"진행상황: {len(partial_results)}/{len(CATEGORIES)} 카테고리 완료")] + updates
 
-    # 버튼 클릭 시 validate_and_generate를 호출
+    # 버튼 클릭 -> validate_and_generate(비동기 + 스트리밍)
     generate_btn.click(
         fn=validate_and_generate,
         inputs=[topic],
         outputs=[status] + [output_boxes[category] for category in CATEGORIES],
-        # Gradio에서 비동기 + 스트리밍 생성을 위해 아래 설정
         api_name="generate_copywriting",
         queue=True
     )