Update app.py

app.py

@@ -448,5 +448,296 @@ with gr.Blocks(title="키워드 기반 창의적 변화 아이디어 생성기",
         outputs=processing_indicator
     )
 
+if __name__ == "__main__":
+    demo.launch(debug=True)
+
+    ],
+    
+    "환경 상호작용": [
+        "온도 반응", "습도 반응", "기압 반응", "중력 반응", "자기장 반응",
+        "빛 반응", "소리 반응", "화학 물질 감지", "기계적 자극 감지", "전기 자극 반응",
+        "방사선 반응", "진동 감지", "pH 반응", "용매 반응", "기체 교환",
+        "환경 오염 반응", "날씨 반응", "계절 변화 반응", "일주기 반응", "생태계 상호작용",
+        "공생/경쟁 반응", "포식/피식 관계", "군집 형성", "영역 설정", "이주/정착 패턴"
+    ],
+    
+    "센서 기능": [
+        "시각 센서/감지", "청각 센서/감지", "촉각 센서/감지", "미각 센서/감지", "후각 센서/감지",
+        "온도 센서/감지", "습도 센서/감지", "압력 센서/감지", "가속도 센서/감지", "회전 센서/감지",
+        "근접 센서/감지", "위치 센서/감지", "운동 센서/감지", "가스 센서/감지", "적외선 센서/감지",
+        "자외선 센서/감지", "방사선 센서/감지", "자기장 센서/감지", "전기장 센서/감지", "화학물질 센서/감지",
+        "생체신호 센서/감지", "진동 센서/감지", "소음 센서/감지", "빛 세기 센서/감지", "빛 파장 센서/감지",
+        "기울기 센서/감지", "pH 센서/감지", "전류 센서/감지", "전압 센서/감지", "이미지 센서/감지",
+        "거리 센서/감지", "깊이 센서/감지", "중력 센서/감지", "속도 센서/감지", "흐름 센서/감지",
+        "수위 센서/감지", "탁도 센서/감지", "염도 센서/감지", "금속 감지", "압전 센서/감지", 
+        "광전 센서/감지", "열전대 센서/감지", "홀 효과 센서/감지", "초음파 센서/감지", "레이더 센서/감지",
+        "라이다 센서/감지", "터치 센서/감지", "제스처 센서/감지", "심박 센서/감지", "혈압 센서/감지"
+    ]
+}
+
+##############################################################################
+# Gemini API 호출 함수 (예: gemini-2.0-flash-thinking-exp-01-21 -> 다른 모델 사용 시 수정)
+##############################################################################
+def query_gemini_api(prompt):
+    try:
+        # 예시: 기존 gemini-2.0... 대신, 다른 모델이 필요하다면 교체하세요.
+        model = genai.GenerativeModel('gemini-2.0-flash-thinking-exp-01-21')
+        
+        response = model.generate_content(prompt)
+
+        # 응답 구조 방어적으로 처리
+        try:
+            if hasattr(response, 'text'):
+                return response.text
+            
+            if hasattr(response, 'candidates') and response.candidates:
+                if len(response.candidates) > 0:
+                    candidate = response.candidates[0]
+                    if hasattr(candidate, 'content'):
+                        content = candidate.content
+                        if hasattr(content, 'parts') and content.parts:
+                            if len(content.parts) > 0:
+                                return content.parts[0].text
+            if hasattr(response, 'parts') and response.parts:
+                if len(response.parts) > 0:
+                    return response.parts[0].text
+
+            return "Unable to generate a response. API response structure is different than expected."
+            
+        except Exception as inner_e:
+            logger.error(f"Error processing response: {inner_e}")
+            return f"An error occurred while processing the response: {str(inner_e)}"
+            
+    except Exception as e:
+        logger.error(f"Error calling Gemini API: {e}")
+        if "API key not valid" in str(e):
+            return "API key is not valid. Please check your GEMINI_API_KEY environment variable."
+        return f"An error occurred while calling the API: {str(e)}"
+
+##############################################################################
+# 설명 확장 함수: "모델/컨셉/형상의 변화에 대한 이해와 혁신 포인트, 기능성 등을 중심"으로
+##############################################################################
+def enhance_with_llm(base_description, obj_name, category):
+    prompt = f"""
+다음은 '{obj_name}'의 '{category}' 관련 간단한 설명입니다:
+"{base_description}"
+위 내용을 보다 구체화하여, 
+1) 창의적인 모델/컨셉/형상의 변화에 대한 이해, 
+2) 혁신 포인트와 기능성 등을 중심으로 
+3~4문장의 아이디어로 확장해 주세요.
+"""
+    return query_gemini_api(prompt)
+
+##############################################################################
+# 단일 키워드(오브젝트)에 대한 "창의적 변화 아이디어" 생성
+##############################################################################
+def generate_single_object_transformations(obj):
+    results = {}
+    for category, transformations in physical_transformation_categories.items():
+        transformation = choose_alternative(random.choice(transformations))
+        base_description = f"{obj}이(가) {transformation} 현상을 보인다"
+        results[category] = {"base": base_description, "enhanced": None}
+    return results
+
+##############################################################################
+# 두 키워드에 대한 "창의적 변화 아이디어" 생성
+##############################################################################
+def generate_two_objects_interaction(obj1, obj2):
+    results = {}
+    for category, transformations in physical_transformation_categories.items():
+        transformation = choose_alternative(random.choice(transformations))
+        template = random.choice([
+            "{obj1}이(가) {obj2}에 결합하여 {change}가 발생했다",
+            "{obj1}과(와) {obj2}이(가) 충돌하면서 {change}가 일어났다"
+        ])
+        base_description = template.format(obj1=obj1, obj2=obj2, change=transformation)
+        results[category] = {"base": base_description, "enhanced": None}
+    return results
+
+##############################################################################
+# 세 키워드에 대한 "창의적 변화 아이디어" 생성
+##############################################################################
+def generate_three_objects_interaction(obj1, obj2, obj3):
+    results = {}
+    for category, transformations in physical_transformation_categories.items():
+        transformation = choose_alternative(random.choice(transformations))
+        template = random.choice([
+            "{obj1}, {obj2}, {obj3}이(가) 삼각형 구조로 결합하여 {change}가 발생했다",
+            "{obj1}이(가) {obj2}와(과) {obj3} 사이에서 매개체 역할을 하며 {change}를 촉진했다"
+        ])
+        base_description = template.format(obj1=obj1, obj2=obj2, obj3=obj3, change=transformation)
+        results[category] = {"base": base_description, "enhanced": None}
+    return results
+
+##############################################################################
+# 생성된 기본 설명을 LLM을 통해 확장
+##############################################################################
+def enhance_descriptions(results, objects):
+    obj_name = " 및 ".join([obj for obj in objects if obj])
+    
+    for category, result in results.items():
+        result["enhanced"] = enhance_with_llm(result["base"], obj_name, category)
+    
+    return results
+
+##############################################################################
+# 사용자 입력(최대 3개 키워드)에 따라 창의적 변화 아이디어 생성
+##############################################################################
+def generate_transformations(text1, text2=None, text3=None):
+    if text2 and text3:
+        results = generate_three_objects_interaction(text1, text2, text3)
+        objects = [text1, text2, text3]
+    elif text2:
+        results = generate_two_objects_interaction(text1, text2)
+        objects = [text1, text2]
+    else:
+        results = generate_single_object_transformations(text1)
+        objects = [text1]
+    
+    return enhance_descriptions(results, objects)
+
+##############################################################################
+# 결과 포맷팅
+##############################################################################
+def format_results(results):
+    formatted = ""
+    for category, result in results.items():
+        formatted += f"## {category}\n**기본 아이디어**: {result['base']}\n\n**확장된 아이디어**: {result['enhanced']}\n\n---\n\n"
+    return formatted
+
+##############################################################################
+# Gradio UI에서 호출할 함수
+##############################################################################
+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
+    
+    if not text1:
+        return "오류: 최소 하나의 키워드를 입력해주세요."
+    
+    keyword_info = f"키워드: {text1}"
+    if text2:
+        keyword_info += f", {text2}"
+    if text3:
+        keyword_info += f", {text3}"
+    
+    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 키 경고 메시지
+##############################################################################
+def get_warning_message():
+    if not GEMINI_API_KEY:
+        return "⚠️ 환경 변수 GEMINI_API_KEY가 설정되지 않았습니다. Gemini API 키를 설정하세요."
+    return ""
+
+##############################################################################
+# Gradio UI
+##############################################################################
+with gr.Blocks(title="키워드 기반 창의적 변화 아이디어 생성기",
+               theme=gr.themes.Soft(primary_hue="teal", secondary_hue="slate", neutral_hue="neutral")) as demo:
+    
+    gr.HTML("""
+    <style>
+    body { background: linear-gradient(135deg, #e0eafc, #cfdef3); font-family: 'Arial', sans-serif; }
+    .gradio-container { padding: 20px; }
+    h1, h2 { text-align: center; }
+    h1 { color: #333; }
+    h2 { color: #555; }
+    .output { background-color: #ffffff; padding: 15px; border-radius: 8px; }
+    .gr-button { background-color: #4CAF50; color: white; border: none; border-radius: 4px; padding: 8px 16px; }
+    .progress-message { color: #2196F3; font-weight: bold; margin-top: 10px; }
+    </style>
+    """)
+    
+    gr.Markdown("# 🚀 키워드 기반 창의적 변화 아이디어 생성기")
+    gr.Markdown("입력한 **키워드**(최대 3개)와 **카테고리**를 바탕으로, **창의적인 모델/컨셉/형상 변화**에 대한 이해와 **혁신 포인트**, **기능성** 등을 중심으로 확장된 아이디어를 제시합니다.")
+    
+    warning = gr.Markdown(get_warning_message())
+    
+    # 좌측 입력 영역
+    with gr.Row():
+        with gr.Column(scale=1):
+            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>
+                <p style="margin-left: 10px; font-weight: bold; color: #3498db;">처리 중입니다...</p>
+            </div>
+            <style>
+            @keyframes spin {
+                0% { transform: rotate(0deg); }
+                100% { transform: rotate(360deg); }
+            }
+            </style>
+            """, visible=False)
+            
+            submit_button = gr.Button("아이디어 생성하기", variant="primary")
+        
+        # 우측 출력 영역
+        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, 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, category_dropdown],
+        outputs=idea_output
+    ).then(
+        fn=hide_processing_indicator,
+        inputs=None,
+        outputs=processing_indicator
+    )
+
 if __name__ == "__main__":
     demo.launch(debug=True)