def main(artist,song):
Blue = colour.Colour(0,0,255)
White = colour.Colour(255,255,255)
words_in_song = {}
filename = "%s-%s.%s"%(p(artist),p(song),filext)
if filename not in os.listdir("./graphs"):
Fetcher = GetLyrics()
lyrics = Fetcher.fetch(artist,song)
if lyrics is None:
print "No Results found"
return
lyriclist = lyrics.split(" ")
G = pgv.AGraph()
G.edge_attr['dir']='forward'
G.edge_attr['arrowtype'] = 'normal'
for index in range(0,len(lyriclist)):
if lyriclist[index] in words_in_song.keys():
words_in_song[lyriclist[index]] += 1
else:
words_in_song[lyriclist[index]] = 1
G.add_node(lyriclist[index])
if index+1 < len(lyriclist):
G.add_edge(lyriclist[index],lyriclist[index+1])
for x in words_in_song.keys():
scaleFactor = int(math.floor((float(words_in_song[x]) / float(len(words_in_song.keys()))) * 255))
G.get_node(x).attr['color'] = Blue.colourLerp(White,scaleFactor).toHex()
G.get_node(x).attr['style'] = "filled"
G.layout(prog='dot')
G.write("./graphs/%s.dot"%filename)
G.draw("./graphs/%s"%filename)
pp = pprint.PrettyPrinter()
pp.pprint(words_in_song)
return "./graphs/%s"%filename
else:
return "./graphs/%s"%filename
def compare(artist1,song1,artist2,song2):
Blue = colour.Colour(0,0,255)
Red = colour.Colour(255,0,0)
Magenta = colour.Colour(255,0,255)
White = colour.Colour(255,255,255)
words_in_song1 = {}
words_in_song2 = {}
filename = "%s-%s--vs--%s-%s.%s"%(p(artist1),p(song1),p(artist2),p(song2),filext)
if filename not in os.listdir("./graphs"):
songOne = GetLyrics()
songTwo = GetLyrics()
lyricsOne = songOne.fetch(artist1,song1)
time.sleep(1)
lyricsTwo = songTwo.fetch(artist2,song2)
if (lyricsOne is None) or (lyricsTwo is None):
print "No Results found"
return
lyricsOne_list = lyricsOne.split(" ")
lyricsTwo_list = lyricsTwo.split(" ")
G = pgv.AGraph()
G.edge_attr['dir']='forward'
G.edge_attr['arrowtype'] = 'normal'
for index in range(0,len(lyricsOne_list)):
if lyricsOne_list[index] in words_in_song1.keys():
words_in_song1[lyricsOne_list[index]] += 1
else:
words_in_song1[lyricsOne_list[index]] = 1
G.add_node(lyricsOne_list[index])
if index+1 < len(lyricsOne_list):
G.add_edge(lyricsOne_list[index],lyricsOne_list[index+1])
for index in range(0,len(lyricsTwo_list)):
if lyricsTwo_list[index] in words_in_song2.keys():
words_in_song2[lyricsTwo_list[index]] += 1
else:
words_in_song2[lyricsTwo_list[index]] = 1
G.add_node(lyricsTwo_list[index])
if index+1 < len(lyricsTwo_list):
G.add_edge(lyricsTwo_list[index],lyricsTwo_list[index+1])
for x in words_in_song1.keys():
scaleFactor = int(math.floor((float(words_in_song1[x]) / float(len(words_in_song1.keys()))) * 255))
G.get_node(x).attr['color'] = Blue.colourLerp(White,scaleFactor).toHex()
G.get_node(x).attr['style'] = "filled"
for x in words_in_song2.keys():
scaleFactor = int(math.floor((float(words_in_song2[x]) / float(len(words_in_song2.keys()))) * 255))
G.get_node(x).attr['color'] = Red.colourLerp(White,scaleFactor).toHex()
G.get_node(x).attr['style'] = "filled"
if x in words_in_song1:
G.get_node(x).attr['color'] = Magenta.colourLerp(White,scaleFactor).toHex()
G.layout(prog='dot')
G.write("./graphs/%s.dot"%filename)
G.draw("./graphs/%s"%filename)
#pp = pprint.PrettyPrinter()
#pp.pprint(words_in_song1)
#pp.pprint(words_in_song2)
return "./graphs/%s"%filename
else:
return "./graphs/%s"%filename