Update app.py

app.py

@@ -10,15 +10,16 @@ import aiohttp
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# OpenAI API 클라이언트 설정
+# OpenAI API 키 설정
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
-# 요청사항에 따라 카테고리 수정
+# 카테고리
 CATEGORIES = [
     "공포 마케팅",
     "스토리텔링"
 ]
 
+# 카테고리별 프롬프트 반환 함수
 def get_category_prompt(category):
     if category == "공포 마케팅":
         return """
@@ -68,7 +69,7 @@ def get_category_prompt(category):
 - 팔 없는 장애인 화가의 꿈을 이뤄준 첨단 그림 도구
 """
 
-# 비동기 API 호출 함수
+# 비동기 OpenAI API 호출
 async def call_api_async(session, content, system_message, max_tokens, temperature, top_p):
     url = "https://api.openai.com/v1/chat/completions"
     headers = {
@@ -85,11 +86,14 @@ async def call_api_async(session, content, system_message, max_tokens, temperatu
         "temperature": temperature,
         "top_p": top_p,
     }
-    async with session.post(url, headers=headers, json=payload) as response:
-        response_json = await response.json()
-        return response_json["choices"][0]["message"]["content"]
+    async with session.post(url, headers=headers, json=payload) as resp:
+        resp_json = await resp.json()
+        # 응답 형식 예외처리
+        if "choices" not in resp_json or len(resp_json["choices"]) == 0:
+            return "에러 발생: API 응답이 올바르지 않습니다."
+        return resp_json["choices"][0]["message"]["content"]
 
-# 여러 카테고리를 "동시에" 요청 보내기
+# 여러 카테고리를 "동시에" 비동기로 처리
 async def generate_copywriting_async(categories, topic):
     max_tokens = 1000
     temperature = 0.8
@@ -97,34 +101,37 @@ async def generate_copywriting_async(categories, topic):
     results = {}
 
     async with aiohttp.ClientSession() as session:
-        # 카테고리별로 태스크를 만들어 "동시에" 실행
-        tasks = []
-        cat_map = {}
-        
-        for category in categories:
-            prompt = get_category_prompt(category)
+        # 딕셔너리 형태로 태스크 등록: { 카테고리: Task }
+        tasks = {}
+        for cat in categories:
+            prompt = get_category_prompt(cat)
             user_content = f"주제: {topic}"
-            # create_task를 통해 태스크 생성 후 배열에 담음
-            task = asyncio.create_task(
+            tasks[cat] = asyncio.create_task(
                 call_api_async(session, user_content, prompt, max_tokens, temperature, top_p)
             )
-            tasks.append(task)
-            cat_map[task] = category
-
-        # 모든 태스크가 한꺼번에 시작된 상태에서,
-        # 먼저 끝나는 태스크부터 결과를 얻어옴 (as_completed)
-        for done_task in asyncio.as_completed(tasks):
-            finished_category = cat_map[done_task]
+        
+        # 모든 태스크를 "동시에" 시작하고,
+        # 가장 먼저 끝난 태스크부터 순서대로 반환
+        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:
-                result = await done_task
-                results[finished_category] = result
+                results[finished_category] = await done_task
             except Exception as e:
                 results[finished_category] = f"에러 발생: {str(e)}"
-            
-            # 현재까지 완료된 카테고리 결과를 yield (부분완료 상태 반환)
+
+            # 지금까지의 부분 결과를 스트리밍
             yield results
 
+#############################
 # Gradio 인터페이스
+#############################
 with gr.Blocks() as iface:
     gr.Markdown("# AI 카피라이팅 생성기")
     
@@ -140,37 +147,35 @@ with gr.Blocks() as iface:
             output_box = gr.Textbox(label=category, visible=True)
             output_boxes[category] = output_box
 
-    # 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
+        # generate_copywriting_async에서 부분적으로 완료될 때마다 partial_results를 yield
         async for partial_results in generate_copywriting_async(CATEGORIES, topic):
-            # partial_results에는 지금까지 완료된 카테고리들의 결과가 담김
-            box_updates = []
+            # partial_results = { "공포 마케팅": "...", ... } 이미 끝난 것들만 저장
+            updates = []
             for cat in CATEGORIES:
                 if cat in partial_results:
-                    box_updates.append(gr.update(value=partial_results[cat]))
+                    updates.append(gr.update(value=partial_results[cat]))
                 else:
-                    box_updates.append(gr.update(value="대기중..."))
+                    updates.append(gr.update(value="대기중..."))
             
-            # 진행상황 안내 (완료된 카테고리 개수 / 전체)
-            yield [gr.update(value=f"진행상황: {len(partial_results)}/{len(CATEGORIES)} 카테고리 완료")] + box_updates
+            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[cat] for cat in CATEGORIES],
-        api_name="generate_copywriting",
+        # Gradio에서 비동기 스트리밍을 지원하기 위해 queue=True
         queue=True
     )
 
 # 인터페이스 실행
-iface.launch()
+iface.launch(server_name="0.0.0.0", server_port=7860, share=False)