Update app.py

app.py

@@ -112,83 +112,31 @@ class GradioClientController:
             return tuple([[] for _ in range(7)])
 
     def _parse_dropdown_result(self, result):
-        """API 응답에서 실제 선택지 데이터만 추출"""
+        """API 응답에서 실제 선택지 데이터만 추출 - 단순화"""
         try:
-            logger.info(f"응답 파싱 시작, 응답 타입: {type(result)}")
+            logger.info(f"응답 파싱: {result}")
             
-            if isinstance(result, (list, tuple)):
-                parsed_choices = []
-                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']
-                            logger.info(f"dict에서 choices 추출: {choices}")
-                        elif '__type__' in item:
-                            # __type__ 구조 분석
-                            logger.info(f"__type__ 구조 발견: {item}")
-                            # 일반적으로 Gradio에서는 choices가 별도 필드에 있음
-                            choices = []
-                        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
+            # API 문서에 따르면 tuple of 7 elements를 반환
+            # 각 요소는 Literal 타입의 리스트여야 함
+            if isinstance(result, (list, tuple)) and len(result) >= 7:
+                # 결과를 그대로 반환 (API가 이미 올바른 형식으로 반환한다고 가정)
+                return result[:7]
             else:
-                logger.warning(f"예상하지 못한 응답 형식: {type(result)}")
+                logger.warning(f"예상하지 못한 응답 형식: {type(result)}, length={len(result) if hasattr(result, '__len__') else 'N/A'}")
                 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:
-        """심플 배경 초기 선택지만 반환"""
-        try:
-            logger.info("심플 배경 선택지 로드 시작")
-            
-            if "심플 배경" in self.background_options:
-                result = self.background_options["심플 배경"]
-                logger.info("캐시에서 심플 배경 선택지 로드")
-            elif self.client:
-                logger.info("API에서 심플 배경 선택지 로드")
-                result = self.client.predict("심플 배경", api_name="/update_dropdowns")
-                logger.info(f"API 응답: {result}")
-                
-                # 응답 파싱
-                parsed_result = self._parse_dropdown_result(result)
-                self.background_options["심플 배경"] = parsed_result
-                
-                if len(parsed_result) > 0:
-                    choices = parsed_result[0]
-                    logger.info(f"파싱된 심플 배경 선택지: {choices}")
-                    return choices
-            else:
-                logger.error("클라이언트가 초기화되지 않음")
-                return []
-                
-        except Exception as e:
-            logger.error(f"심플 배경 선택지 로드 실패: {str(e)}")
-            logger.error(traceback.format_exc())
-        
-        return []
+        """심플 배경 초기 선택지 - 하드코딩으로 테스트"""
+        # API 문서에서 확인된 실제 선택지들
+        return [
+            '화이트 기본', '회색 투톤', '라이트 그레이', '그레이 그라데이션 스포트라이트', 
+            '프리미엄 드라마틱 블랙', '딥블루 유리반사', '파스텔 그라데이션', '스카이블루 파스텔', 
+            '버터옐로우 파스텔', '블루 원색', '레드 원색'
+        ]
     
     def _ensure_client(self) -> bool:
         """클라이언트 연결 상태 확인 및 재연결"""
@@ -198,31 +146,19 @@ class GradioClientController:
         return self.client is not None
     
     def update_dropdowns(self, bg_type: str) -> Tuple:
-        """배경 유형에 따른 드롭다운 업데이트"""
+        """배경 유형에 따른 드롭다운 업데이트 - 단순화"""
         try:
-            logger.info(f"드롭다운 업데이트 시작: {bg_type}")
+            logger.info(f"드롭다운 업데이트: {bg_type}")
             
             if not self._ensure_client():
                 logger.error("클라이언트 연결 실패")
-                # 캐시된 데이터가 있으면 사용
-                if bg_type in self.background_options:
-                    result = self.background_options[bg_type]
-                    logger.info(f"캐시된 드롭다운 데이터 사용: {bg_type}")
-                else:
-                    return tuple([gr.update(visible=False) for _ in range(7)])
-            else:
-                # 실시간으로 API 호출
-                result = self.client.predict(bg_type, api_name="/update_dropdowns")
-                logger.info(f"API 응답 원본: {result}")
-                
-                # 응답 파싱
-                parsed_result = self._parse_dropdown_result(result)
-                self.background_options[bg_type] = parsed_result
-                result = parsed_result
-                
-                logger.info(f"파싱된 결과: {result}")
+                return tuple([gr.update(visible=False) for _ in range(7)])
+            
+            # API 호출
+            result = self.client.predict(bg_type, api_name="/update_dropdowns")
+            logger.info(f"API 응답: {result}")
             
-            # 결과를 Gradio 업데이트 형식으로 변환
+            # API 응답을 그대로 사용
             if isinstance(result, (list, tuple)) and len(result) >= 7:
                 updates = []
                 visibility_map = ["심플 배경", "스튜디오 배경", "자연 환경", "실내 환경", "특수배경", "주얼리", "특수효과"]
@@ -230,33 +166,24 @@ class GradioClientController:
                 for i, choices in enumerate(result[:7]):
                     is_visible = (bg_type == visibility_map[i])
                     
-                    # 선택지가 있을 때만 첫 번째 값을 기본값으로 설정
-                    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}'")
+                    if is_visible and isinstance(choices, list) and len(choices) > 0:
+                        logger.info(f"{visibility_map[i]} 드롭다운: {len(choices)}개 선택지")
                         updates.append(gr.update(
                             visible=True, 
                             choices=choices, 
-                            value=first_choice,
+                            value=choices[0],  # 첫 번째 항목 선택
                             interactive=True
                         ))
                     else:
-                        logger.info(f"드롭다운 {i} ({visibility_map[i]}): visible=False")
-                        updates.append(gr.update(
-                            visible=False, 
-                            choices=[], 
-                            value=None,
-                            interactive=True
-                        ))
+                        updates.append(gr.update(visible=False))
                 
                 return tuple(updates)
             else:
-                logger.warning(f"API에서 예상과 다른 형식의 결과를 반환했습니다: {type(result)}, length={len(result) if hasattr(result, '__len__') else 'N/A'}")
+                logger.error(f"잘못된 API 응답: {result}")
                 return tuple([gr.update(visible=False) for _ in range(7)])
             
         except Exception as e:
             logger.error(f"드롭다운 업데이트 오류: {str(e)}")
-            logger.error(traceback.format_exc())
             return tuple([gr.update(visible=False) for _ in range(7)])
     
     def generate_prompt_only(self, password: str, bg_type: str, simple: str, studio: str, 
@@ -412,9 +339,10 @@ def create_gradio_interface():
                                 value="심플 배경"
                             )
                             
-                            # 드롭다운 컴포넌트들 - 초기에는 빈 상태로 시작
+                            # 드롭다운 컴포넌트들 - 심플 배경은 하드코딩된 선택지로 시작
                             simple_dropdown = gr.Dropdown(
-                                choices=[],
+                                choices=controller.get_initial_simple_choices(),
+                                value="화이트 기본",  # 기본값 설정
                                 label="심플 배경 선택",
                                 visible=True,
                                 interactive=True