Update chart_generation.py

chart_generation.py

@@ -10,7 +10,7 @@ def create_bar_plot(file, x_values, y_values):
     df['Simple_X'] = df[x_values].apply(lambda x: ' '.join(x.split(',')[0].split()[:2]) if type(x)==str else np.nan)
 
     if y_values == "":
-        counts = df['Simple_Source'].value_counts()
+        counts = df['Simple_X'].value_counts()
         fig = plt.figure(figsize=(16, 7))
         counts.plot(kind='bar')
         plt.title(f'Count of First Words in {x_values}')
@@ -19,7 +19,7 @@ def create_bar_plot(file, x_values, y_values):
     else:
         df['Simple_Y'] = df[y_values].apply(lambda x: ' '.join(x.split(',')[0].split()[:2]) if type(x)==str else np.nan)
         
-        count_df = df.groupby(['Simple_Source', 'Simple_Y']).size().unstack(fill_value=0)
+        count_df = df.groupby(['Simple_X', 'Simple_Y']).size().unstack(fill_value=0)
         fig, ax = plt.subplots(figsize=(16, 7))
 
         count_df.plot(kind='bar', stacked=True, ax=ax)