Update app.py

app.py

@@ -12,6 +12,10 @@ from concurrent.futures import ThreadPoolExecutor
 import os
 import tempfile
 from datetime import datetime
+from dotenv import load_dotenv  # dotenv 추가
+
+# .env 파일의 환경 변수를 로드합니다.
+load_dotenv()
 
 # 환경 변수에서 API 키와 시크릿 키를 불러옵니다.
 BASE_URL = "https://api.searchad.naver.com"
@@ -19,6 +23,14 @@ API_KEY = os.environ.get("NAVER_API_KEY")
 SECRET_KEY = os.environ.get("NAVER_SECRET_KEY")
 CUSTOMER_ID = 2666992
 
+# 환경 변수에서 클라이언트 ID와 시크릿을 불러옵니다.
+CLIENT_ID = os.environ.get("NAVER_CLIENT_ID")
+CLIENT_SECRET = os.environ.get("NAVER_CLIENT_SECRET")
+
+# 환경 변수 로드 확인
+if not API_KEY or not SECRET_KEY or not CLIENT_ID or not CLIENT_SECRET:
+    raise ValueError("필수 환경 변수가 설정되지 않았습니다. .env 파일을 확인하세요.")
+
 class NaverAPI:
     def __init__(self, base_url, api_key, secret_key, customer_id):
         self.base_url = base_url
@@ -58,20 +70,25 @@ class NaverAPI:
         return response.json()
 
 def get_blog_count(keyword):
-    client_id = "421ZKFMM5TS1xmvsF7C0"
-    client_secret = "h47UQHAOGV"
+    # 클라이언트 ID와 시크릿을 환경 변수에서 불러옵니다.
+    client_id = CLIENT_ID
+    client_secret = CLIENT_SECRET
     encText = urllib.parse.quote(keyword)
     url = "https://openapi.naver.com/v1/search/blog?query=" + encText
     request = urllib.request.Request(url)
     request.add_header("X-Naver-Client-Id", client_id)
     request.add_header("X-Naver-Client-Secret", client_secret)
-    response = urllib.request.urlopen(request)
-    rescode = response.getcode()
-    if rescode == 200:
-        response_body = response.read()
-        data = json.loads(response_body.decode('utf-8'))
-        return data['total']
-    else:
+    try:
+        response = urllib.request.urlopen(request)
+        rescode = response.getcode()
+        if rescode == 200:
+            response_body = response.read()
+            data = json.loads(response_body.decode('utf-8'))
+            return data['total']
+        else:
+            return 0
+    except Exception as e:
+        print(f"Error fetching blog count for keyword '{keyword}': {e}")
         return 0
 
 def get_keywords_data_chunk(chunk):
@@ -89,9 +106,12 @@ def get_monthly_search_volumes(keywords):
     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:
-            data = future.result()
-            if 'keywordList' in data:
-                all_data.extend(data['keywordList'])
+            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 응답이 유효하지 않습니다.", "", "", "")]  # 블로그 문서 수 칼럼 추가
@@ -120,9 +140,13 @@ def get_monthly_search_volumes(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):
-            keyword, blog_count = future.result()
-            results[i] = (results[i][0], results[i][1], results[i][2], results[i][3], blog_count)
-    
+            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")
+
     return results
 
 def save_to_excel(results, keyword):
@@ -150,4 +174,4 @@ iface = gr.Interface(
     title="네이버 월검색량 검색기",
 )
 
-iface.launch()
+iface.launch(share=True)  # share=True를 추가하여 공개 링크 생성