from pytrends.request import TrendReq

# Only need to run this once, the rest of requests will use the same session.
pytrend = TrendReq()

# Get Google Keyword Suggestions

search_term = 'Dragonite'
suggestions_dict = pytrend.suggestions(keyword=search_term)
#keyword = next((item for item in suggestions_dict if item["type"] == "Pokemon species"), search_term)
suggestions_dict

/g/1215jfxl

pytrend.build_payload(kw_list=['/m/01nrwb'], timeframe='all')
pokemon_interest_over_time = pytrend.interest_over_time()
pokemon_interest_over_time.drop('isPartial', axis=1, inplace=True)
pokemon_interest_over_time


pytrend.build_payload(kw_list=['/m/01nrwb', '/g/121xk3zq'], timeframe='all')

# Interest Over Time
interest_over_time_df = pytrend.interest_over_time()
interest_over_time_df.drop('/m/01nrwb', axis=1, inplace=True)
interest_over_time_df.drop('isPartial', axis=1, inplace=True)
interest_over_time_df 

pokemon_interest_over_time = pd.concat([pokemon_interest_over_time, interest_over_time_df], axis=1)

pokemon_interest_over_time.loc['2019-12-01']

import pandas as pd

df = pd.read_csv("pokemon.csv")
df.head()

for index, row in df.iterrows():
    dex = f"{row['pokedex_number']:03d}"
    url = f"https://datacrayon.com/dataset/pokemon_thumbs/{dex}.png"
    df.at[index,'URL'] = url
    
    

df = df.drop_duplicates(subset=['pokedex_number'])

df.at[40,'name']

for index, row in df.iterrows():
    search_term = row['name']
    suggestions_dict = pytrend.suggestions(keyword=search_term)
    keyword = next((item for item in suggestions_dict if item["type"].lower() == "Pokemon species".lower()), None)
    
    if(keyword == None):
        keyword = next((item for item in suggestions_dict if "Pokemon species".lower() in item["type"].lower()), None)


    
    print(f"{index} - {search_term} > {keyword}")
    
    df.at[index,'keyword'] = keyword['mid'] if keyword != None else search_term
    
    

df.to_csv("pokemon_trends_keywords_unmodified.csv")

df = pd.read_csv("pokemon_trends_keywords.csv")

pokemon_interest_over_time = pd.read_csv("pokemon_interest_over_time.csv", index_col='date')

pokemon_interest_over_time.index = pokemon_interest_over_time.index.astype(str)
pokemon_interest_over_time.index.dtype

pokemon_interest_over_time.drop('/m/014wp5', axis=1, inplace=True)
pokemon_interest_over_time.drop('/m/014wp5.1', axis=1, inplace=True)
#pokemon_interest_over_time.drop('/m/040nd1', axis=1, inplace=True)
pokemon_interest_over_time.drop('/g/1215jfxl.1', axis=1, inplace=True)
pokemon_interest_over_time.drop('/g/1215jfxl', axis=1, inplace=True)
pokemon_interest_over_time.drop('/m/040nd1', axis=1, inplace=True)

#pytrend.build_payload(kw_list=['/m/01nrwb'], timeframe='all')
#pokemon_interest_over_time = pytrend.interest_over_time()
#pokemon_interest_over_time.drop('isPartial', axis=1, inplace=True)
#pokemon_interest_over_time

#pokemon_interest_over_time.drop('/m/0279f9', axis=1, inplace=True)




import time

for index, row in df.iterrows():
    search_term = row['keyword']
    if((search_term != "/m/01nrwb") and search_term not in pokemon_interest_over_time.columns):
        print(f"{index} - {row['name']}, {row['keyword']}")
        pytrend.build_payload(kw_list=['/m/01nrwb', search_term], timeframe='all')

        # Interest Over Time
        interest_over_time_df = pytrend.interest_over_time()
        interest_over_time_df.drop('/m/01nrwb', axis=1, inplace=True)
        interest_over_time_df.drop('isPartial', axis=1, inplace=True)
        interest_over_time_df 
        
        interest_over_time_df.index = interest_over_time_df.index.astype(str)
        pokemon_interest_over_time = pd.concat([pokemon_interest_over_time, interest_over_time_df], axis=1)
        pokemon_interest_over_time.to_csv("pokemon_interest_over_timeplus.csv")
        time.sleep(2.4)
    

    

pokemon_interest_over_time

import pandas as pd
df = pd.read_csv("pokemon_trends_keywords.csv")
pokemon_interest_over_time = pd.read_csv("pokemon_interest_over_timeplus.csv", index_col="date")
pokemon_interest_over_time = pokemon_interest_over_time.cumsum()

names = ["Bug", "Dark", "Dragon", "Electric", "Fairy", "Fighting", "Fire", "Flying", "Ghost", "Grass", "Ground", "Ice", "Normal", "Poison", "Psychic", "Rock", "Steel", "Water"]
colors = ["#A6B91A", "#705746", "#6F35FC", "#F7D02C", "#D685AD", "#C22E28", "#EE8130", "#A98FF3", "#735797", "#7AC74C", "#E2BF65", "#96D9D6", "#A8A77A", "#A33EA1", "#F95587", "#B6A136", "#B7B7CE", "#6390F0"]
colors[names.index(type1)]

from colorthief import ColorThief
import urllib.request

urllib.request.urlretrieve('https://datacrayon.com/datasets/pokemon_thumbs/151.png', "pokeimg/151.png")

color_thief = ColorThief('./pokeimg/151.png')
dominant_color = color_thief.get_color(quality=1)
color = '#%02x%02x%02x' % dominant_color

from colorthief import ColorThief
from pathlib import Path


samples=[]
nodes=[]

for col in pokemon_interest_over_time.columns:
    #print(col)
    pokemon = df.loc[df['keyword'] == col]
    name = pokemon.name.values[0]

    dex = pokemon.pokedex_number.values[0]

    #print(name)
    type1 = pokemon.type_1.values[0]

    url = pokemon.URL.values[0]


    # COLOR THIEF
    my_file = Path(f"pokeimg/{dex}.png")

    if not my_file.is_file():
        urllib.request.urlretrieve(url, f"pokeimg/{dex}.png")

    color_thief = ColorThief(f'./pokeimg/{dex}.png')
    dominant_color = color_thief.get_color(quality=1)
    color = '#%02x%02x%02x' % dominant_color

    
    nodes.append({"name":name,
              #    "color": colors[names.index(type1)],
                  "color": color,
                 "icon": url})    
for index, row in pokemon_interest_over_time.iterrows():
    print(index)
    for col in pokemon_interest_over_time.columns:
        pokemon = df.loc[df['keyword'] == col]
        name = pokemon.name.values[0]
                          
        samples.append({"order": f'{index.replace("-", ".")[:-3].split(".")[0]}.{index.replace("-", ".")[:-3].split(".")[1]}',
                       "name": name,
                       "value":row[col]})

url

sequence = list(dict.fromkeys([d['order'] for d in samples]))
sequence.sort()
sequence

name

from plotapi import BarFight

BarFight.set_license("your username", "your license key")

BarFight(samples,
         interval=500, icon_padding=0,nodes=nodes, colors="rainbow", height=1000, top_n=20,
         background_color="white", border="10px solid #C9605A", skip_first=True).show()