Update app.py

app.py

@@ -114,34 +114,49 @@ class GradioClientController:
     def _parse_dropdown_result(self, result):
         """API 응답에서 실제 선택지 데이터만 추출"""
         try:
+            logger.info(f"응답 파싱 시작, 응답 타입: {type(result)}")
+            
             if isinstance(result, (list, tuple)):
                 parsed_choices = []
-                for item in result:
+                for i, item in enumerate(result):
+                    logger.info(f"항목 {i}: 타입={type(item)}, 값={item}")
+                    
                     if isinstance(item, dict):
                         # Gradio 응답에서 choices 추출
                         if 'choices' in item:
                             choices = item['choices']
-                        elif '__type__' in item and 'choices' in str(item):
-                            # __type__ 구조에서 choices 찾기
-                            choices = item.get('choices', [])
+                            logger.info(f"dict에서 choices 추출: {choices}")
+                        elif '__type__' in item:
+                            # __type__ 구조 분석
+                            logger.info(f"__type__ 구조 발견: {item}")
+                            # 일반적으로 Gradio에서는 choices가 별도 필드에 있음
+                            choices = []
                         else:
-                            # 직접 리스트인 경우
-                            choices = item if isinstance(item, list) else []
+                            # dict의 다른 값들 확인
+                            choices = []
+                            for key, value in item.items():
+                                if isinstance(value, list) and key != '__type__':
+                                    choices = value
+                                    break
                     elif isinstance(item, list):
                         # 이미 리스트인 경우
                         choices = item
+                        logger.info(f"리스트 형태 choices: {choices}")
                     else:
                         # 문자열이거나 기타 타입
                         choices = [str(item)] if item else []
+                        logger.info(f"기타 타입을 문자열로 변환: {choices}")
                     
                     parsed_choices.append(choices)
                 
+                logger.info(f"파싱 완료: {parsed_choices}")
                 return parsed_choices
             else:
                 logger.warning(f"예상하지 못한 응답 형식: {type(result)}")
                 return [[] for _ in range(7)]
         except Exception as e:
             logger.error(f"드롭다운 결과 파싱 오류: {str(e)}")
+            logger.error(traceback.format_exc())
             return [[] for _ in range(7)]
 
     def get_initial_simple_choices(self) -> list:
@@ -194,7 +209,7 @@ class GradioClientController:
                     result = self.background_options[bg_type]
                     logger.info(f"캐시된 드롭다운 데이터 사용: {bg_type}")
                 else:
-                    return tuple([gr.update() for _ in range(7)])
+                    return tuple([gr.update(visible=False) for _ in range(7)])
             else:
                 # 실시간으로 API 호출
                 result = self.client.predict(bg_type, api_name="/update_dropdowns")
@@ -214,45 +229,35 @@ class GradioClientController:
                 
                 for i, choices in enumerate(result[:7]):
                     is_visible = (bg_type == visibility_map[i])
-                    first_choice = choices[0] if choices and len(choices) > 0 else None
-                    
-                    logger.info(f"드롭다운 {i} ({visibility_map[i]}): visible={is_visible}, choices_count={len(choices) if choices else 0}, first_choice={first_choice}")
                     
-                    updates.append(gr.update(
-                        visible=is_visible, 
-                        choices=choices if choices else [], 
-                        value=first_choice
-                    ))
+                    # 선택지가 있을 때만 첫 번째 값을 기본값으로 설정
+                    if is_visible and choices and len(choices) > 0:
+                        first_choice = choices[0]
+                        logger.info(f"드롭다운 {i} ({visibility_map[i]}): visible=True, choices={len(choices)}개, default='{first_choice}'")
+                        updates.append(gr.update(
+                            visible=True, 
+                            choices=choices, 
+                            value=first_choice,
+                            interactive=True
+                        ))
+                    else:
+                        logger.info(f"드롭다운 {i} ({visibility_map[i]}): visible=False")
+                        updates.append(gr.update(
+                            visible=False, 
+                            choices=[], 
+                            value=None,
+                            interactive=True
+                        ))
                 
                 return tuple(updates)
             else:
                 logger.warning(f"API에서 예상과 다른 형식의 결과를 반환했습니다: {type(result)}, length={len(result) if hasattr(result, '__len__') else 'N/A'}")
-                return tuple([gr.update() for _ in range(7)])
+                return tuple([gr.update(visible=False) for _ in range(7)])
             
         except Exception as e:
             logger.error(f"드롭다운 업데이트 오류: {str(e)}")
             logger.error(traceback.format_exc())
-            # 캐시된 데이터 시도
-            if bg_type in self.background_options:
-                try:
-                    result = self.background_options[bg_type]
-                    updates = []
-                    visibility_map = ["심플 배경", "스튜디오 배경", "자연 환경", "실내 환경", "특수배경", "주얼리", "특수효과"]
-                    
-                    for i, choices in enumerate(result[:7]):
-                        is_visible = (bg_type == visibility_map[i])
-                        updates.append(gr.update(
-                            visible=is_visible, 
-                            choices=choices if choices else [], 
-                            value=choices[0] if choices and len(choices) > 0 else None
-                        ))
-                    
-                    logger.info(f"캐시된 데이터로 드롭다운 업데이트: {bg_type}")
-                    return tuple(updates)
-                except:
-                    pass
-                    
-            return tuple([gr.update() for _ in range(7)])
+            return tuple([gr.update(visible=False) for _ in range(7)])
     
     def generate_prompt_only(self, password: str, bg_type: str, simple: str, studio: str, 
                            nature: str, indoor: str, special: str, jewelry: str, 
@@ -514,12 +519,13 @@ def create_gradio_interface():
                     info = gr.Textbox(label="처리 정보", interactive=False)
                     error = gr.Textbox(label="오류 메시지", interactive=False, visible=True)
             
-            # 드롭다운 변경 이벤트
+            # 드롭다운 변경 이벤트 - 즉시 업데이트
             background_type.change(
                 fn=controller.update_dropdowns,
                 inputs=[background_type],
                 outputs=[simple_dropdown, studio_dropdown, nature_dropdown, 
-                        indoor_dropdown, special_dropdown, jewelry_dropdown, special_effects_dropdown]
+                        indoor_dropdown, special_dropdown, jewelry_dropdown, special_effects_dropdown],
+                queue=False  # 큐를 사용하지 않아 더 빠른 응답
             )
             
             # 프롬프트 생성 버튼 클릭 이벤트