import numpy as np                   # for multi-dimensional containers 
import pandas as pd                  # for DataFrames
import itertools
from plotapi import Chord

username = "shahin@plotapi.com" 
password = "PLOTAPI-B-7ef36302-6439-40ec-8218-9d1e6d9fb0aa"
Chord.set_license(username, password)

data_url = 'pokemon.csv'
data = pd.read_csv(data_url)

data["url"] = "https://datacrayon.com/datasets/pokemon_img/"+data["image_name"]
#data['display_name'] = ""

data.head()

#data['name'] = data['name'].str.slice(0,10)

#banlist = [
#    " Form",
#    " Forme",
#    " Style",
#    " Pattern",
#    " Reversion",
#    " Cloak",
#    " Mode"
#]
#
#for idx, pok in data.iterrows():
#    if(pok['name'] in pok['form']):
#        data.at[idx, 'display_name'] = pok['form']
#    else:
#        data.at[idx, 'display_name'] = pok['form'] + " " + pok['name']
#
#    
#    for word in banlist:
#         data.at[idx, 'display_name'] = data.at[idx, 'display_name'].replace(word,"")
#    

data.loc[98]

data.shape
data.columns

pd.DataFrame(data.columns.values.tolist())

types = pd.DataFrame(data[['type_1', 'type_2']].values)
types

#types = types.dropna()

types = types.replace('\n','', regex=True)
types

names =np.unique(pd.concat([types[col] for col in types]).dropna()).tolist()
pd.DataFrame(names)

types

##for t in types:
#    matrix.at[t[0], t[1]] += 1
    
#matrix = matrix.values.tolist()

colors = ["#A6B91A", "#705746", "#6F35FC", "#F7D02C", "#D685AD",
          "#C22E28", "#EE8130", "#A98FF3", "#735797", "#7AC74C",
          "#E2BF65", "#96D9D6", "#A8A77A", "#A33EA1", "#F95587",
          "#B6A136", "#B7B7CE", "#6390F0"]

names

data = data.replace('\n','', regex=True)

data

data

data

data

data.loc[98]

matrix = pd.DataFrame(0, index=names, columns=names)

details = pd.DataFrame([[[] for i in range(len(names))] for i in range(len(names))], index=names, columns=names)

details_thumbs = pd.DataFrame([[[] for i in range(len(names))] for i in range(len(names))], index=names, columns=names)

details



names_nested = names.copy()
print
for count_x, item_x in enumerate(names):
    
    for count_y, item_y in enumerate(names_nested):
        details_urls = data[
            (data['type_1'].isin([item_x, item_y])) &
            (data['type_2'].isin([item_y, item_x]))]['url'].to_list()
        
        details_names = data[
            (data['type_1'].isin([item_x, item_y])) &
            (data['type_2'].isin([item_y, item_x]))]['display_name'].to_list()
        
        urls_names = np.column_stack((details_urls, details_names))
        if(urls_names.size > 0):
            details[item_y][item_x] = details_names
            details_thumbs[item_y][item_x] = details_urls
            
            #print(item_x + ":" + item_y)
            #print(details_names)
            matrix[item_y][item_x] = len(details_names)


        #else:
            #details[item_x][item_y] = []
            #details_thumbs[item_x][item_y] = []
   # names_nested.remove(item_x)
            
            
for count_x, item_x in enumerate(names):
    
    details_urls = data[
    (data['type_1'].isin([item_x])) &
    (data['type_2'].isna())]['url'].to_list()

    details_names = data[
        (data['type_1'].isin([item_x])) &
        (data['type_2'].isna())]['display_name'].to_list()

    urls_names = np.column_stack((details_urls, details_names))
    if(urls_names.size > 0):
        details[item_x][item_x] = details_names
        details_thumbs[item_x][item_x] = details_urls

        matrix[item_x][item_x] = len(details_urls)

    #else:
     #   details[item_x][item_x] = []
      #  details_thumbs[item_x][item_x] = []
    
details=pd.DataFrame(details).values.tolist()
details_thumbs=pd.DataFrame(details_thumbs).values.tolist()

names_nested

matrix.sum()

print(names)

type_colors = dict(zip(names, colors))
type_colors

for index, row in data.iterrows():
    dex = f"{row['pokedex_id']:03d}"
    url = f"https://www.pokemon.com/uk/pokedex/{dex}"
    data.at[index,'dex_site'] = f'<a href="{url}">{row["display_name"]}</a>'

data['dex_site']


data_table = data[['type_1', "type_2", "pokedex_id","display_generation","dex_site"]]
data_table["URL"] = '<img src="'+data['url'] +'">'


data_table = data_table.fillna('')

data_table["display_generation"] = '<span class="gen">'+data_table["display_generation"]+'</span>'



data_table["pokedex_id"] = '<span class="'+data_table["type_1"]+'">'+data_table["type_1"]+'</span> <span class="'+data_table["type_2"]+'">'+data_table["type_2"]+'</span>'

    
data_table.columns = ['Type 1', 'Type 2', 'Type', 'Gen', 'Name', '']


data_table

data

print(data_table)

matrix = matrix.values.tolist()
details_thumbs = details_thumbs
data_table = data_table.to_csv(index=False)

import json

data = {"matrix": matrix,
       "details_thumbs": details_thumbs,
       "data_table": data_table}

with open("pokemon_types.json", "w") as fp:
    json.dump(data, fp)

with open("pokemon_types.json", "r") as fp:
    b = json.load(fp)
    
b['matrix']

Chord(
    matrix,
    names,
    colors="monsters",
    details_thumbs=details_thumbs,
    margin=30,
    noun="Pokemon!",
    thumbs_width=50,
    curved_labels=True,
    thumbs_margin=1,
    popup_width=600,
    thumbs_font_size=10,
    credit=True,
    arc_numbers=True,
    reverse_gradients=False,
    data_table_column_width=80,
    data_table=data_table,
    animated_intro=True,
    data_table_show_indices=False
).show()

Chord(
    matrix.values.tolist(),
    names,
    colors="viridis",
    details=details,
    details_thumbs=details_thumbs,
    margin=80,
    noun="Pokédsssdasmdsadson",
    thumbs_width=50,
    thumbs_margin=1,
    popup_width=601,
    thumbs_font_size=10,
    credit=True,
    arc_numbers=True,
    reverse_gradients=False,
        data_table=data_table.to_csv( index=False)
).show()

Chord(
    matrix.values.tolist(),
    names,
    colors="viridis",
    details=details,
    details_thumbs=details_thumbs,
    margin=80,
    noun="Posdsxxsdsfsskémon",
    thumbs_width=50,
    thumbs_margin=1,
    popup_width=600,
    thumbs_font_size=10,
    credit=True,
    arc_numbers=True,
    reverse_gradients=False,
).show_png()

matrix.values.tolist()

matrix['Bug']