Update app.py

app.py

@@ -70,7 +70,7 @@ def get_category_prompt(category):
 
 async def call_api_async(session, content, system_message, max_tokens, temperature, top_p):
     payload = {
-        "model": "gpt-4o-mini",  # 모델 이름을 실제 사용 가능한 것으로 변경하세요.
+        "model": "gpt-4o-mini",  # 실제 사용 가능한 모델 이름으로 변경하세요.
         "messages": [
             {"role": "system", "content": system_message},
             {"role": "user", "content": content},
@@ -103,7 +103,11 @@ async def generate_copywriting_async(categories, topic):
             task = asyncio.create_task(call_api_async(session, user_content, prompt, max_tokens, temperature, top_p))
             tasks.append((category, task))
         
-        for category, task in tasks:
+        # Mapping tasks to categories
+        task_to_category = {task: category for category, task in tasks}
+
+        for task in asyncio.as_completed(task_to_category.keys()):
+            category = task_to_category[task]
             try:
                 copywriting = await task
                 results[category] = copywriting
@@ -134,22 +138,54 @@ with gr.Blocks() as iface:
 
     async def validate_and_generate(topic_input):
         try:
-            # 상태 업데이트: 생성 중
-            status_update = "카피라이팅 생성 중..."
-            outputs_initial = ["" for _ in CATEGORIES]
-            yield [gr.update(value=status_update)] + [gr.update(value="") for _ in CATEGORIES]
-            
-            # 카피라이팅 생성
-            results = await generate_copywriting_async(CATEGORIES, topic_input)
-            
-            # 상태 업데이트: 완료 및 결과 표시
-            status_complete = "카피라이팅 생성이 완료되었습니다."
-            outputs_final = [results.get(category, "") for category in CATEGORIES]
-            yield [gr.update(value=status_complete)] + [gr.update(value=copy) for copy in outputs_final]
+            # 초기 상태 설정: "카피라이팅 생성 중..." 및 빈 출력 박스
+            current_outputs = ["카피라이팅 생성 중..."] + [""] * len(CATEGORIES)
+            yield [gr.Markdown.update(value=current_outputs[0])] + [gr.Textbox.update(value=o) for o in current_outputs[1:]]
+
+            # 비동기 카피라이팅 생성
+            async with aiohttp.ClientSession() as session:
+                tasks = []
+                for category in CATEGORIES:
+                    prompt = get_category_prompt(category)
+                    user_content = f"주제: {topic_input}"
+                    task = asyncio.create_task(call_api_async(session, user_content, prompt, 1000, 0.8, 0.95))
+                    tasks.append((category, task))
+                
+                # 태스크를 카테고리에 매핑
+                task_to_category = {task: category for category, task in tasks}
+
+                for task in asyncio.as_completed(task_to_category.keys()):
+                    category = task_to_category[task]
+                    try:
+                        copywriting = await task
+                        index = CATEGORIES.index(category) + 1  # +1은 status를 제외하기 위함
+                        current_outputs[index] = copywriting
+                        # 업데이트할 출력 리스트 생성
+                        updates = [gr.Markdown.update(value="카피라이팅 생성 중...")] + [
+                            gr.Textbox.update(value=current_outputs[i + 1]) if i + 1 == index else gr.Textbox.update()
+                            for i in range(len(CATEGORIES))
+                        ]
+                        yield updates
+                    except Exception as e:
+                        logger.error(f"Error generating copywriting for {category}: {str(e)}")
+                        index = CATEGORIES.index(category) + 1
+                        current_outputs[index] = f"오류 발생: {str(e)}"
+                        updates = [gr.Markdown.update(value="카피라이팅 생성 중...")] + [
+                            gr.Textbox.update(value=current_outputs[i + 1]) if i + 1 == index else gr.Textbox.update()
+                            for i in range(len(CATEGORIES))
+                        ]
+                        yield updates
+
+            # 최종 상태 업데이트: "카피라이팅 생성이 완료되었습니다."
+            final_updates = [gr.Markdown.update(value="카피라이팅 생성이 완료되었습니다.")] + [
+                gr.Textbox.update(value=current_outputs[i + 1]) for i in range(len(CATEGORIES))
+            ]
+            yield final_updates
+
         except Exception as e:
             logger.error(f"Error during copywriting generation: {str(e)}")
             error_message = f"오류 발생: {str(e)}"
-            yield [gr.update(value=error_message)] + [gr.update(value="") for _ in CATEGORIES]
+            yield [gr.Markdown.update(value=error_message)] + [gr.Textbox.update(value="") for _ in CATEGORIES]
 
     generate_btn.click(
         fn=validate_and_generate,