Update app.py

app.py

@@ -5,17 +5,16 @@ from collections import defaultdict
 import plotly.express as px
 import time
 
-def analyze_coloring_trends(region='US'):
+def analyze_coloring_trends(search_query='adult coloring books', region='US'):
     # Initialize Google Trends connection
     pytrends = TrendReq(hl='en-US', tz=360)
     
-    # Define categories to analyze
-    coloring_categories = [
-        'adult coloring books', 'mandala coloring', 
-        'animal coloring', 'flower coloring', 
-        'kids coloring books'
-    ]
-    
+    # 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']:
+        coloring_categories = [search_query]
+    else:
+        coloring_categories = [search_query]
+
     trending_keywords = defaultdict(int)
     results = []
 
@@ -58,7 +57,7 @@ def analyze_coloring_trends(region='US'):
     # Check if DataFrame has any data
     if df.empty:
         print("No data was retrieved.")
-        return "No data available for the selected categories in this region.", "", ""
+        return "No data available for the selected category in this region.", "", ""
     
     # Create bar chart visualization
     fig = px.bar(
@@ -79,8 +78,6 @@ def analyze_coloring_trends(region='US'):
     recommendations = f"""
     Top Trending Categories in {region}:
     1. {coloring_categories[0]}
-    2. {coloring_categories[1]}
-    3. {coloring_categories[2]}
     
     Recommended Actions:
     - Focus on these trending keywords
@@ -91,14 +88,19 @@ def analyze_coloring_trends(region='US'):
     
     return fig, top_keywords_str, recommendations
 
-# Create Gradio interface with a refined list of supported regions
+# Create Gradio interface with search bar
 iface = gr.Interface(
     fn=analyze_coloring_trends,
     inputs=[
+        gr.Textbox(
+            label="Enter Search Term",
+            value="adult coloring books",  # Default search term
+            placeholder="e.g. adult coloring books, animal coloring, etc."
+        ),
         gr.Dropdown(
             choices=['US', 'UK', 'CA', 'AU', 'DE', 'FR', 'ES', 'IT', 'JP', 'BR', 'IN'],
             label="Select Region",
-            value="US"  # Set default region using 'value'
+            value="US"  # Set default region
         )
     ],
     outputs=[