Update app.py

app.py

@@ -17,6 +17,18 @@ from dotenv import load_dotenv
 load_dotenv()
 
 # 로깅 설정 (API 엔드포인트 정보는 제외)
+class SecureFilter(logging.Filter):
+    """API 엔드포인트 정보를 로그에서 제거하는 필터"""
+    def filter(self, record):
+        # API 엔드포인트 관련 로그 필터링
+        message = record.getMessage()
+        if any(keyword in message.lower() for keyword in ['hf.space', 'happydoggg', 'gradio_api', 'http request']):
+            return False
+        return True
+
+# 보안 필터 적용
+secure_filter = SecureFilter()
+
 logging.basicConfig(
     level=logging.INFO,
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
@@ -27,6 +39,11 @@ logging.basicConfig(
 )
 
 logger = logging.getLogger(__name__)
+logger.addFilter(secure_filter)
+
+# HTTP 요청 로그도 숨김
+logging.getLogger("httpx").setLevel(logging.WARNING)
+logging.getLogger("gradio_client").setLevel(logging.WARNING)
 
 # API 클라이언트 초기화
 def get_api_client():
@@ -123,8 +140,14 @@ def get_app_temp_dir():
     return os.environ.get('CONTROL_TOWER_TEMP', tempfile.gettempdir())
 
 def get_session_id():
-    """세션 ID 생성"""
-    return str(uuid.uuid4())
+    """세션 ID 생성 - 원본 API와 동일"""
+    try:
+        client = get_api_client()
+        result = client.predict(api_name="/get_session_id")
+        return result
+    except Exception as e:
+        logger.error(f"세션 ID 생성 API 호출 오류: {e}")
+        return str(uuid.uuid4())
 
 def cleanup_session_files(session_id, delay=300):
     """세션별 임시 파일 정리 함수"""
@@ -199,99 +222,92 @@ def create_session_temp_file(session_id, suffix='.xlsx'):
     register_session_file(session_id, temp_file_path)
     return temp_file_path
 
-def wrapper_modified(keyword, korean_only, apply_main_keyword_option, exclude_zero_volume, session_id):
-    """키워드 검색 및 처리 래퍼 함수 (API 클라이언트 사용)"""
-    update_session_activity(session_id)
-    
+# ========== 원본 API와 정확히 동일한 함수들 ==========
+
+def search_with_loading(keyword, korean_only, apply_main_keyword, exclude_zero_volume):
+    """원본 API: /search_with_loading (4개 매개변수, 1개 반환값)"""
+    try:
+        client = get_api_client()
+        result = client.predict(
+            keyword=keyword,
+            korean_only=korean_only,
+            apply_main_keyword=apply_main_keyword,
+            exclude_zero_volume=exclude_zero_volume,
+            api_name="/search_with_loading"
+        )
+        return result
+    except Exception as e:
+        logger.error(f"search_with_loading API 호출 오류: {e}")
+        return gr.update(visible=True)
+
+def process_search_results(keyword, korean_only, apply_main_keyword, exclude_zero_volume):
+    """원본 API: /process_search_results (4개 매개변수, 6개 반환값)"""
     try:
         client = get_api_client()
-        
-        # API 호출
         result = client.predict(
             keyword=keyword,
             korean_only=korean_only,
-            apply_main_keyword=apply_main_keyword_option,
+            apply_main_keyword=apply_main_keyword,
             exclude_zero_volume=exclude_zero_volume,
             api_name="/process_search_results"
         )
         
-        # 결과 처리
-        table_html, cat_choices, vol_choices, selected_cat, download_file = result
+        # 원본 API 문서에 따르면 6개 반환값이지만 실제로는 5개일 수 있음
+        # 안전하게 처리
+        if len(result) == 6:
+            table_html, cat_choices, vol_choices, selected_cat, download_file, progress_html = result
+        elif len(result) == 5:
+            table_html, cat_choices, vol_choices, selected_cat, download_file = result
+            progress_html = ""
+        else:
+            logger.error(f"예상과 다른 반환값 개수: {len(result)}")
+            return (
+                "<p>서비스 응답 형식에 문제가 발생했습니다.</p>",
+                ["전체 보기"], ["전체"], "전체 보기", None, ""
+            )
         
         # 다운로드 파일이 있는 경우 로컬로 복사
-        excel_path = None
+        local_download_file = None
         if download_file:
-            excel_path = create_session_temp_file(session_id, '.xlsx')
-            # 원격 파일을 로컬로 복사
+            session_id = get_session_id()
+            local_download_file = create_session_temp_file(session_id, '.xlsx')
             try:
-                import shutil
-                shutil.copy2(download_file, excel_path)
+                shutil.copy2(download_file, local_download_file)
             except Exception as e:
                 logger.error(f"파일 복사 오류: {e}")
-                excel_path = None
+                local_download_file = None
         
-        # 가상의 DataFrame 상태 (실제로는 사용되지 않음)
-        df_state = pd.DataFrame()
+        return table_html, cat_choices, vol_choices, selected_cat, local_download_file, progress_html
         
-        if table_html and "검색 결과가 없습니다" not in table_html:
-            return (gr.update(value=table_html), gr.update(choices=cat_choices), 
-                   gr.update(choices=vol_choices), df_state, 
-                   gr.update(choices=cat_choices, value=selected_cat), excel_path,
-                   gr.update(visible=True), gr.update(visible=True), keyword)
-        else:
-            return (gr.update(value="<p>검색 결과가 없습니다. 다른 키워드로 시도해보세요.</p>"), 
-                   gr.update(choices=["전체 보기"]), gr.update(choices=["전체"]), 
-                   df_state, gr.update(choices=["전체 보기"], value="전체 보기"), None,
-                   gr.update(visible=False), gr.update(visible=False), keyword)
-                   
     except Exception as e:
         logger.error(f"API 호출 오류: {e}")
-        return (gr.update(value="<p>서비스 연결에 문제가 발생했습니다. 잠시 후 다시 시도해주세요.</p>"), 
-               gr.update(choices=["전체 보기"]), gr.update(choices=["전체"]), 
-               pd.DataFrame(), gr.update(choices=["전체 보기"], value="전체 보기"), None,
-               gr.update(visible=False), gr.update(visible=False), "")
-
-def analyze_with_auto_download(analysis_keywords, selected_category, state_df, session_id):
-    """카테고리 일치 분석 실행 및 자동 다운로드 (API 클라이언트 사용)"""
-    update_session_activity(session_id)
-    
+        return (
+            "<p>서비스 연결에 문제가 발생했습니다. 잠시 후 다시 시도해주세요.</p>",
+            ["전체 보기"], ["전체"], "전체 보기", None, ""
+        )
+
+def filter_and_sort_table(selected_cat, keyword_sort, total_volume_sort, usage_count_sort, selected_volume_range, exclude_zero_volume):
+    """원본 API: /filter_and_sort_table (6개 매개변수, 1개 반환값)"""
     try:
         client = get_api_client()
-        
-        # API 호출
         result = client.predict(
-            analysis_keywords=analysis_keywords,
-            selected_category=selected_category,
-            api_name="/process_analyze_results"
+            selected_cat=selected_cat,
+            keyword_sort=keyword_sort,
+            total_volume_sort=total_volume_sort,
+            usage_count_sort=usage_count_sort,
+            selected_volume_range=selected_volume_range,
+            exclude_zero_volume=exclude_zero_volume,
+            api_name="/filter_and_sort_table"
         )
-        
-        analysis_result, download_file = result
-        
-        # 다운로드 파일이 있는 경우 로컬로 복사
-        excel_path = None
-        if download_file:
-            excel_path = create_session_temp_file(session_id, '.xlsx')
-            try:
-                import shutil
-                shutil.copy2(download_file, excel_path)
-            except Exception as e:
-                logger.error(f"분석 결과 파일 복사 오류: {e}")
-                excel_path = None
-        
-        return analysis_result, excel_path, gr.update(visible=True)
-        
+        return result
     except Exception as e:
-        logger.error(f"분석 API 호출 오류: {e}")
-        return "분석 서비스 연결에 문제가 발생했습니다. 잠시 후 다시 시도해주세요.", None, gr.update(visible=True)
+        logger.error(f"filter_and_sort_table API 호출 오류: {e}")
+        return "<p>필터링 서비스 연결에 문제가 발생했습니다.</p>"
 
-def filter_and_sort_table(df, selected_cat, keyword_sort, total_volume_sort, usage_count_sort, selected_volume_range, exclude_zero_volume, session_id):
-    """테이블 필터링 및 정렬 함수 (API 클라이언트 사용)"""
-    update_session_activity(session_id)
-    
+def filter_and_sort_table_1(selected_cat, keyword_sort, total_volume_sort, usage_count_sort, selected_volume_range, exclude_zero_volume):
+    """원본 API: /filter_and_sort_table_1"""
     try:
         client = get_api_client()
-        
-        # API 호출
         result = client.predict(
             selected_cat=selected_cat,
             keyword_sort=keyword_sort,
@@ -299,122 +315,222 @@ def filter_and_sort_table(df, selected_cat, keyword_sort, total_volume_sort, usa
             usage_count_sort=usage_count_sort,
             selected_volume_range=selected_volume_range,
             exclude_zero_volume=exclude_zero_volume,
-            api_name="/filter_and_sort_table"
+            api_name="/filter_and_sort_table_1"
         )
-        
         return result
-        
     except Exception as e:
-        logger.error(f"필터링 API 호출 오류: {e}")
+        logger.error(f"filter_and_sort_table_1 API 호출 오류: {e}")
         return "<p>필터링 서비스 연결에 문제가 발생했습니다.</p>"
 
-def update_category_selection(selected_cat, session_id):
-    """카테고리 필터 선택 시 분석할 카테고리도 같은 값으로 업데이트 (API 클라이언트 사용)"""
-    update_session_activity(session_id)
-    
+def filter_and_sort_table_2(selected_cat, keyword_sort, total_volume_sort, usage_count_sort, selected_volume_range, exclude_zero_volume):
+    """원본 API: /filter_and_sort_table_2"""
+    try:
+        client = get_api_client()
+        result = client.predict(
+            selected_cat=selected_cat,
+            keyword_sort=keyword_sort,
+            total_volume_sort=total_volume_sort,
+            usage_count_sort=usage_count_sort,
+            selected_volume_range=selected_volume_range,
+            exclude_zero_volume=exclude_zero_volume,
+            api_name="/filter_and_sort_table_2"
+        )
+        return result
+    except Exception as e:
+        logger.error(f"filter_and_sort_table_2 API 호출 오류: {e}")
+        return "<p>필터링 서비스 연결에 문제가 발생했습니다.</p>"
+
+def filter_and_sort_table_3(selected_cat, keyword_sort, total_volume_sort, usage_count_sort, selected_volume_range, exclude_zero_volume):
+    """원본 API: /filter_and_sort_table_3"""
+    try:
+        client = get_api_client()
+        result = client.predict(
+            selected_cat=selected_cat,
+            keyword_sort=keyword_sort,
+            total_volume_sort=total_volume_sort,
+            usage_count_sort=usage_count_sort,
+            selected_volume_range=selected_volume_range,
+            exclude_zero_volume=exclude_zero_volume,
+            api_name="/filter_and_sort_table_3"
+        )
+        return result
+    except Exception as e:
+        logger.error(f"filter_and_sort_table_3 API 호출 오류: {e}")
+        return "<p>필터링 서비스 연결에 문제가 발생했습니다.</p>"
+
+def filter_and_sort_table_4(selected_cat, keyword_sort, total_volume_sort, usage_count_sort, selected_volume_range, exclude_zero_volume):
+    """원본 API: /filter_and_sort_table_4"""
+    try:
+        client = get_api_client()
+        result = client.predict(
+            selected_cat=selected_cat,
+            keyword_sort=keyword_sort,
+            total_volume_sort=total_volume_sort,
+            usage_count_sort=usage_count_sort,
+            selected_volume_range=selected_volume_range,
+            exclude_zero_volume=exclude_zero_volume,
+            api_name="/filter_and_sort_table_4"
+        )
+        return result
+    except Exception as e:
+        logger.error(f"filter_and_sort_table_4 API 호출 오류: {e}")
+        return "<p>필터링 서비스 연결에 문제가 발생했습니다.</p>"
+
+def update_category_selection(selected_cat):
+    """원본 API: /update_category_selection (1개 매개변수, 1개 반환값)"""
     try:
         client = get_api_client()
-        
         result = client.predict(
             selected_cat=selected_cat,
             api_name="/update_category_selection"
         )
-        
         return gr.update(value=result)
-        
     except Exception as e:
-        logger.error(f"카테고리 선택 API 호출 오류: {e}")
+        logger.error(f"update_category_selection API 호출 오류: {e}")
         return gr.update(value=selected_cat)
 
-def reset_interface(session_id):
-    """인터페이스 리셋 함수 - 세션별 데이터 초기화"""
-    update_session_activity(session_id)
-    
-    # 세션별 임시 파일 정리
-    if session_id in session_temp_files:
-        for file_path in session_temp_files[session_id]:
+def analyze_with_loading(analysis_keywords, selected_category):
+    """원본 API: /analyze_with_loading (2개 매개변수, 1개 반환값)"""
+    try:
+        client = get_api_client()
+        result = client.predict(
+            analysis_keywords=analysis_keywords,
+            selected_category=selected_category,
+            api_name="/analyze_with_loading"
+        )
+        return result
+    except Exception as e:
+        logger.error(f"analyze_with_loading API 호출 오류: {e}")
+        return gr.update(visible=True)
+
+def process_analyze_results(analysis_keywords, selected_category):
+    """원본 API: /process_analyze_results (2개 매개변수, 2개 반환값)"""
+    try:
+        client = get_api_client()
+        result = client.predict(
+            analysis_keywords=analysis_keywords,
+            selected_category=selected_category,
+            api_name="/process_analyze_results"
+        )
+        
+        analysis_result, download_file = result
+        
+        # 다운로드 파일이 있는 경우 로컬로 복사
+        local_download_file = None
+        if download_file:
+            session_id = get_session_id()
+            local_download_file = create_session_temp_file(session_id, '.xlsx')
             try:
-                if os.path.exists(file_path):
-                    os.remove(file_path)
-                    logger.info(f"세션 {session_id[:8]}... 리셋 시 파일 삭제: {file_path}")
+                shutil.copy2(download_file, local_download_file)
             except Exception as e:
-                logger.error(f"세션 {session_id[:8]}... 리셋 시 파일 삭제 오류: {e}")
-        session_temp_files[session_id] = []
-    
+                logger.error(f"분석 결과 파일 복사 오류: {e}")
+                local_download_file = None
+        
+        return analysis_result, local_download_file
+        
+    except Exception as e:
+        logger.error(f"process_analyze_results API 호출 오류: {e}")
+        return "분석 서비스 연결에 문제가 발생했습니다. 잠시 후 다시 시도해주세요.", None
+
+def reset_interface():
+    """원본 API: /reset_interface (0개 매개변수, 16개 반환값)"""
     try:
         client = get_api_client()
-        
         result = client.predict(api_name="/reset_interface")
-        
         return result
-        
     except Exception as e:
-        logger.error(f"리셋 API 호출 오류: {e}")
-        # 기본 리셋 값 반환
+        logger.error(f"reset_interface API 호출 오류: {e}")
+        # 기본 리셋 값 반환 (16개)
         return (
             "", # 검색 키워드
             True, # 한글만 추출
             False, # 검색량 0 키워드 제외
             "메인키워드 적용", # 조합 방식
             "", # HTML 테이블
-            ["전체 보기"], # 카테고리 필터
-            "전체 보기", # 카테고리 필터 선택
-            ["전체"], # 검색량 구간 필터
-            "전체", # 검색량 구간 선택
+            ["전체 보기"], # 카테고리 필터 choices
+            "전체 보기", # 카테고리 필터 value
+            ["전체"], # 검색량 구간 필터 choices
+            "전체", # 검색량 구간 필터 value
             "정렬 없음", # 총검색량 정렬
             "정렬 없음", # 키워드 사용횟수 정렬
-            ["전체 보기"], # 분석할 카테고리
-            "전체 보기", # 분석할 카테고리 선택
+            ["전체 보기"], # 분석할 카테고리 choices
+            "전체 보기", # 분석할 카테고리 value
             "", # 키워드 입력
             "", # 분석 결과
-            None, # 다운로드 파일
-            gr.update(visible=False), # 키워드 분석 섹션
-            gr.update(visible=False), # 분석 결과 출력 섹션
-            "" # 키워드 상태
+            None # 다운로드 파일
         )
 
-# 래퍼 함수들
-def search_with_loading(keyword, korean_only, apply_main_keyword, exclude_zero_volume, session_id):
-    update_session_activity(session_id)
+# ========== UI 처리 래퍼 함수들 ==========
+
+def wrapper_search_with_loading(keyword, korean_only, apply_main_keyword, exclude_zero_volume):
+    """검색 로딩 UI 처리"""
+    result = search_with_loading(keyword, korean_only, apply_main_keyword, exclude_zero_volume)
     return (
-        gr.update(visible=True),
-        gr.update(visible=False)
+        gr.update(visible=True),  # progress_section
+        gr.update(visible=False)  # empty_table_html
     )
 
-def process_search_results(keyword, korean_only, apply_main_keyword, exclude_zero_volume, session_id):
-    update_session_activity(session_id)
-    
-    result = wrapper_modified(keyword, korean_only, apply_main_keyword, exclude_zero_volume, session_id)
+def wrapper_process_search_results(keyword, korean_only, apply_main_keyword, exclude_zero_volume):
+    """검색 결과 처리 UI"""
+    result = process_search_results(keyword, korean_only, apply_main_keyword, exclude_zero_volume)
     
-    table_html, cat_choices, vol_choices, df, selected_cat, excel, keyword_section_vis, cat_section_vis, new_keyword_state = result
+    # 안전하게 결과 처리
+    if len(result) >= 5:
+        table_html, cat_choices, vol_choices, selected_cat, download_file = result[:5]
+        progress_html = result[5] if len(result) > 5 else ""
+    else:
+        logger.error(f"검색 결과 처리 오류: 예상보다 적은 반환값 {len(result)}")
+        table_html = "<p>검색 결과 처리에 문제가 발생했습니다.</p>"
+        cat_choices = ["전체 보기"]
+        vol_choices = ["전체"]
+        selected_cat = "전체 보기"
+        download_file = None
+        progress_html = ""
     
-    if table_html and "검색 결과가 없습니다" not in table_html:
-        empty_placeholder_vis = False
+    # UI 표시 여부 결정
+    if table_html and "검색 결과가 없습니다" not in table_html and "문제가 발생했습니다" not in table_html:
         keyword_section_visibility = True
+        category_section_visibility = True
+        empty_placeholder_vis = False
         execution_section_visibility = True
     else:
-        empty_placeholder_vis = True
         keyword_section_visibility = False
+        category_section_visibility = False
+        empty_placeholder_vis = True
         execution_section_visibility = False
     
+    # 가상의 state_df (원본과 호환성을 위해)
+    state_df = pd.DataFrame()
+    
     return (
-        table_html, cat_choices, vol_choices, df, selected_cat, excel,
-        gr.update(visible=keyword_section_visibility),
-        gr.update(visible=cat_section_vis),
-        gr.update(visible=False),
-        gr.update(visible=empty_placeholder_vis),
-        gr.update(visible=execution_section_visibility),
-        new_keyword_state
+        table_html,  # table_output
+        cat_choices,  # category_filter choices
+        vol_choices,  # search_volume_filter choices
+        state_df,  # state_df
+        selected_cat,  # selected_category value
+        download_file,  # download_output
+        gr.update(visible=keyword_section_visibility),  # keyword_analysis_section
+        gr.update(visible=category_section_visibility),  # category_analysis_section
+        gr.update(visible=False),  # progress_section
+        gr.update(visible=empty_placeholder_vis),  # empty_table_html
+        gr.update(visible=execution_section_visibility),  # execution_section
+        keyword  # keyword_state
     )
 
-def analyze_with_loading(analysis_keywords, selected_category, state_df, session_id):
-    update_session_activity(session_id)
-    return gr.update(visible=True)
+def wrapper_analyze_with_loading(analysis_keywords, selected_category, state_df):
+    """분석 로딩 UI 처리"""
+    result = analyze_with_loading(analysis_keywords, selected_category)
+    return gr.update(visible=True)  # progress_section
 
-def process_analyze_results(analysis_keywords, selected_category, state_df, session_id):
-    update_session_activity(session_id)
-    results = analyze_with_auto_download(analysis_keywords, selected_category, state_df, session_id)
-    return results + (gr.update(visible=False),)
+def wrapper_process_analyze_results(analysis_keywords, selected_category, state_df):
+    """분석 결과 처리 UI"""
+    analysis_result, download_file = process_analyze_results(analysis_keywords, selected_category)
+    return (
+        analysis_result,  # analysis_result
+        download_file,  # download_output
+        gr.update(visible=True),  # analysis_output_section
+        gr.update(visible=False)  # progress_section
+    )
 
 # 세션 정리 스케줄러
 def start_session_cleanup_scheduler():
@@ -423,7 +539,6 @@ def start_session_cleanup_scheduler():
         while True:
             time.sleep(600)  # 10분마다 실행
             cleanup_old_sessions()
-            # 추가로 허깅페이스 임시 폴더도 주기적 정리
             cleanup_huggingface_temp_folders()
     
     threading.Thread(target=cleanup_scheduler, daemon=True).start()
@@ -487,9 +602,6 @@ def create_app():
     )) as demo:
         gr.HTML(fontawesome_html)
         
