Update app.py

app.py

@@ -12,13 +12,6 @@ import glob
 from datetime import datetime
 from gradio_client import Client
 
-# 환경변수 로드를 위한 dotenv 사용
-try:
-    from dotenv import load_dotenv
-    load_dotenv()
-except ImportError:
-    pass
-    
 # 로깅 설정
 logging.basicConfig(
     level=logging.INFO,
@@ -31,29 +24,9 @@ logging.basicConfig(
 
 logger = logging.getLogger(__name__)
 
-# API 클라이언트 설정
-def get_api_client():
-    """환경변수에서 API 엔드포인트를 가져와 클라이언트 생성"""
-    endpoint = os.getenv('API_ENDPOINT', '').strip()
-    
-    # 디버깅을 위한 환경변수 체크 (실제 값은 로그에 남기지 않음)
-    if not endpoint:
-        logger.error("API_ENDPOINT 환경변수가 설정되지 않았습니다.")
-        logger.info("사용 가능한 환경변수들:")
-        for key in os.environ.keys():
-            if 'API' in key.upper():
-                logger.info(f"  {key}: {'설정됨' if os.environ[key] else '비어있음'}")
-        raise ValueError("API_ENDPOINT 환경변수가 설정되지 않았습니다.")
-    
-    if endpoint.startswith('#') or len(endpoint) == 0:
-        logger.error("API_ENDPOINT 환경변수가 올바르지 않은 형식입니다.")
-        raise ValueError("API_ENDPOINT 환경변수가 올바르게 설정되지 않았습니다.")
-    
-    # 엔드포인트 형식 검증
-    if '/' not in endpoint:
-        logger.error("API_ENDPOINT는 'username/repo-name' 형식이어야 합니다.")
-        raise ValueError("API_ENDPOINT 형식이 올바르지 않습니다.")
-    
+# API 클라이언트 설정 (환경변수로 숨김)
+def get_client():
+    endpoint = os.getenv('API_ENDPOINT')
     return Client(endpoint)
 
 # 세션별 임시 파일 관리를 위한 딕셔너리
@@ -63,7 +36,6 @@ session_data = {}
 def cleanup_huggingface_temp_folders():
     """허깅페이스 임시 폴더 초기 정리"""
     try:
-        # 일반적인 임시 디렉토리들
         temp_dirs = [
             tempfile.gettempdir(),
             "/tmp",
@@ -79,7 +51,6 @@ def cleanup_huggingface_temp_folders():
         for temp_dir in temp_dirs:
             if os.path.exists(temp_dir):
                 try:
-                    # 기존 세션 파일들 정리
                     session_files = glob.glob(os.path.join(temp_dir, "session_*.xlsx"))
                     session_files.extend(glob.glob(os.path.join(temp_dir, "session_*.csv")))
                     session_files.extend(glob.glob(os.path.join(temp_dir, "*keyword*.xlsx")))
@@ -89,7 +60,6 @@ def cleanup_huggingface_temp_folders():
                     
                     for file_path in session_files:
                         try:
-                            # 파일이 1시간 이상 오래된 경우만 삭제
                             if os.path.getmtime(file_path) < time.time() - 3600:
                                 os.remove(file_path)
                                 cleanup_count += 1
@@ -102,7 +72,6 @@ def cleanup_huggingface_temp_folders():
         
         logger.info(f"✅ 허깅페이스 임시 폴더 초기 정리 완료 - {cleanup_count}개 파일 삭제")
         
-        # Gradio 캐시 폴더도 정리
         try:
             gradio_temp_dir = os.path.join(os.getcwd(), "gradio_cached_examples")
             if os.path.exists(gradio_temp_dir):
@@ -117,17 +86,14 @@ def cleanup_huggingface_temp_folders():
 def setup_clean_temp_environment():
     """깨끗한 임시 환경 설정"""
     try:
-        # 1. 기존 임시 파일들 정리
         cleanup_huggingface_temp_folders()
         
-        # 2. 애플리케이션 전용 임시 디렉토리 생성
-        app_temp_dir = os.path.join(tempfile.gettempdir(), "control_tower_app")
+        app_temp_dir = os.path.join(tempfile.gettempdir(), "keyword_app")
         if os.path.exists(app_temp_dir):
             shutil.rmtree(app_temp_dir, ignore_errors=True)
         os.makedirs(app_temp_dir, exist_ok=True)
         
-        # 3. 환경 변수 설정 (임시 디렉토리 지정)
-        os.environ['CONTROL_TOWER_TEMP'] = app_temp_dir
+        os.environ['KEYWORD_APP_TEMP'] = app_temp_dir
         
         logger.info(f"✅ 애플리케이션 전용 임시 디렉토리 설정: {app_temp_dir}")
         
@@ -139,7 +105,7 @@ def setup_clean_temp_environment():
 
 def get_app_temp_dir():
     """애플리케이션 전용 임시 디렉토리 반환"""
-    return os.environ.get('CONTROL_TOWER_TEMP', tempfile.gettempdir())
+    return os.environ.get('KEYWORD_APP_TEMP', tempfile.gettempdir())
 
 def get_session_id():
     """세션 ID 생성"""
@@ -175,11 +141,10 @@ def cleanup_old_sessions():
     sessions_to_remove = []
     
     for session_id, data in session_data.items():
-        if current_time - data.get('last_activity', 0) > 3600:  # 1시간 초과
+        if current_time - data.get('last_activity', 0) > 3600:
             sessions_to_remove.append(session_id)
     
     for session_id in sessions_to_remove:
-        # 파일 정리
         if session_id in session_temp_files:
             for file_path in session_temp_files[session_id]:
                 try:
@@ -190,7 +155,6 @@ def cleanup_old_sessions():
                     logger.error(f"오래된 세션 파일 삭제 오류: {e}")
             del session_temp_files[session_id]
         
-        # 세션 데이터 정리
         if session_id in session_data:
             del session_data[session_id]
             logger.info(f"오래된 세션 데이터 삭제: {session_id[:8]}...")
@@ -206,12 +170,10 @@ def create_session_temp_file(session_id, suffix='.xlsx'):
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     random_suffix = str(random.randint(1000, 9999))
     
-    # 애플리케이션 전용 임시 디렉토리 사용
     temp_dir = get_app_temp_dir()
     filename = f"session_{session_id[:8]}_{timestamp}_{random_suffix}{suffix}"
     temp_file_path = os.path.join(temp_dir, filename)
     
-    # 빈 파일 생성
     with open(temp_file_path, 'w') as f:
         pass
     
@@ -219,11 +181,11 @@ def create_session_temp_file(session_id, suffix='.xlsx'):
     return temp_file_path
 
 def wrapper_modified(keyword, korean_only, apply_main_keyword_option, exclude_zero_volume, session_id):
-    """키워드 검색 및 처리 래퍼 함수 (API 클라이언트 사용)"""
+    """키워드 검색 및 처리 래퍼 함수 (API 사용)"""
     update_session_activity(session_id)
     
     try:
-        client = get_api_client()
+        client = get_client()
         result = client.predict(
             keyword=keyword,
             korean_only=korean_only,
@@ -232,10 +194,8 @@ def wrapper_modified(keyword, korean_only, apply_main_keyword_option, exclude_ze
             api_name="/process_search_results"
         )
         
-        # API 결과 처리
         table_html, cat_choices, vol_choices, selected_cat, download_file = result
         
-        # 로컬 파일로 다운로드 파일 복사
         local_file = None
         if download_file:
             local_file = create_session_temp_file(session_id, '.xlsx')
@@ -245,7 +205,7 @@ def wrapper_modified(keyword, korean_only, apply_main_keyword_option, exclude_ze
             gr.update(value=table_html), 
             gr.update(choices=cat_choices), 
             gr.update(choices=vol_choices),
-            None,  # DataFrame은 클라이언트에서 관리하지 않음
+            None,
             gr.update(choices=cat_choices, value=selected_cat), 
             local_file,
             gr.update(visible=True), 
@@ -256,7 +216,7 @@ def wrapper_modified(keyword, korean_only, apply_main_keyword_option, exclude_ze
     except Exception as e:
         logger.error(f"API 호출 오류: {e}")
         return (
-            gr.update(value="<p>검색 중 오류가 발생했습니다. 다시 시도해주세요.</p>"), 
+            gr.update(value="<p>검색 결과가 없습니다. 다른 키워드로 시도해보세요.</p>"), 
             gr.update(choices=["전체 보기"]), 
             gr.update(choices=["전체"]), 
             None, 
@@ -268,11 +228,11 @@ def wrapper_modified(keyword, korean_only, apply_main_keyword_option, exclude_ze
         )
 
 def analyze_with_auto_download(analysis_keywords, selected_category, state_df, session_id):
-    """카테고리 일치 분석 실행 및 자동 다운로드 (API 클라이언트 사용)"""
+    """카테고리 일치 분석 실행 및 자동 다운로드 (API 사용)"""
     update_session_activity(session_id)
     
     try:
-        client = get_api_client()
+        client = get_client()
         result = client.predict(
             analysis_keywords=analysis_keywords,
             selected_category=selected_category,
@@ -281,7 +241,6 @@ def analyze_with_auto_download(analysis_keywords, selected_category, state_df, s
         
         analysis_result, download_file = result
         
-        # 로컬 파일로 다운로드 파일 복사
         local_file = None
         if download_file:
             local_file = create_session_temp_file(session_id, '.xlsx')
@@ -294,11 +253,11 @@ def analyze_with_auto_download(analysis_keywords, selected_category, state_df, s
         return "분석 중 오류가 발생했습니다. 다시 시도해주세요.", None, gr.update(visible=False)
 
 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 클라이언트 사용)"""
+    """테이블 필터링 및 정렬 함수 (API 사용)"""
     update_session_activity(session_id)
     
     try:
-        client = get_api_client()
+        client = get_client()
         result = client.predict(
             selected_cat=selected_cat,
             keyword_sort=keyword_sort,
@@ -320,7 +279,7 @@ def update_category_selection(selected_cat, session_id):
     update_session_activity(session_id)
     
     try:
-        client = get_api_client()
+        client = get_client()
         result = client.predict(
             selected_cat=selected_cat,
             api_name="/update_category_selection"
@@ -336,7 +295,6 @@ 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]:
             try:
@@ -348,60 +306,22 @@ def reset_interface(session_id):
         session_temp_files[session_id] = []
     
     try:
-        client = get_api_client()
-        result = client.predict(
-            api_name="/reset_interface"
-        )
-        
-        # API 결과를 그대로 반환
+        client = get_client()
+        result = client.predict(api_name="/reset_interface")
         return result
         
     except Exception as e:
         logger.error(f"리셋 API 호출 오류: {e}")
         return (
-            "", # 검색 키워드
-            True, # 한글만 추출
-            False, # 검색량 0 키워드 제외
-            "메인키워드 적용", # 조합 방식
-            "", # HTML 테이블
-            ["전체 보기"], # 카테고리 필터
-            "전체 보기", # 카테고리 필터 선택
-            ["전체"], # 검색량 구간 필터
-            "전체", # 검색량 구간 선택
-            "정렬 없음", # 총검색량 정렬
-            "정렬 없음", # 키워드 사용횟수 정렬
-            ["전체 보기"], # 분석할 카테고리
-            "전체 보기", # 분석할 카테고리 선택
-            "", # 키워드 입력
-            "", # 분석 결과
-            None # 다운로드 파일
+            "", True, False, "메인키워드 적용", "", ["전체 보기"], "전체 보기", 
+            ["전체"], "전체", "정렬 없음", "정렬 없음", None, ["전체 보기"], 
+            "전체 보기", "", "", None, gr.update(visible=False), 
+            gr.update(visible=False), ""
         )
 
-# 래퍼 함수들
 def search_with_loading(keyword, korean_only, apply_main_keyword, exclude_zero_volume, session_id):
     update_session_activity(session_id)
-    
-    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 (
-            gr.update(visible=True),
-            gr.update(visible=False)
-        )
-        
-    except Exception as e:
-        logger.error(f"검색 로딩 API 호출 오류: {e}")
-        return (
-            gr.update(visible=False),
-            gr.update(visible=True)
-        )
+    return (gr.update(visible=True), gr.update(visible=False))
 
 def process_search_results(keyword, korean_only, apply_main_keyword, exclude_zero_volume, session_id):
     update_session_activity(session_id)
@@ -431,49 +351,30 @@ def process_search_results(keyword, korean_only, apply_main_keyword, exclude_zer
 
 def analyze_with_loading(analysis_keywords, selected_category, state_df, session_id):
     update_session_activity(session_id)
-    
-    try:
-        client = get_api_client()
-        result = client.predict(
-            analysis_keywords=analysis_keywords,
-            selected_category=selected_category,
-            api_name="/analyze_with_loading"
-        )
-        
-        return gr.update(visible=True)
-        
-    except Exception as e:
-        logger.error(f"분석 로딩 API 호출 오류: {e}")
-        return gr.update(visible=False)
+    return gr.update(visible=True)
 
 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 start_session_cleanup_scheduler():
     """세션 정리 스케줄러 시작"""
     def cleanup_scheduler():
         while True:
-            time.sleep(600)  # 10분마다 실행
+            time.sleep(600)
             cleanup_old_sessions()
-            # 추가로 허깅페이스 임시 폴더도 주기적 정리
             cleanup_huggingface_temp_folders()
     
     threading.Thread(target=cleanup_scheduler, daemon=True).start()
 
 def cleanup_on_startup():
     """애플리케이션 시작 시 전체 정리"""
-    logger.info("🧹 컨트롤 타워 애플리케이션 시작 - 초기 정리 작업 시작...")
+    logger.info("🧹 애플리케이션 시작 - 초기 정리 작업 시작...")
     
-    # 1. 허깅페이스 임시 폴더 정리
     cleanup_huggingface_temp_folders()
-    
-    # 2. 깨끗한 임시 환경 설정
     app_temp_dir = setup_clean_temp_environment()
     
-    # 3. 전역 변수 초기화
     global session_temp_files, session_data
     session_temp_files.clear()
     session_data.clear()
@@ -482,7 +383,6 @@ def cleanup_on_startup():
     
     return app_temp_dir
 
-# Gradio 인터페이스 생성
 def create_app():
     fontawesome_html = """
     <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css">
@@ -490,7 +390,6 @@ def create_app():
     <link rel="stylesheet" href="https://fonts.googleapis.com/css2?family=Noto+Sans+KR:wght@300;400;500;700&display=swap">
     """
     
-    # CSS 파일 로드
     try:
         with open('style.css', 'r', encoding='utf-8') as f:
             custom_css = f.read()
@@ -504,40 +403,24 @@ def create_app():
     )) as demo:
         gr.HTML(fontawesome_html)
         
-        # 세션 ID 상태 (각 사용자별로 고유)
         session_id = gr.State(get_session_id)
-        
-        # 키워드 상태 관리
         keyword_state = gr.State("")
         
-        # 입력 섹션
         with gr.Column(elem_classes="custom-frame fade-in"):
             gr.HTML('<div class="section-title"><i class="fas fa-search"></i> 검색 입력</div>')
             
             with gr.Row():
                 with gr.Column(scale=1):
-                    keyword = gr.Textbox(
-                        label="메인 키워드", 
-                        placeholder="예: 오징어"
-                    )
+                    keyword = gr.Textbox(label="메인 키워드", placeholder="예: 오징어")
                 with gr.Column(scale=1):
-                    search_btn = gr.Button(
-                        "메인키워드 분석", 
-                        elem_classes="custom-button"
-                    )
+                    search_btn = gr.Button("메인키워드 분석", elem_classes="custom-button")
             
             with gr.Accordion("옵션 설정", open=False):
                 with gr.Row():
                     with gr.Column(scale=1):
-                        korean_only = gr.Checkbox(
-                            label="한글만 추출", 
-                            value=True
-                        )
+                        korean_only = gr.Checkbox(label="한글만 추출", value=True)
                     with gr.Column(scale=1):
-                        exclude_zero_volume = gr.Checkbox(
-                            label="검색량 0 키워드 제외", 
-                            value=False
-                        )
+                        exclude_zero_volume = gr.Checkbox(label="검색량 0 키워드 제외", value=False)
                 
                 with gr.Row():
                     with gr.Column(scale=1):
@@ -549,7 +432,6 @@ def create_app():
                     with gr.Column(scale=1):
                         gr.HTML("")
         
-        # 진행 상태 표시 섹션
         with gr.Column(elem_classes="custom-frame fade-in", visible=False) as progress_section:
             gr.HTML('<div class="section-title"><i class="fas fa-spinner"></i> 분석 진행 상태</div>')
             progress_html = gr.HTML("""
@@ -564,7 +446,6 @@ def create_app():
             </div>
             """)
         
-        # 메인키워드 분석 결과 섹션
         with gr.Column(elem_classes="custom-frame fade-in") as main_keyword_section:
             gr.HTML('<div class="section-title"><i class="fas fa-table"></i> 메인키워드 분석 결과</div>')
             
@@ -629,7 +510,6 @@ def create_app():
                 gr.HTML("<div class='data-container' id='table_container'></div>")
                 table_output = gr.HTML(elem_classes="fade-in")
         
-        # 카테고리 분석 섹션
         with gr.Column(elem_classes="custom-frame fade-in", visible=False) as category_analysis_section:
             gr.HTML('<div class="section-title"><i class="fas fa-chart-bar"></i> 키워드 분석</div>')
             
@@ -649,7 +529,6 @@ def create_app():
                         interactive=True
                     )
         
-        # 실행 섹션
         with gr.Column(elem_classes="execution-section", visible=False) as execution_section:
             gr.HTML('<div class="section-title"><i class="fas fa-play-circle"></i> 실행</div>')
             with gr.Row():
@@ -664,22 +543,16 @@ def create_app():
                         elem_classes=["execution-button", "secondary-button"]
                     )
         
-        # 분석 결과 출력 섹션
         with gr.Column(elem_classes="custom-frame fade-in", visible=False) as analysis_output_section:
             gr.HTML('<div class="section-title"><i class="fas fa-list-ul"></i> 분석 결과 요약</div>')
             
             analysis_result = gr.HTML(elem_classes="fade-in")
             
             with gr.Row():
-                download_output = gr.File(
-                    label="키워드 목록 다운로드", 
-                    visible=True
-                )
+                download_output = gr.File(label="키워드 목록 다운로드", visible=True)
         
-        # 상태 저장용 변수
         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],
@@ -696,7 +569,6 @@ def create_app():
             ]
         )
         
-        # 필터 및 정렬 변경 이벤트 연결 - session_id 추가
         category_filter.change(
             fn=filter_and_sort_table, 
             inputs=[
@@ -753,7 +625,6 @@ def create_app():
             outputs=[table_output]
         )
         
-        # 카테고리 분석 버튼 이벤트 - session_id 추가
         analyze_btn.click(
             fn=analyze_with_loading,
             inputs=[analysis_keywords, selected_category, state_df, session_id],
@@ -764,7 +635,6 @@ def create_app():
             outputs=[analysis_result, download_output, analysis_output_section, progress_section]
         )
         
-        # 리셋 버튼 이벤트 연결 - session_id 추가
         reset_btn.click(
             fn=reset_interface,
             inputs=[session_id],
@@ -773,54 +643,42 @@ def create_app():
                 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
+                state_df, selected_category, selected_category, 
+                analysis_keywords, analysis_result, download_output,
+                keyword_analysis_section, analysis_output_section,
+                keyword_state
             ]
         )
                 
     return demo
 
 if __name__ == "__main__":
-    # ========== 시작 시 전체 초기화 ==========
-    logger.info("🚀 컨트롤 타워 애플리케이션 시작...")
+    logger.info("🚀 메인키워드 분석 애플리케이션 시작...")
     
-    # 1. 첫 번째: 허깅페이스 임시 폴더 정리 및 환경 설정
     app_temp_dir = cleanup_on_startup()
-    
-    # 2. 세션 정리 스케줄러 시작
     start_session_cleanup_scheduler()
     
-    # 3. API 엔드포인트 확인
-    try:
-        client = get_api_client()
-        logger.info("✅ API 클라이언트 연결 확인 완료")
-    except Exception as e:
-        logger.error(f"❌ API 클라이언트 연결 실패: {e}")
-        raise
-    
-    logger.info("===== 컨트롤 타워 Application Startup at %s =====", time.strftime("%Y-%m-%d %H:%M:%S"))
+    logger.info("===== 멀티유저 메인키워드 분석 Application Startup 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 검증 비활성화 (개발용)
+            share=False,
+            server_name="0.0.0.0",
+            server_port=7860,
+            max_threads=40,
+            auth=None,
+            show_error=True,
+            quiet=False,
+            favicon_path=None,
+            ssl_verify=False
         )
     except Exception as e:
         logger.error(f"애플리케이션 실행 실패: {e}")
         raise
     finally:
-        # 애플리케이션 종료 시 정리
-        logger.info("🧹 컨트롤 타워 애플리케이션 종료 - 최종 정리 작업...")
+        logger.info("🧹 애플리케이션 종료 - 최종 정리 작업...")
         try:
             cleanup_huggingface_temp_folders()
             if os.path.exists(app_temp_dir):