Update app.py

app.py

@@ -9,10 +9,18 @@ def analyze_coloring_trends(search_query='adult coloring books', region='US'):
     # Initialize Google Trends connection
     pytrends = TrendReq(hl='en-US', tz=360)
     
-    # Define a list of predefined categories (if user doesn't enter a custom query)
-    if search_query.lower() in ['adult coloring books', 'mandala coloring', 'animal coloring', 'flower coloring', 'kids coloring books']:
+    # Predefined categories for the search query
+    predefined_categories = [
+        'adult coloring books', 'mandala coloring', 
+        'animal coloring', 'flower coloring', 
+        'kids coloring books'
+    ]
+    
+    # If the search query matches one of the predefined categories, use it
+    if search_query.lower() in predefined_categories:
         coloring_categories = [search_query]
     else:
+        # Otherwise, use the user-provided search term as a custom query
         coloring_categories = [search_query]
 
     trending_keywords = defaultdict(int)
@@ -54,9 +62,8 @@ def analyze_coloring_trends(search_query='adult coloring books', region='US'):
     # Create DataFrame from results
     df = pd.DataFrame(results)
     
-    # Check if DataFrame has any data
+    # If DataFrame is empty, handle the case gracefully
     if df.empty:
-        print("No data was retrieved.")
         return "No data available for the selected category in this region.", "", ""
     
     # Create bar chart visualization