-        # 세션 ID 상태 (각 사용자별로 고유)
-        session_id = gr.State(get_session_id)
-        
         # 키워드 상태 관리
         keyword_state = gr.State("")
         
@@ -662,14 +774,14 @@ def create_app():
         # 상태 저장용 변수
         state_df = gr.State()
         
-        # 이벤트 연결 - 모든 함수에 session_id 추가
+        # 이벤트 연결
         search_btn.click(
-            fn=search_with_loading,
-            inputs=[keyword, korean_only, apply_main_keyword, exclude_zero_volume, session_id],
+            fn=wrapper_search_with_loading,
+            inputs=[keyword, korean_only, apply_main_keyword, exclude_zero_volume],
             outputs=[progress_section, empty_table_html]
         ).then(
-            fn=process_search_results,
-            inputs=[keyword, korean_only, apply_main_keyword, exclude_zero_volume, session_id],
+            fn=wrapper_process_search_results,
+            inputs=[keyword, korean_only, apply_main_keyword, exclude_zero_volume],
             outputs=[
                 table_output, category_filter, search_volume_filter, 
                 state_df, selected_category, download_output,
@@ -679,29 +791,29 @@ def create_app():
             ]
         )
         
-        # 필터 및 정렬 변경 이벤트 연결 - session_id 추가
+        # 필터 및 정렬 변경 이벤트 연결
         category_filter.change(
             fn=filter_and_sort_table, 
             inputs=[
-                state_df, category_filter, gr.Textbox(value="정렬 없음", visible=False), 
+                category_filter, gr.Textbox(value="정렬 없음", visible=False), 
                 total_volume_sort, usage_count_sort, 
-                search_volume_filter, exclude_zero_volume, session_id
+                search_volume_filter, exclude_zero_volume
             ], 
             outputs=[table_output]
         )
         
         category_filter.change(
             fn=update_category_selection,
-            inputs=[category_filter, session_id],
+            inputs=[category_filter],
             outputs=[selected_category]
         )
         
         total_volume_sort.change(
             fn=filter_and_sort_table, 
             inputs=[
-                state_df, category_filter, gr.Textbox(value="정렬 없음", visible=False), 
+                category_filter, gr.Textbox(value="정렬 없음", visible=False), 
                 total_volume_sort, usage_count_sort, 
-                search_volume_filter, exclude_zero_volume, session_id
+                search_volume_filter, exclude_zero_volume
             ], 
             outputs=[table_output]
         )
