Update app.py

app.py

@@ -44,6 +44,9 @@ def choose_alternative(transformation):
     else:
         return random.choice([left, right])
 
+# 창의적인 모델/컨셉/형상 변화 아이디어를 위한 카테고리 (기존 '물리적 변화' 사전 유지)
+
+
 # 창의적인 모델/컨셉/형상 변화 아이디어를 위한 카테고리 (기존 '물리적 변화' 사전 유지)
 physical_transformation_categories = {
     "공간 이동": [
@@ -283,30 +286,20 @@ def generate_three_objects_interaction(obj1, obj2, obj3):
 ##############################################################################
 # 생성된 기본 설명을 LLM을 통해 확장
 ##############################################################################
-def enhance_descriptions(results, objects, progress=None):
+def enhance_descriptions(results, objects):
+    # progress 인자 제거
     obj_name = " 및 ".join([obj for obj in objects if obj])
-    total_categories = len(results)
     
-    for i, (category, result) in enumerate(results.items()):
-        # 프로그레스 업데이트 (진행률을 0.2~0.9 사이로 매핑)
-        if progress:
-            progress_value = 0.2 + (i / total_categories) * 0.7
-            progress(progress_value, f"카테고리 '{category}' 처리 중...")
-        
+    for category, result in results.items():
         result["enhanced"] = enhance_with_llm(result["base"], obj_name, category)
     
-    # 완료 메시지
-    if progress:
-        progress(0.95, "결과 정리 중...")
-    
     return results
 
 ##############################################################################
 # 사용자 입력(최대 3개 키워드)에 따라 창의적 변화 아이디어 생성
 ##############################################################################
-def generate_transformations(text1, text2=None, text3=None, progress=None):
-    if progress:
-        progress(0.1, "아이디어 기본 구조 생성 중...")
+def generate_transformations(text1, text2=None, text3=None):
+    # progress 인자 제거
     
     if text2 and text3:
         results = generate_three_objects_interaction(text1, text2, text3)
@@ -318,10 +311,7 @@ def generate_transformations(text1, text2=None, text3=None, progress=None):
         results = generate_single_object_transformations(text1)
         objects = [text1]
     
-    if progress:
-        progress(0.2, "Gemini API를 통한 아이디어 확장 중...")
-    
-    return enhance_descriptions(results, objects, progress)
+    return enhance_descriptions(results, objects)
 
 ##############################################################################
 # 결과 포맷팅
@@ -343,7 +333,8 @@ def process_inputs(text1, text2, text3, progress=gr.Progress()):
     if not text1:
         return "오류: 최소 하나의 키워드를 입력해주세요."
     
-    progress(0.05, "아이디어 생성 준비 중...")
+    # 진행 상태 메시지 (중첩 리스트 대신 단순 문자열 사용)
+    status_message = ""
     
     # 키워드 정보 확인
     keyword_info = f"키워드: {text1}"
@@ -352,16 +343,26 @@ def process_inputs(text1, text2, text3, progress=gr.Progress()):
     if text3:
         keyword_info += f", {text3}"
     
-    progress(0.1, "창의적인 모델/컨셉/형상 변화 아이디어 생성 시작...")
-    
-    # 결과 생성
-    results = generate_transformations(text1, text2, text3, progress)
-    
-    progress(0.95, "결과 포맷팅 중...")
-    formatted = format_results(results)
-    
-    progress(1.0, "완료!")
-    return formatted
+    # 결과 생성 (progress 인자 전달하지 않음)
+    try:
+        # 진행 상태를 단계별로 표시
+        progress(0.05, desc="아이디어 생성 준비 중...")
+        time.sleep(0.3)  # 시각적 효과를 위한 짧은 지연
+        
+        progress(0.1, desc="창의적인 모델/컨셉/형상 변화 아이디어 생성 시작...")
+        
+        # text1, text2, text3만 전달하고 progress는 전달하지 않음
+        results = generate_transformations(text1, text2, text3)
+        
+        progress(0.8, desc="결과 포맷팅 중...")
+        formatted = format_results(results)
+        
+        progress(1.0, desc="완료!")
+        return formatted
+        
+    except Exception as e:
+        logger.error(f"Error in process_inputs: {e}")
+        return f"처리 중 오류가 발생했습니다: {str(e)}"
 
 ##############################################################################
 # API 키 경고 메시지
@@ -402,9 +403,24 @@ with gr.Blocks(title="키워드 기반 창의적 변화 아이디어 생성기",
             text_input2 = gr.Textbox(label="키워드 2 (선택)", placeholder="예: 인공지능")
             text_input3 = gr.Textbox(label="키워드 3 (선택)", placeholder="예: 헬스케어")
             
-            # 상태 메시지를 표시할 영역 추가
+            # 입력과 버튼 사이에 상태 메시지를 표시할 영역 추가
             status_msg = gr.Markdown("💡 '아이디어 생성하기' 버튼을 클릭하면 프로그레스 바��� 나타납니다.")
             
+            # 처리 중일 때만 표시되는 상태 메시지
+            with gr.Box(visible=False) as processing_box:
+                processing_indicator = gr.HTML("""
+                <div style="display: flex; justify-content: center; align-items: center; margin: 10px 0;">
+                    <div style="border: 5px solid #f3f3f3; border-top: 5px solid #3498db; border-radius: 50%; width: 30px; height: 30px; animation: spin 2s linear infinite;"></div>
+                    <p style="margin-left: 10px; font-weight: bold; color: #3498db;">처리 중입니다...</p>
+                </div>
+                <style>
+                @keyframes spin {
+                    0% { transform: rotate(0deg); }
+                    100% { transform: rotate(360deg); }
+                }
+                </style>
+                """)
+            
             submit_button = gr.Button("아이디어 생성하기", variant="primary")
         
         # 우측 출력 영역
@@ -414,19 +430,37 @@ with gr.Blocks(title="키워드 기반 창의적 변화 아이디어 생성기",
     # 예시 입력
     gr.Examples(
         examples=[
-            ["스마트폰", "", ""],
-            ["커피", "책", ""],
-            ["자동차", "로봇", "인공지능"],
+
+            ["자동차", "", ""],
+            ["스마트폰", "인공지능", ""],
+            ["드론", "인공지능", ""],
             ["운동화", "웨어러블", "건강"],
+
         ],
         inputs=[text_input1, text_input2, text_input3],
     )
     
-    # 버튼 이벤트
+    # 처리 시작 전 함수 (로딩 표시기 보이게)
+    def show_processing_indicator():
+        return gr.update(visible=True)
+    
+    # 처리 완료 후 함수 (로딩 표시기 숨기기)
+    def hide_processing_indicator():
+        return gr.update(visible=False)
+    
+    # 버튼 이벤트 - 여러 함수 순차 실행
     submit_button.click(
+        fn=show_processing_indicator,
+        inputs=None,
+        outputs=processing_box
+    ).then(
         fn=process_inputs,
         inputs=[text_input1, text_input2, text_input3],
         outputs=idea_output
+    ).then(
+        fn=hide_processing_indicator,
+        inputs=None,
+        outputs=processing_box
     )
 
 if __name__ == "__main__":