Update app.py

app.py

@@ -115,25 +115,29 @@ def get_search_volumes(keyword):
     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)
-            
-            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)
+            # 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)
+
+                    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:
@@ -219,10 +223,10 @@ def get_monthly_search_volumes(keywords, include_related_keywords=True):
             for future in blog_futures:
                 try:
                     keyword, blog_count = future.result()
-                    temp_results.append((keyword, 0, 0, 0, blog_count))
+                    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, 0, 0, 0, "Error"))
+                    temp_results.append((keyword, results[0][1], results[0][2], results[0][3], "Error"))
             results = temp_results
 
     return results