Update app.py

app.py

@@ -17,5 +17,37 @@ with st.sidebar:
     selected_year = st.selectbox('Select a year', year_list, index=len(year_list)-1)
     df_selected_year = df[df.Year == selected_year]
 
+    cik_list = list(df.CIK.unique())[::-1]
+    cik_list = sorted(cik_list)
+
+    selected_cik = st.selectbox('Select a company', cik_list, index=len(cik_list)-1)
+    df_selected_cik = df[df.CIK == selected_cik]
+
     color_theme_list = ['blues', 'cividis', 'greens', 'inferno', 'magma', 'plasma', 'reds', 'rainbow', 'turbo', 'viridis']
-    selected_color_theme = st.selectbox('Select a color theme', color_theme_list)
+    selected_color_theme = st.selectbox('Select a color theme', color_theme_list)
+
+def make_esgScoresPlot(df, cik)
+    to_display = df[df['CIK'] == cik]
+    to_display['Year'] = to_display['Year'].astype('string')
+    to_display.sort_values('Year', inplace=True)
+    to_display.set_index('Year', inplace=True)
+
+    # display subset
+    print(to_display)
+    print(to_display.index.dtype)
+
+    ax = sns.lineplot(data=to_display[['e_score', 's_score', 'g_score']] )
+    ax.xlabel('year')
+    ax.ylabel('score')
+    ax.title('ESG scores for ')
+    ax.legend(title='Legend', loc='upper left')
+    return ax
+
+col = st.columns((1.5, 4.5, 2), gap='medium')
+
+with col[1]:
+    st.markdown('#### ESG Scores')
+    
+    esgScoresPlot = make_esgScoresPlot(df, selected_cik)
+    st.altair_chart(esgScoresPlot, use_container_width=True)
+