Update app.py

app.py

@@ -1,5 +1,3 @@
-
-
 import os
 import gradio as gr
 import random
@@ -44,10 +42,7 @@ def choose_alternative(transformation):
     else:
         return random.choice([left, right])
 
-# 창의적인 모델/컨셉/형상 변화 아이디어를 위한 카테고리 (기존 '물리적 변화' 사전 유지)
-
-
-# 창의적인 모델/컨셉/형상 변화 아이디어를 위한 카테고리 (기존 '물리적 변화' 사전 유지)
+# 창의적인 모델/컨셉/형상 변화 아이디어를 위한 카테고리
 physical_transformation_categories = {
     "공간 이동": [
         "앞/뒤 이동", "좌/우 이동", "위/아래 이동", "세로축 회전(고개 끄덕임)", 
@@ -287,7 +282,6 @@ def generate_three_objects_interaction(obj1, obj2, obj3):
 # 생성된 기본 설명을 LLM을 통해 확장
 ##############################################################################
 def enhance_descriptions(results, objects):
-    # progress 인자 제거
     obj_name = " 및 ".join([obj for obj in objects if obj])
     
     for category, result in results.items():
@@ -299,8 +293,6 @@ def enhance_descriptions(results, objects):
 # 사용자 입력(최대 3개 키워드)에 따라 창의적 변화 아이디어 생성
 ##############################################################################
 def generate_transformations(text1, text2=None, text3=None):
-    # progress 인자 제거
-    
     if text2 and text3:
         results = generate_three_objects_interaction(text1, text2, text3)
         objects = [text1, text2, text3]
@@ -325,7 +317,7 @@ def format_results(results):
 ##############################################################################
 # Gradio UI에서 호출할 함수
 ##############################################################################
-def process_inputs(text1, text2, text3, progress=gr.Progress()):
+def process_inputs(text1, text2, text3, selected_category, progress=gr.Progress()):
     text1 = text1.strip() if text1 else None
     text2 = text2.strip() if text2 else None
     text3 = text3.strip() if text3 else None
@@ -333,36 +325,30 @@ def process_inputs(text1, text2, text3, progress=gr.Progress()):
     if not text1:
         return "오류: 최소 하나의 키워드를 입력해주세요."
     
-    # 진행 상태 메시지 (중첩 리스트 대신 단순 문자열 사용)
-    status_message = ""
-    
-    # 키워드 정보 확인
     keyword_info = f"키워드: {text1}"
     if text2:
         keyword_info += f", {text2}"
     if text3:
         keyword_info += f", {text3}"
     
-    # 결과 생성 (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)}"
+    progress(0.05, desc="아이디어 생성 준비 중...")
+    time.sleep(0.3)  # 시각적 효과를 위한 짧은 지연
+    
+    progress(0.1, desc="창의적인 모델/컨셉/형상 변화 아이디어 생성 시작...")
+    
+    results = generate_transformations(text1, text2, text3)
+    
+    # 선택한 카테고리에 해당하는 결과만 필터링
+    if selected_category in results:
+        results = {selected_category: results[selected_category]}
+    else:
+        return "선택한 카테고리가 결과에 존재하지 않습니다."
+    
+    progress(0.8, desc="결과 포맷팅 중...")
+    formatted = format_results(results)
+    
+    progress(1.0, desc="완료!")
+    return formatted
 
 ##############################################################################
 # API 키 경고 메시지
@@ -392,7 +378,7 @@ with gr.Blocks(title="키워드 기반 창의적 변화 아이디어 생성기",
     """)
     
     gr.Markdown("# 🚀 키워드 기반 창의적 변화 아이디어 생성기")
-    gr.Markdown("입력한 **키워드**(최대 3개)를 바탕으로, **창의적인 모델/컨셉/형상 변화**에 대한 이해와 **혁신 포인트**, **기능성** 등을 중심으로 확장된 아이디어를 제시합니다.")
+    gr.Markdown("입력한 **키워드**(최대 3개)와 **카테고리**를 바탕으로, **창의적인 모델/컨셉/형상 변화**에 대한 이해와 **혁신 포인트**, **기능성** 등을 중심으로 확장된 아이디어를 제시합니다.")
     
     warning = gr.Markdown(get_warning_message())
     
@@ -402,11 +388,16 @@ with gr.Blocks(title="키워드 기반 창의적 변화 아이디어 생성기",
             text_input1 = gr.Textbox(label="키워드 1 (필수)", placeholder="예: 스마트폰")
             text_input2 = gr.Textbox(label="키워드 2 (선택)", placeholder="예: 인공지능")
             text_input3 = gr.Textbox(label="키워드 3 (선택)", placeholder="예: 헬스케어")
+            # 카테고리 선택 드롭다운 추가
+            category_dropdown = gr.Dropdown(
+                label="카테고리 선택",
+                choices=list(physical_transformation_categories.keys()),
+                value=list(physical_transformation_categories.keys())[0],
+                info="출력할 카테고리를 선택하세요."
+            )
             
-            # 입력과 버튼 사이에 상태 메시지를 표시할 영역 추가
             status_msg = gr.Markdown("💡 '아이디어 생성하기' 버튼을 클릭하면 아이디어 생성이 시작됩니다.")
             
-            # 처리 중일 때만 표시되는 상태 메시지
             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>
@@ -426,34 +417,30 @@ with gr.Blocks(title="키워드 기반 창의적 변화 아이디어 생성기",
         with gr.Column(scale=2):
             idea_output = gr.Markdown(label="아이디어 결과")
 
-    # 예시 입력
     gr.Examples(
         examples=[
-            ["스마트폰", "", ""],
-            ["자동차", "", ""],            
-            ["자동차", "인공지능", ""],
-            ["드론", "인공지능", ""],
-            ["운동화", "웨어러블", "건강"],
+            ["스마트폰", "", "", list(physical_transformation_categories.keys())[0]],
+            ["자동차", "", "", list(physical_transformation_categories.keys())[0]],            
+            ["자동차", "인공지능", "", list(physical_transformation_categories.keys())[0]],
+            ["드론", "인공지능", "", list(physical_transformation_categories.keys())[0]],
+            ["운동화", "웨어러블", "건강", list(physical_transformation_categories.keys())[0]],
         ],
-        inputs=[text_input1, text_input2, text_input3],
+        inputs=[text_input1, text_input2, text_input3, category_dropdown],
     )
     
-    # 처리 시작 전 함수 (로딩 표시기 보이게)
     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_indicator
     ).then(
         fn=process_inputs,
-        inputs=[text_input1, text_input2, text_input3],
+        inputs=[text_input1, text_input2, text_input3, category_dropdown],
         outputs=idea_output
     ).then(
         fn=hide_processing_indicator,
@@ -462,4 +449,4 @@ with gr.Blocks(title="키워드 기반 창의적 변화 아이디어 생성기",
     )
 
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    demo.launch(debug=True)