Update app.py

app.py

@@ -212,11 +212,14 @@ def wrapper_modified(keyword, korean_only, apply_main_keyword_option, exclude_ze
             api_name="/process_search_results"
         )
         
-        # API 응답 확인 및 처리
+        # API 응답 확인 및 처리 (6개 값)
         logger.info(f"API 응답 타입: {type(result)}, 길이: {len(result) if isinstance(result, (list, tuple)) else 'N/A'}")
         
-        if isinstance(result, (list, tuple)) and len(result) >= 5:
+        if isinstance(result, (list, tuple)) and len(result) >= 6:
+            table_html, cat_choices, vol_choices, selected_cat, download_file, extra = result[:6]
+        elif isinstance(result, (list, tuple)) and len(result) >= 5:
             table_html, cat_choices, vol_choices, selected_cat, download_file = result[:5]
+            extra = None
         else:
             # 응답이 예상과 다른 경우 기본값 사용
             logger.warning(f"예상과 다른 API 응답: {result}")
@@ -226,6 +229,23 @@ def wrapper_modified(keyword, korean_only, apply_main_keyword_option, exclude_ze
             selected_cat = "전체 보기"
             download_file = None
         
+        # choices 형식 처리 (중첩 리스트인 경우 첫 번째 값만 사용)
+        if isinstance(cat_choices, dict) and 'choices' in cat_choices:
+            cat_choices = [choice[0] if isinstance(choice, list) else choice for choice in cat_choices['choices']]
+        elif isinstance(cat_choices, list) and cat_choices and isinstance(cat_choices[0], list):
+            cat_choices = [choice[0] for choice in cat_choices]
+        
+        if isinstance(vol_choices, dict) and 'choices' in vol_choices:
+            vol_choices = [choice[0] if isinstance(choice, list) else choice for choice in vol_choices['choices']]
+        elif isinstance(vol_choices, list) and vol_choices and isinstance(vol_choices[0], list):
+            vol_choices = [choice[0] for choice in vol_choices]
+        
+        # selected_cat 처리
+        if isinstance(selected_cat, dict) and 'value' in selected_cat:
+            selected_cat = selected_cat['value']
+        elif isinstance(selected_cat, list):
+            selected_cat = selected_cat[0] if selected_cat else "전체 보기"
+        
         local_file = None
         if download_file:
             local_file = create_session_temp_file(session_id, '.xlsx')