Update chart_generation.py

chart_generation.py

@@ -7,17 +7,19 @@ import numpy as np
 def create_bar_plot(file, x_values, y_values):
     df = pd.read_excel(file)
 
-    df['Simple_Source'] = df[x_values].apply(lambda x: x.split()[0].split(',')[0] if type(x)==str else np.nan)
+    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()
-        fig = plt.figure(figsize=(10, 6))
+        fig = plt.figure(figsize=(16, 7))
         counts.plot(kind='bar')
         plt.title(f'Count of First Words in {x_values}')
         plt.xlabel('First Word')
         plt.ylabel('Count')
     else:
-        count_df = df.groupby(['Simple_Source', y_values]).size().unstack(fill_value=0)
+        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)
         fig, ax = plt.subplots(figsize=(16, 7))
 
         count_df.plot(kind='bar', stacked=True, ax=ax)