Update app.py

app.py

@@ -2,8 +2,27 @@ import streamlit as st
 from pytrends.request import TrendReq
 import pandas as pd
 import matplotlib.pyplot as plt
-from modules.pytrendsexpander import pytrendsExpander
 
+class pytrendsExpander:
+    def __init__(self, keyword):
+        self.keyword = keyword
+        self.pytrends = TrendReq()
+        self.df = pd.DataFrame()
+
+    def expand_search(self, m):
+        pytrends = self.pytrends
+
+        # Get seed keyword
+        kw_list = [self.keyword]
+        pytrends.build_payload(kw_list, cat=0, timeframe='today 1-m', geo='US', gprop='')
+
+        # Get results from the seed keyword
+        queries = pytrends.related_queries()
+
+        # Store the results in the data frame
+        df_results = pd.DataFrame(queries.get(self.keyword).get('top').rename_axis('ranking').reset_index())
+
+        return df_results.head(m)
 
 # Set up the pytrends API
 pytrends = TrendReq(hl='en-US', tz=360)