Update app.py

app.py

@@ -75,6 +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)
@@ -86,7 +93,7 @@ def get_blog_count(keyword):
         if rescode == 200:
             response_body = response.read()
             data = json.loads(response_body.decode('utf-8'))
-            return data['total']
+            return data.get('total', 0)
         else:
             return 0
     except Exception as e:
@@ -100,54 +107,127 @@ def get_keywords_data_chunk(chunk):
 def get_blog_count_parallel(keyword):
     return (keyword, get_blog_count(keyword))
 
-def get_monthly_search_volumes(keywords):
-    all_data = []
-    chunk_size = 10  # 키워드를 10개씩 나누어 요청
+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:
+            # keywordList에서 입력한 키워드와 일치하는 항목을 찾습니다.
+            for item in data['keywordList']:
+                if item['relKeyword'].strip().lower() == keyword.strip().lower():
+                    monthly_pc = item.get('monthlyPcQcCnt', 0)
+                    monthly_mobile = item.get('monthlyMobileQcCnt', 0)
 
-    # 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 응답이 유효하지 않습니다.", "", "", "")]  # 블로그 문서 수 칼럼 추가
+                    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
+                    return (keyword, monthly_pc, monthly_mobile, total_searches)
+            # 입력한 키워드와 일치하는 항목이 없을 경우
+            return (keyword, 0, 0, 0)
+        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 = []
-    unique_keywords = set()
-    for item in all_data:
-        keyword = item['relKeyword']
-        if keyword not in unique_keywords:
-            unique_keywords.add(keyword)
-            monthly_pc = item['monthlyPcQcCnt']
-            monthly_mobile = item['monthlyMobileQcCnt']
-            
-            if isinstance(monthly_pc, str):
-                monthly_pc = int(monthly_pc.replace(',', '').replace('< 10', '0'))
-            if isinstance(monthly_mobile, str):
-                monthly_mobile = int(monthly_mobile.replace(',', '').replace('< 10', '0'))
-            
-            total_searches = monthly_pc + monthly_mobile
-            results.append((keyword, monthly_pc, monthly_mobile, total_searches))
+
+    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 응답이 유효하지 않습니다.", "", "", "")]
         
-        if len(results) >= 100:
-            break
+        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:
-        blog_futures = [executor.submit(get_blog_count_parallel, result[0]) for result in results]
-        for i, future in enumerate(blog_futures):
-            try:
-                keyword, blog_count = future.result()
-                results[i] = (results[i][0], results[i][1], results[i][2], results[i][3], blog_count)
-            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")
+        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:
+                    keyword, blog_count = future.result()
+                    results[i] = (results[i][0], results[i][1], results[i][2], results[i][3], blog_count)
+                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:
+            blog_futures = [executor.submit(get_blog_count_parallel, result[0]) for result in results]
+            temp_results = []
+            for future in blog_futures:
+                try:
+                    keyword, blog_count = future.result()
+                    temp_results.append((keyword, results[0][1], results[0][2], results[0][3], blog_count))
+                except Exception as e:
+                    print(f"Error fetching blog count for keyword '{keyword}': {e}")
+                    temp_results.append((keyword, results[0][1], results[0][2], results[0][3], "Error"))
+            results = temp_results
 
     return results
 
@@ -160,20 +240,26 @@ def save_to_excel(results, keyword):
     df.to_excel(file_path, index=False)
     return file_path
 
-def display_search_volumes(keywords):
-    keyword_list = [keyword.strip() for keyword in keywords.split(',')]
-    results = get_monthly_search_volumes(keyword_list)
+def display_search_volumes(keywords, include_related):
+    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
 
 iface = gr.Interface(
     fn=display_search_volumes, 
-    inputs=gr.Textbox(placeholder="키워드를 입력하세요"), 
+    inputs=[
+        gr.Textbox(placeholder="키워드를 입력하세요 (쉼표로 구분)", lines=2),
+        gr.Checkbox(label="연관검색어 포함", value=True)  # 연관검색어 토글 추가
+    ], 
     outputs=[
         gr.Dataframe(headers=["키워드", "PC월검색량", "모바일월검색량", "토탈월검색량", "블로그문서수"]),
         gr.File(label="다운로드 엑셀 파일")
     ],
     title="네이버 월검색량 검색기",
+    description="키워드의 월 검색량과 블로그 문서 수를 확인할 수 있습니다. 연관검색어를 포함할지 선택하세요.",
 )
 
-iface.launch(share=True)  # share=True를 추가하여 공개 링크 생성
+iface.launch(share=True)  # share=True를 추가하여 공개 링크 생성