Update app.py

app.py

@@ -42,6 +42,7 @@ def choose_alternative(transformation):
     else:
         return random.choice([left, right])
 
+
 # 창의적인 모델/컨셉/형상 변화 아이디어를 위한 카테고리 (기존 '물리적 변화' 사전 유지)
 physical_transformation_categories = {
     "공간 이동": [
@@ -184,6 +185,118 @@ physical_transformation_categories = {
     ]
 }
 
+import os
+import gradio as gr
+import random
+import time
+import logging
+import google.generativeai as genai
+
+# 로깅 설정
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.FileHandler("api_debug.log"),
+        logging.StreamHandler()
+    ]
+)
+logger = logging.getLogger("idea_generator")
+
+# Gemini API 키는 환경 변수 GEMINI_API_KEY에서 가져옵니다.
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+genai.configure(api_key=GEMINI_API_KEY)
+
+# 슬래시("/")가 포함된 변환 문자열에서 두 옵션 중 하나만 선택하는 헬퍼 함수
+def choose_alternative(transformation):
+    if "/" not in transformation:
+        return transformation
+    parts = transformation.split("/")
+    if len(parts) != 2:
+        return random.choice([part.strip() for part in parts])
+    left = parts[0].strip()
+    right = parts[1].strip()
+    if " " in left:
+        tokens = left.split(" ", 1)
+        prefix = tokens[0]
+        if not right.startswith(prefix):
+            option1 = left
+            option2 = prefix + " " + right
+        else:
+            option1 = left
+            option2 = right
+        return random.choice([option1, option2])
+    else:
+        return random.choice([left, right])
+
+# 창의적인 모델/컨셉/형상 변화 아이디어를 위한 카테고리 (기존 '물리적 변화' 사전 유지)
+physical_transformation_categories = {
+    "공간 이동": [
+        "앞/뒤 이동", "좌/우 이동", "위/아래 이동", "세로축 회전(고개 끄덕임)", 
+        "가로축 회전(고개 젓기)", "길이축 회전(옆으로 기울임)", "원 운동", "나선형 이동", 
+        "관성에 의한 미끄러짐", "회전축 변화", "불규칙 회전", "흔들림 운동", "포물선 이동",
+        "무중력 부유", "수면 위 부유", "점프/도약", "슬라이딩", "롤링", "자유 낙하",
+        "왕복 운동", "탄성 튕김", "관통", "회피 움직임", "지그재그 이동", "스윙 운동"
+    ],
+    
+    # [기존 카테고리들 유지]
+    # ... 다른 모든 카테고리들은 원래 코드와 동일하게 유지 ...
+    
+    "크기와 형태 변화": [
+        "부피 늘어남/줄어듦", "길이 늘어남/줄어듦", "너비 늘어남/줄어듦", "높이 늘어남/줄어듦",
+        "밀도 변화", "무게 증가/감소", "모양 변형", "상태 변화", "불균등 변형",
+        "복잡한 형태 변형", "비틀림/꼬임", "불균일한 확장/축소", "모서리 둥글게/날카롭게",
+        "깨짐/갈라짐", "여러 조각 나눠짐", "물 저항", "먼지 저항", "찌그러짐/복원",
+        "접힘/펼쳐짐", "압착/팽창", "늘어남/수축", "구겨짐/평평해짐", "뭉개짐/단단해짐",
+        "말림/펴짐", "꺾임/구부러짐"
+    ],
+    
+    "표면 및 외관 변화": [
+        "색상 변화", "질감 변화", "투명/불투명 변화", "반짝임/무광 변화", 
+        "빛 반사 정도 변화", "무늬 변화", "각도에 따른 색상 변화", "빛에 따른 색상 변화", 
+        "온도에 따른 색상 변화", "홀로그램 효과", "표면 각도별 빛 반사", "표면 모양 변형", 
+        "초미세 표면 구조 변화", "자가 세정 효과", "얼룩/패턴 생성", "흐림/선명함 변화",
+        "광택/윤기 변화", "색조/채도 변화", "발광/형광", "빛 산란 효과", 
+        "빛 흡수 변화", "반투명 효과", "그림자 효과 변화", "자외선 반응 변화",
+        "야광 효과"
+    ],
+    
+    "물질의 상태 변화": [
+        "고체/액체/기체 전환", "결정화/용해", "산화/부식", "딱딱해짐/부드러워짐", 
+        "특수 상태 전환", "무정형/결정형 전환", "성분 분리", "미세 입자 형성/분해", 
+        "젤 형성/풀어짐", "준안정 상태 변화", "분자 자가 정렬/분해", "상태변화 지연 현상", 
+        "녹음", "굳음", "증발/응축", "승화/증착", "침전/부유", "분산/응집",
+        "건조/습윤", "팽윤/수축", "동결/해동", "풍화/침식", "충전/방전",
+        "결합/분리", "발효/부패"
+    ],
+    
+    "열 관련 변화": [
+        "온도 상승/하강", "열에 의한 팽창/수축", "열 전달/차단", "압력 상승/하강", 
+        "열 변화에 따른 자화", "무질서도 변화", "열전기 현상", "자기장에 의한 열 변화", 
+        "상태변화 중 열 저장/방출", "열 스트레스 발생/해소", "급격한 온도 변화 영향", 
+        "복사열에 의한 냉각/가열", "발열/흡열", "열 분포 변화", "열 반���/흡수",
+        "냉각 응축", "열 활성화", "열 변색", "열 팽창 계수 변화", "열 안정성 변화",
+        "내열성/내한성", "자기발열", "열적 평형/불균형", "열적 변형", "열 분산/집중"
+    ],
+    
+    "움직임 특성 변화": [
+        "가속/감속", "일정
+    # 축약된 부분, 실제 코드에서는 모든 카테고리를 유지해야 합니다
+    # 다른 모든 카테고리들은 원래 코드와 동일하게 유지해야 합니다
+    "센서 기능": [
+        "시각 센서/감지", "청각 센서/감지", "촉각 센서/감지", "미각 센서/감지", "후각 센서/감지",
+        "온도 센서/감지", "습도 센서/감지", "압력 센서/감지", "가속도 센서/감지", "회전 센서/감지",
+        "근접 센서/감지", "위치 센서/감지", "운동 센서/감지", "가스 센서/감지", "적외선 센서/감지",
+        "자외선 센서/감지", "방사선 센서/감지", "자기장 센서/감지", "전기장 센서/감지", "화학물질 센서/감지",
+        "생체신호 센서/감지", "진동 센서/감지", "소음 센서/감지", "빛 세기 센서/감지", "빛 파장 센서/감지",
+        "기울기 센서/감지", "pH 센서/감지", "전류 센서/감지", "전압 센서/감지", "이미지 센서/감지",
+        "거리 센서/감지", "깊이 센서/감지", "중력 센서/감지", "속도 센서/감지", "흐름 센서/감지",
+        "수위 센서/감지", "탁도 센서/감지", "염도 센서/감지", "금속 감지", "압전 센서/감지", 
+        "광전 센서/감지", "열전대 센서/감지", "홀 효과 센서/감지", "초음파 센서/감지", "레이더 센서/감지",
+        "라이다 센서/감지", "터치 센서/감지", "제스처 센서/감지", "심박 센서/감지", "혈압 센서/감지"
+    ]
+}
+
 ##############################################################################
 # Gemini API 호출 함수 (예: gemini-2.0-flash-thinking-exp-01-21 -> 다른 모델 사용 시 수정)
 ##############################################################################
@@ -281,16 +394,31 @@ def generate_three_objects_interaction(obj1, obj2, obj3):
 ##############################################################################
 # 생성된 기본 설명을 LLM을 통해 확장
 ##############################################################################
-def enhance_descriptions(results, objects):
+def enhance_descriptions(results, objects, progress=None):
     obj_name = " 및 ".join([obj for obj in objects if obj])
-    for category, result in results.items():
+    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}' 처리 중...")
+        
         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):
