import matplotlib.pyplot as plt
import sys

filename = sys.argv[1]
title = sys.argv[2]

label=[]
value=[]
color = []
with open(filename,'r') as f:
    for line in f:
        tokens = line.split()
        c = tokens[-1]
        problem_data = ' '.join(tokens[0:-1])
        l,v = problem_data.split(':')
        if(int(v)>0):
            label.append(l+'('+v+')')
            value.append(v)
            color.append(c)


wedges, texts, autotexts = plt.pie(value, colors=color,autopct=lambda p: '{:.1f}%'.format(round(p)) if p > 0 else '',pctdistance=1.2,wedgeprops= {"edgecolor":"black",
                     'linewidth': 1,
                     'antialiased': False})

plt.legend(wedges, label,
          loc="center left",
          bbox_to_anchor=(1.1, 0, 0.5,1))
# Add title
plt.title(title,pad=20)
if(len(sys.argv)<=3):
    plt.legend(label, bbox_to_anchor=(1.1,0.5), loc="center right", fontsize=10, 
           bbox_transform=plt.gcf().transFigure,shadow=True)
else:
    plt.legend(label,loc="lower center", bbox_to_anchor=(0.5,-0.1), fontsize=10, 
           bbox_transform=plt.gcf().transFigure,shadow=True)
# Show plot
# plt.tight_layout()
plt.savefig(filename.replace('.txt','.png'),dpi=600,transparent=True, bbox_inches="tight")