kunthawat/ALwrity
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: handle empty data in get_related_topics_and_save_csv to prevent list index out of range error

Browse Source
This commit is contained in:
ajaysi (aider)
2024-09-14 19:21:25 +05:30
parent 5be849cfcb
commit 0ec58fc7c6
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lib/ai_web_researcher/google_trends_researcher.py
View File

@@ -339,6 +339,11 @@ def get_related_topics_and_save_csv(search_keywords):
logger.error(f"Failed to get pytrends realted topics: {err}") logger.error(f"Failed to get pytrends realted topics: {err}")
return None return None
# Check if data contains expected keys
if not data or 'top' not in list(data.values())[0] or 'rising' not in list(data.values())[0]:
logger.error("No related topics found.")
return pd.DataFrame()
# Extract data from the result # Extract data from the result
top_topics = list(data.values())[0]['top'] top_topics = list(data.values())[0]['top']
rising_topics = list(data.values())[0]['rising'] rising_topics = list(data.values())[0]['rising']