Update app.py

app.py

@@ -100,12 +100,12 @@ 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):
+def get_monthly_search_volumes(keywords, include_related_keywords=True):
     all_data = []
     chunk_size = 10  # 키워드를 10개씩 나누어 요청
 
-    if include_related:
-        # 연관검색어 포함 시, API를 통해 연관 키워드를 가져옴
+    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:
@@ -116,8 +116,14 @@ def get_monthly_search_volumes(keywords, include_related):
                 except Exception as e:
                     print(f"Error fetching keywords data chunk: {e}")
     else:
-        # 연관검색어 미포함 시, 입력 키워드만 사용
-        all_data = [{'relKeyword': keyword, 'monthlyPcQcCnt': '0', 'monthlyMobileQcCnt': '0'} for keyword in keywords]
+        # 연관검색어를 포함하지 않으므로 입력 키워드만 처리
+        for keyword in keywords:
+            # 가상의 데이터 구조을 맞추기 위해
+            all_data.append({
+                'relKeyword': keyword,
+                'monthlyPcQcCnt': '0',  # 실제 API에서 값을 가져오려면 별도 요청 필요
+                'monthlyMobileQcCnt': '0'
+            })
 
     if not all_data:
         return [("Error", "데이터가 반환되지 않았거나 API 응답이 유효하지 않습니다.", "", "", "")]  # 블로그 문서 수 칼럼 추가
@@ -139,19 +145,33 @@ def get_monthly_search_volumes(keywords, include_related):
             total_searches = monthly_pc + monthly_mobile
             results.append((keyword, monthly_pc, monthly_mobile, total_searches))
         
-        if len(results) >= 100:
+        if len(results) >= 100 and include_related_keywords:
             break
 
-    # 블로그 문서 수 병렬 요청
-    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:
+        # 블로그 문서 수 병렬 요청
+        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")
+    else:
+        # 블로그 문서 수 병렬 요청 (연관검색어가 아닐 경우 각 키워드에 대해)
+        with ThreadPoolExecutor(max_workers=5) as executor:
+            blog_futures = [executor.submit(get_blog_count_parallel, keyword) for keyword in results]
+            temp_results = []
+            for future in blog_futures:
+                try:
+                    keyword, blog_count = future.result()
+                    temp_results.append((keyword, 0, 0, 0, blog_count))
+                except Exception as e:
+                    print(f"Error fetching blog count for keyword '{keyword}': {e}")
+                    temp_results.append((keyword, 0, 0, 0, "Error"))
+            results = temp_results
 
     return results
 
@@ -166,7 +186,7 @@ def save_to_excel(results, keyword):
 
 def display_search_volumes(keywords, include_related):
     keyword_list = [keyword.strip() for keyword in keywords.split(',')]
-    results = get_monthly_search_volumes(keyword_list, include_related)
+    results = get_monthly_search_volumes(keyword_list, include_related_keywords=include_related)
     file_path = save_to_excel(results, keywords)
     return results, file_path
 
@@ -174,14 +194,14 @@ iface = gr.Interface(
     fn=display_search_volumes, 
     inputs=[
         gr.Textbox(placeholder="키워드를 입력하세요"),
-        gr.Checkbox(label="연관검색어 포함", value=True)  # 연관검색어 ON/OFF 체크박스 추가
+        gr.Checkbox(label="연관검색어 포함", value=True)  # 연관검색어 토글 추가
     ], 
     outputs=[
         gr.Dataframe(headers=["키워드", "PC월검색량", "모바일월검색량", "토탈월검색량", "블로그문서수"]),
         gr.File(label="다운로드 엑셀 파일")
     ],
     title="네이버 월검색량 검색기",
-    description="키워드를 입력하고 연관검색어 포함 여부를 선택하세요."
+    description="키워드의 월 검색량과 블로그 문서 수를 확인할 수 있습니다. 연관검색어를 포함할지 선택하세요.",
 )
 
 iface.launch(share=True)  # share=True를 추가하여 공개 링크 생성