Update chart_generation.py

chart_generation.py

@@ -18,7 +18,10 @@ def create_bar_plot(file, x_values, y_values):
         plt.ylabel('Count')
     else:
         count_df = df.groupby(['Type', 'Simple_Source']).size().unstack(fill_value=0)
-        fig = count_df.plot(kind='bar', stacked=True, figsize=(10, 7))
+        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')