@@ -709,9 +821,9 @@ def create_app():
         usage_count_sort.change(
             fn=filter_and_sort_table, 
             inputs=[
-                state_df, category_filter, gr.Textbox(value="정렬 없음", visible=False), 
+                category_filter, gr.Textbox(value="정렬 없음", visible=False), 
                 total_volume_sort, usage_count_sort, 
-                search_volume_filter, exclude_zero_volume, session_id
+                search_volume_filter, exclude_zero_volume
             ], 
             outputs=[table_output]
         )
@@ -719,9 +831,9 @@ def create_app():
         search_volume_filter.change(
             fn=filter_and_sort_table, 
             inputs=[
-                state_df, category_filter, gr.Textbox(value="정렬 없음", visible=False), 
+                category_filter, gr.Textbox(value="정렬 없음", visible=False), 
                 total_volume_sort, usage_count_sort, 
-                search_volume_filter, exclude_zero_volume, session_id
+                search_volume_filter, exclude_zero_volume
             ], 
             outputs=[table_output]
         )
@@ -729,87 +841,36 @@ def create_app():
         exclude_zero_volume.change(
             fn=filter_and_sort_table, 
             inputs=[
-                state_df, category_filter, gr.Textbox(value="정렬 없음", visible=False), 
+                category_filter, gr.Textbox(value="정렬 없음", visible=False), 
                 total_volume_sort, usage_count_sort, 
-                search_volume_filter, exclude_zero_volume, session_id
+                search_volume_filter, exclude_zero_volume
             ], 
             outputs=[table_output]
         )
         
