Update appStore/tapp_display.py

appStore/tapp_display.py

@@ -49,8 +49,8 @@ def target_display():
             count_ghg = sum(hits['GHGLabel'] == True)
             count_netzero = sum(hits['NetzeroLabel'] == True)
             count_nonghg = sum(hits['NonGHGLabel'] == True)
-            count_Mitigation = sum(hits['MitigationLabel'] == True)
-            count_Adaptation = sum(hits['AdaptationLabel'] == True)
+            count_mitigation = sum(hits['MitigationLabel'] == True)
+            count_adaptation = sum(hits['AdaptationLabel'] == True)
             
 
             c1, c2 = st.columns([1,1])
@@ -75,3 +75,61 @@ def target_display():
             st.dataframe(hits)
         else:
             st.info("🤔 No Targets Found")
+
+
+
+def targets():
+    if 'key1' in st.session_state:
+        df = st.session_state['key1'].copy()
+        idx = df['NetzeroLabel_Score'].idxmax()
+        netzero_placeholder = df.loc[idx, 'text']
+        df = df.drop(df.filter(regex='Score').columns, axis=1)
+        df = df[df.TargetLabel==True].reset_index(drop=True)
+        df['keep'] = True
+        df.drop(columns = ['ActionLabel','PolicyLabel','PlansLabel'], inplace=True)
+        st.session_state['target_hits'] = df
+        st.session_state['netzero'] = netzero_placeholder
+    
+def target_display():
+    if 'key1' in st.session_state:
+        st.caption(""" **{}** is splitted into **{}** paragraphs/text chunks."""\
+                          .format(os.path.basename(st.session_state['filename']),
+                                 len(st.session_state['key0'])))   
+        
+        hits  = st.session_state['target_hits']
+        if len(hits) !=0:
+            # collecting some statistics
+            count_target = sum(hits['TargetLabel'] == True)
+            count_ghg = sum(hits['GHGLabel'] == True)
+            count_netzero = sum(hits['NetzeroLabel'] == True)
+            count_nonghg = sum(hits['NonGHGLabel'] == True)
+            count_mitigation = sum(hits['MitigationLabel'] == True)
+            count_adaptation = sum(hits['AdaptationLabel'] == True)
+            
+
+            c1, c2 = st.columns([1,1])
+            with c1:
+                st.write('**Target Related Paragraphs**: `{}`'.format(count_target))
+                st.write('**Netzero Related Paragraphs**: `{}`'.format(count_netzero))
+                st.write('**Mitigation Related Paragraphs**: `{}`'.format(count_mitigation))
+            with c2:
+                st.write('**GHG Target Related Paragraphs**: `{}`'.format(count_ghg))
+                st.write('**NonGHG Target Related Paragraphs**: `{}`'.format(count_nonghg))
+                st.write('**Adaptation Related Paragraphs**: `{}`'.format(count_adaptation))
+            st.write('----------------')
+            st.markdown("<h4 style='text-align: left; color: black;'> Sectoral Target Related Paragraphs Count </h4>", unsafe_allow_html=True)
+            cols = list(hits.columns)
+            sector_cols = list(set(cols) - {'TargetLabel','MitigationLabel','AdaptationLabel','GHGLabel','NetzeroLabel','NonGHGLabel','text','keep','page'})
+            placeholder= []
+            for col in sector_cols:
+                placeholder.append({'Sector':col,'Count':sum(hits[col] == True)})
+            sector_df = pd.DataFrame.from_dict(placeholder)
+            fig = px.bar(sector_df, x='Sector', y='Count')
+            st.plotly_chart(fig,use_container_width= True)
+                
+            st.dataframe(hits)
+        else:
+            st.info("🤔 No Targets Found")
+
+
+