+def generate_transformations(text1, text2=None, text3=None, progress=None):
+    if progress:
+        progress(0.1, "아이디어 기본 구조 생성 중...")
+    
     if text2 and text3:
         results = generate_three_objects_interaction(text1, text2, text3)
         objects = [text1, text2, text3]
@@ -300,7 +428,11 @@ def generate_transformations(text1, text2=None, text3=None):
     else:
         results = generate_single_object_transformations(text1)
         objects = [text1]
-    return enhance_descriptions(results, objects)
+    
+    if progress:
+        progress(0.2, "Gemini API를 통한 아이디어 확장 중...")
+    
+    return enhance_descriptions(results, objects, progress)
 
 ##############################################################################
 # 결과 포맷팅
@@ -314,29 +446,33 @@ def format_results(results):
 ##############################################################################
 # Gradio UI에서 호출할 함수
 ##############################################################################
-def process_inputs(text1, text2, text3):
-    messages = []
-    messages.append("입력값 확인 중...")
-    time.sleep(0.3)
+def process_inputs(text1, text2, text3, 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:
-        messages.append("오류: 최소 하나의 키워드를 입력해주세요.")
-        return "\n\n".join(messages)
+        return "오류: 최소 하나의 키워드를 입력해주세요."
     
-    messages.append("창의적인 모델/컨셉/형상 변화 아이디어 생성 중...")
-    time.sleep(0.3)
-    results = generate_transformations(text1, text2, text3)
+    progress(0.05, "아이디어 생성 준비 중...")
     
-    messages.append("결과 포맷팅 중...")
-    time.sleep(0.3)
-    formatted = format_results(results)
+    # 키워드 정보 확인
+    keyword_info = f"키워드: {text1}"
+    if text2:
+        keyword_info += f", {text2}"
+    if text3:
+        keyword_info += f", {text3}"
+    
+    progress(0.1, "창의적인 모델/컨셉/형상 변화 아이디어 생성 시작...")
     
-    messages.append("완료!")
-    messages.append(formatted)
+    # 결과 생성
+    results = generate_transformations(text1, text2, text3, progress)
     
-    return "\n\n".join(messages)
+    progress(0.95, "결과 포맷팅 중...")
+    formatted = format_results(results)
+    
+    progress(1.0, "완료!")
+    return formatted
 
 ##############################################################################
 # API 키 경고 메시지
@@ -361,6 +497,7 @@ with gr.Blocks(title="키워드 기반 창의적 변화 아이디어 생성기",
     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>
     """)
     
@@ -375,12 +512,15 @@ 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="예: 헬스케어")
-            submit_button = gr.Button("아이디어 생성하기")
+            
+            # 상태 메시지를 표시할 영역 추가
+            status_msg = gr.Markdown("💡 '아이디어 생성하기' 버튼을 클릭하면 프로그레스 바가 나타납니다.")
+            
+            submit_button = gr.Button("아이디어 생성하기", variant="primary")
         
-        # 우측 출력 영역 (탭은 하나만 사용)
+        # 우측 출력 영역
         with gr.Column(scale=2):
-            with gr.TabItem("창의적인 모델/컨셉/형상 변화 아이디어", id="creative_tab"):
-                idea_output = gr.Markdown(label="아이디어 결과")
+            idea_output = gr.Markdown(label="아이디어 결과")
 
     # 예시 입력
     gr.Examples(
@@ -393,10 +533,12 @@ with gr.Blocks(title="키워드 기반 창의적 변화 아이디어 생성기",
         inputs=[text_input1, text_input2, text_input3],
     )
     
-    # 버튼 이벤트: 첫 번째 탭에만 연결
-    submit_button.click(fn=process_inputs,
-                        inputs=[text_input1, text_input2, text_input3],
-                        outputs=idea_output)
+    # 버튼 이벤트
+    submit_button.click(
+        fn=process_inputs,
+        inputs=[text_input1, text_input2, text_input3],
+        outputs=idea_output
+    )
 
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    demo.launch(debug=True)