-        # 카테고리 분석 버튼 이벤트 - session_id 추가
+        # 카테고리 분석 버튼 이벤트
         analyze_btn.click(
-            fn=analyze_with_loading,
-            inputs=[analysis_keywords, selected_category, state_df, session_id],
+            fn=wrapper_analyze_with_loading,
+            inputs=[analysis_keywords, selected_category, state_df],
             outputs=[progress_section]
         ).then(
-            fn=process_analyze_results,
-            inputs=[analysis_keywords, selected_category, state_df, session_id],
+            fn=wrapper_process_analyze_results,
+            inputs=[analysis_keywords, selected_category, state_df],
             outputs=[analysis_result, download_output, analysis_output_section, progress_section]
         )
         
-        # 리셋 버튼 이벤트 연결 - session_id 추가
+        # 리셋 버튼 이벤트 연결
         reset_btn.click(
             fn=reset_interface,
-            inputs=[session_id],
+            inputs=[],
             outputs=[
                 keyword, korean_only, exclude_zero_volume, apply_main_keyword, 
                 table_output, category_filter, category_filter, 
                 search_volume_filter, search_volume_filter, 
                 total_volume_sort, usage_count_sort, 
                 selected_category, selected_category, 
-                analysis_keywords, analysis_result, download_output,
-                keyword_analysis_section, analysis_output_section,
-                keyword_state
+                analysis_keywords, analysis_result, download_output
             ]
         )
                 
