Update app.py

app.py

@@ -68,7 +68,7 @@ def get_category_prompt(category):
 - 팔 없는 장애인 화가의 꿈을 이뤄준 첨단 그림 도구
 """
 
-# 기존의 동기적 호출 대신 비동기 API 호출
+# 비동기 API 호출 함수
 async def call_api_async(session, content, system_message, max_tokens, temperature, top_p):
     url = "https://api.openai.com/v1/chat/completions"
     headers = {
@@ -86,10 +86,10 @@ async def call_api_async(session, content, system_message, max_tokens, temperatu
         "top_p": top_p,
     }
     async with session.post(url, headers=headers, json=payload) as response:
-        result = await response.json()
-        return result["choices"][0]["message"]["content"]
+        response_json = await response.json()
+        return response_json["choices"][0]["message"]["content"]
 
-# 여러 카테고리에 대해 병렬 비동기 호출 (두 항목을 동시에 시작)
+# 여러 카테고리를 "동시에" 요청 보내기
 async def generate_copywriting_async(categories, topic):
     max_tokens = 1000
     temperature = 0.8
@@ -97,32 +97,34 @@ async def generate_copywriting_async(categories, topic):
     results = {}
 
     async with aiohttp.ClientSession() as session:
-        # 카테고리별로 동시에 시작할 태스크 생성
-        tasks = {}
+        # 카테고리별로 태스크를 만들어 "동시에" 실행
+        tasks = []
+        cat_map = {}
+        
         for category in categories:
             prompt = get_category_prompt(category)
             user_content = f"주제: {topic}"
-            tasks[category] = asyncio.create_task(
+            # create_task를 통해 태스크 생성 후 배열에 담음
+            task = asyncio.create_task(
                 call_api_async(session, user_content, prompt, max_tokens, temperature, top_p)
             )
+            tasks.append(task)
+            cat_map[task] = category
 
-        # 모든 태스크가 동시에 시작되며, 먼저 끝난 태스크부터 결과 반환
-        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
-
+        # 모든 태스크가 한꺼번에 시작된 상태에서,
+        # 먼저 끝나는 태스크부터 결과를 얻어옴 (as_completed)
+        for done_task in asyncio.as_completed(tasks):
+            finished_category = cat_map[done_task]
             try:
-                results[finished_category] = await done_task
+                result = await done_task
+                results[finished_category] = result
             except Exception as e:
                 results[finished_category] = f"에러 발생: {str(e)}"
-
-            # 현재까지 완료된 결과를 yield
+            
+            # 현재까지 완료된 카테고리 결과를 yield (부분완료 상태 반환)
             yield results
 
-# Gradio 인터페이스 부분
+# Gradio 인터페이스
 with gr.Blocks() as iface:
     gr.Markdown("# AI 카피라이팅 생성기")
     
@@ -138,32 +140,34 @@ with gr.Blocks() as iface:
             output_box = gr.Textbox(label=category, visible=True)
             output_boxes[category] = output_box
 
-    # validate_and_generate 함수를 '비동기 + 스트리밍' 형태로 작성
+    # 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]
         
-        # 비동기로 카테고리별 카피라이팅 생성
+        # 'generate_copywriting_async'에서 완료될 때마다 partial_results를 yield
         async for partial_results in generate_copywriting_async(CATEGORIES, topic):
-            updates = []
+            # partial_results에는 지금까지 완료된 카테고리들의 결과가 담김
+            box_updates = []
             for cat in CATEGORIES:
                 if cat in partial_results:
-                    updates.append(gr.update(value=partial_results[cat]))
+                    box_updates.append(gr.update(value=partial_results[cat]))
                 else:
-                    updates.append(gr.update(value="대기중..."))
+                    box_updates.append(gr.update(value="대기중..."))
             
-            # 진행상황(완료된 카테고리 수 / 전체 카테고리 수)
-            yield [gr.update(value=f"진행상황: {len(partial_results)}/{len(CATEGORIES)} 카테고리 완료")] + updates
+            # 진행상황 안내 (완료된 카테고리 개수 / 전체)
+            yield [gr.update(value=f"진행상황: {len(partial_results)}/{len(CATEGORIES)} 카테고리 완료")] + box_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],
+        outputs=[status] + [output_boxes[cat] for cat in CATEGORIES],
         api_name="generate_copywriting",
         queue=True
     )