Update app.py

app.py

@@ -75,13 +75,13 @@ def get_blog_count(keyword):
     # 클라이언트 ID와 시크릿을 환경 변수에서 불러옵니다.
     client_id = CLIENT_ID
     client_secret = CLIENT_SECRET
-    
+
     # keyword가 바이트 타입일 경우 디코딩
     if isinstance(keyword, bytes):
         keyword = keyword.decode('utf-8')
     elif not isinstance(keyword, str):
         keyword = str(keyword)
-    
+
     encText = urllib.parse.quote(keyword)
     url = "https://openapi.naver.com/v1/search/blog?query=" + encText
     request = urllib.request.Request(url)
@@ -107,40 +107,15 @@ def get_keywords_data_chunk(chunk):
 def get_blog_count_parallel(keyword):
     return (keyword, get_blog_count(keyword))
 
-def get_monthly_search_volumes(keywords, include_related_keywords=True):
-    all_data = []
-    chunk_size = 10  # 키워드를 10개씩 나누어 요청
-
-    if include_related_keywords:
-        # API 병렬 요청
-        with ThreadPoolExecutor(max_workers=5) as executor:
-            futures = [executor.submit(get_keywords_data_chunk, keywords[i:i+chunk_size]) for i in range(0, len(keywords), chunk_size)]
-            for future in futures:
-                try:
-                    data = future.result()
-                    if 'keywordList' in data:
-                        all_data.extend(data['keywordList'])
-                except Exception as e:
-                    print(f"Error fetching keywords data chunk: {e}")
-    else:
-        # 연관검색어를 포함하지 않으므로 입력 키워드만 처리
-        for keyword in keywords:
-            # 가상의 데이터 구조을 맞추기 위해
-            all_data.append({
-                'relKeyword': keyword,
-                'monthlyPcQcCnt': '0',  # 실제 API에서 값을 가져오려면 별도 요청 필요
-                'monthlyMobileQcCnt': '0'
-            })
-
-    if not all_data:
-        return [("Error", "데이터가 반환되지 않았거나 API 응답이 유효하지 않습니다.", "", "", "")]  # 블로그 문서 수 칼럼 추가
-
-    results = []
-    unique_keywords = set()
-    for item in all_data:
-        keyword = item['relKeyword']
-        if keyword not in unique_keywords:
-            unique_keywords.add(keyword)
+def get_search_volumes(keyword):
+    """
+    단일 키워드의 월 검색량을 가져오는 함수.
+    """
+    api = NaverAPI(BASE_URL, API_KEY, SECRET_KEY, CUSTOMER_ID)
+    try:
+        data = api.get_keywords_data([keyword])
+        if 'keywordList' in data and len(data['keywordList']) > 0:
+            item = data['keywordList'][0]
             monthly_pc = item.get('monthlyPcQcCnt', 0)
             monthly_mobile = item.get('monthlyMobileQcCnt', 0)
             
@@ -158,14 +133,78 @@ def get_monthly_search_volumes(keywords, include_related_keywords=True):
                     monthly_mobile = 0
             
             total_searches = monthly_pc + monthly_mobile
-            results.append((keyword, monthly_pc, monthly_mobile, total_searches))
-        
-        if len(results) >= 100 and include_related_keywords:
-            break
+            return (keyword, monthly_pc, monthly_mobile, total_searches)
+        else:
+            return (keyword, 0, 0, 0)
+    except Exception as e:
+        print(f"Error fetching search volumes for keyword '{keyword}': {e}")
+        return (keyword, 0, 0, 0)
+
+def get_monthly_search_volumes(keywords, include_related_keywords=True):
+    all_data = []
+    results = []
 
     if include_related_keywords:
