Update chart_generation.py

chart_generation.py

@@ -17,13 +17,13 @@ def create_bar_plot(file, x_values, y_values):
         plt.xlabel('First Word')
         plt.ylabel('Count')
     else:
-        count_df = df.groupby(['Type', 'Simple_Source']).size().unstack(fill_value=0)
+        count_df = df.groupby(['Simple_Source', y_values]).size().unstack(fill_value=0)
         fig, ax = plt.subplots(figsize=(10, 7))
 
         count_df.plot(kind='bar', stacked=True, ax=ax)
 
         
-        plt.legend(title='Shop', bbox_to_anchor=(1.05, 1), loc='upper left')
+        plt.legend(title=y_values, bbox_to_anchor=(1.05, 1), loc='upper left')
         
         plt.xlabel(y_values)
         plt.ylabel('Number of Contributions')