Update chart_generation.py

chart_generation.py

@@ -18,7 +18,7 @@ def create_bar_plot(file, x_values, y_values):
         plt.ylabel('Count')
     else:
         count_df = df.groupby(['Simple_Source', y_values]).size().unstack(fill_value=0)
-        fig, ax = plt.subplots(figsize=(10, 7))
+        fig, ax = plt.subplots(figsize=(12, 7))
 
         count_df.plot(kind='bar', stacked=True, ax=ax)
 
@@ -29,11 +29,13 @@ def create_bar_plot(file, x_values, y_values):
         plt.ylabel('Number of Contributions')
         plt.title(f'Number of Contributions by {y_values} and {x_values}')
 
-        for i, bar in enumerate(fig.patches):
+        for i, bar in enumerate(ax.patches):
             h = bar.get_height()
             w = bar.get_width()
             x = bar.get_x()
             y = bar.get_y()
             if h > 0:
-                plt.text(x + w/2, y + h/2, int(h), ha='center', va='center')
+                ax.text(x + w/2, y + h/2, int(h), ha='center', va='center')
+
+        plt.tight_layout()
     return fig