-    return demo
-
-if __name__ == "__main__":
-    # ========== 시작 시 전체 초기화 ==========
-    logger.info("🚀 컨트롤 타워 애플리케이션 시작...")
-    
-    # 1. 첫 번째: 허깅페이스 임시 폴더 정리 및 환경 설정
-    app_temp_dir = cleanup_on_startup()
-    
-    # 2. 세션 정리 스케줄러 시작
-    start_session_cleanup_scheduler()
-    
-    # 3. API 연결 테스트
-    try:
-        test_client = get_api_client()
-        logger.info("✅ API 연결 테스트 성공")
-    except Exception as e:
-        logger.error(f"❌ API 연결 실패: {e}")
-        raise
-    
-    logger.info("===== 컨트롤 타워 애플리케이션 시작 완료 at %s =====", time.strftime("%Y-%m-%d %H:%M:%S"))
-    logger.info(f"📁 임시 파일 저장 위치: {app_temp_dir}")
-    
-    # ========== 앱 실행 ==========
-    try:
-        app = create_app()
-        app.launch(
-            share=False,  # 보안을 위해 share 비활성화
-            server_name="0.0.0.0",  # 모든 IP에서 접근 허용
-            server_port=7860,  # 포트 지정
-            max_threads=40,  # 멀티유저를 위한 스레드 수 증가
-            auth=None,  # 필요시 인증 추가 가능
-            show_error=True,  # 에러 표시
-            quiet=False,  # 로그 표시
-            favicon_path=None,  # 파비콘 설정
-            ssl_verify=False  # SSL 검증 비활성화 (개발용)
-        )
-    except Exception as e:
-        logger.error(f"애플리케이션 실행 실패: {e}")
-        raise
-    finally:
-        # 애플리케이션 종료 시 정리
-        logger.info("🧹 컨트롤 타워 애플리케이션 종료 - 최종 정리 작업...")
-        try:
-            cleanup_huggingface_temp_folders()
-            if os.path.exists(app_temp_dir):
-                shutil.rmtree(app_temp_dir, ignore_errors=True)
-            logger.info("✅ 최종 정리 완료")
-        except Exception as e:
-            logger.error(f"최종 정리 중 오류: {e}")
+    return demo