Update app.py

app.py

@@ -28,19 +28,36 @@ with st.sidebar:
     selected_cik = st.selectbox('Select a company', cik_list, index=len(cik_list)-1)
     df_selected_cik = df[df.CIK == selected_cik]
 
-col = st.columns((1.5, 4.5, 2), gap='medium')
+col1, col2 = st.columns(2)
 
-with col[1]:
+with col1:
     st.markdown('#### ESG Scores')
     
-    to_display = df[df['CIK'] == selected_cik]
     df_selected_cik['Year'] = df_selected_cik['Year'].astype('string')
     df_selected_cik.sort_values('Year', inplace=True)
     df_selected_cik.set_index('Year', inplace=True)
 
-    lp = sns.lineplot(data=df_selected_cik[['e_score', 's_score', 'g_score']] )
-    plt.xlabel('year')
-    plt.ylabel('score')
-    plt.legend(title='Legend', loc='upper left')
-    st.pyplot(plt)
-    
+    sns.lineplot(data=df_selected_cik[['e_score', 's_score', 'g_score']] )
+    plt_lp.xlabel('year')
+    plt_lp.ylabel('score')
+    plt_lp.legend(title='Legend', loc='upper left')
+    st.pyplot(plt_lp)
+
+with col2:
+    st.markdown('### ESG Word Clouds')
+    
+    wc_list = ['E_Text', 'S_Text', 'G_Text']
+
+    for x in wc_list:
+        row = to_display.loc[selected_year]
+        text = row.loc[x]
+        text = re.sub(r'[^A-Za-z\s]', '', text)    
+        text = text.lower()
+        stopwords = set(STOPWORDS)
+        text = ' '.join(word for word in text.split() if word not in stopwords)
+        
+        wordcloud = WordCloud(width=800, height=400, background_color='white').generate(text)
+        
+        plt.figure(figsize=(10, 5))
+        plt.imshow(wordcloud, interpolation='bilinear')
+        plt.axis('off')