Update appStore/tapp_display.py

appStore/tapp_display.py

@@ -127,4 +127,93 @@ def action_display():
             st.info("🤔 No Actions Found")
 
 
+
+def policy():
+    if 'key1' in st.session_state:
+        df = st.session_state['key1'].copy()
+        df = df.drop(df.filter(regex='Score').columns, axis=1)
+        df = df[df.PolicyLabel==True].reset_index(drop=True)
+        df['keep'] = True
+        df.drop(columns = ['TargetLabel','ActionLabel','PlansLabel','GHGLabel','NetzeroLabel','NonGHGLabel'], inplace=True)
+        st.session_state['policy_hits'] = df
+    
+def policy_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['policy_hits']
+        if len(hits) !=0:
+            # collecting some statistics
+            count_action = sum(hits['PolicyLabel'] == True)
+            count_mitigation = sum(hits['MitigationLabel'] == True)
+            count_adaptation = sum(hits['AdaptationLabel'] == True)
+            
+
+            c1, c2 = st.columns([1,1])
+            with c1:
+                st.write('**Policy Related Paragraphs**: `{}`'.format(count_action))
+                st.write('**Mitigation Related Paragraphs**: `{}`'.format(count_mitigation))
+            with c2:
+                st.write('**Adaptation Related Paragraphs**: `{}`'.format(count_adaptation))
+            st.write('----------------')
+            st.markdown("<h4 style='text-align: left; color: black;'> Sectoral Policy Related Paragraphs Count </h4>", unsafe_allow_html=True)
+            cols = list(hits.columns)
+            sector_cols = list(set(cols) - {'PolicyLabel','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 Policy Found")
+
+def plans():
+    if 'key1' in st.session_state:
+        df = st.session_state['key1'].copy()
+        df = df.drop(df.filter(regex='Score').columns, axis=1)
+        df = df[df.PlansLabel==True].reset_index(drop=True)
+        df['keep'] = True
+        df.drop(columns = ['TargetLabel','PolicyLabel','ActionLabel','GHGLabel','NetzeroLabel','NonGHGLabel'], inplace=True)
+        st.session_state['plan_hits'] = df
+    
+def plans_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['plan_hits']
+        if len(hits) !=0:
+            # collecting some statistics
+            count_action = sum(hits['PlansLabel'] == True)
+            count_mitigation = sum(hits['MitigationLabel'] == True)
+            count_adaptation = sum(hits['AdaptationLabel'] == True)
+            
+
+            c1, c2 = st.columns([1,1])
+            with c1:
+                st.write('**Plans Related Paragraphs**: `{}`'.format(count_action))
+                st.write('**Mitigation Related Paragraphs**: `{}`'.format(count_mitigation))
+            with c2:
+                st.write('**Adaptation Related Paragraphs**: `{}`'.format(count_adaptation))
+            st.write('----------------')
+            st.markdown("<h4 style='text-align: left; color: black;'> Sectoral Plans Related Paragraphs Count </h4>", unsafe_allow_html=True)
+            cols = list(hits.columns)
+            sector_cols = list(set(cols) - {'PlanLabel','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 Plans Found")
+