#!/usr/local/bin/python
import csv
import igraph
from igraph import *
from itertools import chain
#reader = csv.DictReader(open("wekarules-k-k-h-all.csv"), dialect="excel")
#reader = csv.DictReader(open("drug-hash-4viz.csv"), dialect="excel")
#reader = csv.DictReader(open("mjd.ALL.2162014.csv"), dialect="excel")
reader = csv.DictReader(open("network-sample-problematic.csv"), dialect="excel")#network-sample-problematic.csv#kh-all-graph-validated.csv
g = igraph.Graph.DictList(vertices=None, edges=reader)
#del g.es()[None]
g.to_undirected()
g.simplify()

layout = g.layout_fruchterman_reingold() 

def is_hashtag(vertex):
    return vertex["name"].startswith("#")
                    
visual_style = {}

visual_style["vertex_color"] = ["red" if is_hashtag(vertex) else "blue" for vertex in g.vs]
visual_style["vertex_size"] = [11 if is_hashtag(vertex) else 23 for vertex in g.vs]
visual_style["vertex_label"] = ["" if is_hashtag(vertex) else g.vs["name"] for vertex in g.vs]
visual_style["vertex_label_size"] = 7
visual_style["edge_width"] = 1
visual_style["edge_color"] =["pink"]
visual_style["layout"] = layout
visual_style["bbox"] = (1700, 1700)
visual_style["margin"] = 50
plot(g, "/Users/ahamed/Publications/JBI-SI-GoodTemplate/ddi-network-vis-kh-labeled.pdf",**visual_style)
#plot(g, **visual_style)