diff --git a/lib/ai_web_researcher/google_trends_researcher.py b/lib/ai_web_researcher/google_trends_researcher.py index aa6ce6b6..0bb86371 100644 --- a/lib/ai_web_researcher/google_trends_researcher.py +++ b/lib/ai_web_researcher/google_trends_researcher.py @@ -339,6 +339,11 @@ def get_related_topics_and_save_csv(search_keywords): logger.error(f"Failed to get pytrends realted topics: {err}") 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 top_topics = list(data.values())[0]['top'] rising_topics = list(data.values())[0]['rising']