Update app.py

app.py

@@ -1,67 +1,42 @@
 import gradio as gr
 from pytrends.request import TrendReq
-import pandas as pd
-import plotly.express as px
-import time
+import matplotlib.pyplot as plt
+import os
 
-# Function to fetch trends and plot data
-def fetch_trends(search_query='adult coloring books', region='US'):
-    pytrends = TrendReq(hl='en-US', tz=360)
+def get_trends(keyword):
+    # Initialize the pytrends object
+    pytrends = TrendReq(hl="en-US", tz=360)
     
-    # Set up the payload with the given query and region
-    pytrends.build_payload([search_query], cat=0, timeframe='today 12-m', geo=region)
+    # Build the payload and fetch the trends data
+    pytrends.build_payload(kw_list=[keyword])
+    df = pytrends.interest_over_time()
     
-    # Fetch related queries
-    related_queries = pytrends.related_queries()
+    # If there is no data, return a message
+    if df.empty:
+        return "No data available for this keyword. Please try a different one."
     
-    # Check if there are any top related queries
-    if related_queries and related_queries.get(search_query, {}).get('top') is not None:
-        # Process the top related queries into a dataframe
-        df = related_queries[search_query]['top']
-        df = df[['query', 'value']]  # Filter only relevant columns
-        df = df.nlargest(10, 'value')  # Get top 10 trending queries
-
-        # Create the Plotly bar chart for visualization
-        fig = px.bar(
-            df, x='query', y='value', 
-            title=f"Top 10 Trending Keywords for '{search_query}' in {region}",
-            labels={'query': 'Keyword', 'value': 'Popularity Score'},
-            color='value', color_continuous_scale='Viridis'
-        )
-        fig.update_layout(xaxis_tickangle=-45)
-        
-        # Convert dataframe to a string for display
-        top_keywords_str = df.to_string(index=False)
-        
-        # Return the plot and the keyword list
-        return fig, top_keywords_str
+    # Create a plot
+    plt.figure(figsize=(20, 10))
+    df[keyword].plot()
+    plt.title(f"Google Trends for '{keyword}'", fontsize=20)
+    plt.xlabel("Date", fontsize=16)
+    plt.ylabel("Interest over time", fontsize=16)
+    plt.grid(True)
+    
+    # Save the plot to a temporary file
+    plot_file = "/tmp/trends_plot.png"
+    plt.savefig(plot_file)
+    plt.close()  # Close the plot to avoid display issues
     
-    else:
-        return None, "No data found for the query in this region."
+    return plot_file  # Return the file path for Gradio to display the image
 
-# Create Gradio Interface
+# Set up the Gradio interface
 iface = gr.Interface(
-    fn=fetch_trends,
-    inputs=[
-        gr.Textbox(
-            label="Enter Search Term",
-            value="adult coloring books",  # Default search term
-            placeholder="e.g. adult coloring books, mandala coloring, etc."
-        ),
-        gr.Dropdown(
-            choices=['US', 'UK', 'CA', 'AU', 'DE', 'FR', 'ES', 'IT', 'JP', 'BR', 'IN'],
-            label="Select Region",
-            value="US"  # Set default region
-        )
-    ],
-    outputs=[
-        gr.Plot(label="Trending Keywords Visualization"),
-        gr.Textbox(label="Top 10 Trending Keywords", lines=10)
-    ],
-    title="Google Trends Keyword Analyzer",
-    description="Analyze trending keywords for any search term in various regions using Google Trends",
-    theme="default"
+    fn=get_trends,
+    inputs=gr.Textbox(lines=1, label="Enter a keyword"),
+    outputs=gr.Image(type="file"),  # Output is now an image file
+    title="Google Trends Visualizer",
+    description="Enter a keyword to see its Google Trends data over time."
 )
 
-# Launch the interface
 iface.launch()