-        # 블로그 문서 수 병렬 요청
+        chunk_size = 10  # 키워드를 10개씩 나누어 요청
+        # API 병렬 요청
+        with ThreadPoolExecutor(max_workers=5) as executor:
+            futures = [executor.submit(get_keywords_data_chunk, keywords[i:i+chunk_size]) for i in range(0, len(keywords), chunk_size)]
+            for future in futures:
+                try:
+                    data = future.result()
+                    if 'keywordList' in data:
+                        all_data.extend(data['keywordList'])
+                except Exception as e:
+                    print(f"Error fetching keywords data chunk: {e}")
+
+        if not all_data:
+            return [("Error", "데이터가 반환되지 않았거나 API 응답이 유효하지 않습니다.", "", "", "")]
+        
+        unique_keywords = set()
+        for item in all_data:
+            keyword = item['relKeyword']
+            if keyword not in unique_keywords:
+                unique_keywords.add(keyword)
+                monthly_pc = item.get('monthlyPcQcCnt', 0)
+                monthly_mobile = item.get('monthlyMobileQcCnt', 0)
+                
+                if isinstance(monthly_pc, str):
+                    monthly_pc = monthly_pc.replace(',', '').replace('< 10', '0')
+                    try:
+                        monthly_pc = int(monthly_pc)
+                    except ValueError:
+                        monthly_pc = 0
+                if isinstance(monthly_mobile, str):
+                    monthly_mobile = monthly_mobile.replace(',', '').replace('< 10', '0')
+                    try:
+                        monthly_mobile = int(monthly_mobile)
+                    except ValueError:
+                        monthly_mobile = 0
+                
+                total_searches = monthly_pc + monthly_mobile
+                results.append((keyword, monthly_pc, monthly_mobile, total_searches))
+            
+            if len(results) >= 100:
+                break
+
+    else:
+        # 연관검색어를 포함하지 않으므로 입력 키워드만 처리
         with ThreadPoolExecutor(max_workers=5) as executor:
+            futures = [executor.submit(get_search_volumes, keyword) for keyword in keywords]
+            for future in futures:
+                try:
+                    result = future.result()
+                    results.append(result)
+                except Exception as e:
+                    print(f"Error fetching search volumes for keyword '{keyword}': {e}")
+                    results.append((keyword, 0, 0, 0))
+    
+    if not results:
+        return [("Error", "데이터가 반환되지 않았거나 API 응답이 유효하지 않습니다.", "", "", "")]
+    
+    # 블로그 문서 수 병렬 요청
+    with ThreadPoolExecutor(max_workers=5) as executor:
+        if include_related_keywords:
             blog_futures = [executor.submit(get_blog_count_parallel, result[0]) for result in results]
             for i, future in enumerate(blog_futures):
                 try:
@@ -174,10 +213,8 @@ def get_monthly_search_volumes(keywords, include_related_keywords=True):
                 except Exception as e:
                     print(f"Error fetching blog count for keyword '{results[i][0]}': {e}")
                     results[i] = (results[i][0], results[i][1], results[i][2], results[i][3], "Error")
-    else:
-        # 블로그 문서 수 병렬 요청 (연관검색어가 아닐 경우 각 키워드에 대해)
-        with ThreadPoolExecutor(max_workers=5) as executor:
-            blog_futures = [executor.submit(get_blog_count_parallel, keyword) for keyword in results]
+        else:
+            blog_futures = [executor.submit(get_blog_count_parallel, result[0]) for result in results]
             temp_results = []
             for future in blog_futures:
                 try:
@@ -200,7 +237,9 @@ def save_to_excel(results, keyword):
     return file_path
 
 def display_search_volumes(keywords, include_related):
-    keyword_list = [keyword.strip() for keyword in keywords.split(',')]
+    keyword_list = [keyword.strip() for keyword in keywords.split(',') if keyword.strip()]
+    if not keyword_list:
+        return [("Error", "입력된 키워드가 없습니다.", "", "", "")], None
     results = get_monthly_search_volumes(keyword_list, include_related_keywords=include_related)
     file_path = save_to_excel(results, keywords)
     return results, file_path
@@ -208,7 +247,7 @@ def display_search_volumes(keywords, include_related):
 iface = gr.Interface(
     fn=display_search_volumes, 
     inputs=[
-        gr.Textbox(placeholder="키워드를 입력하세요"),
+        gr.Textbox(placeholder="키워드를 입력하세요 (쉼표로 구분)", lines=2),
         gr.Checkbox(label="연관검색어 포함", value=True)  # 연관검색어 토글 추가
     ], 
     outputs=[