Update appStore/iki_sheets.py

appStore/iki_sheets.py

@@ -186,4 +186,42 @@ def mitigation_display():
             st.dataframe(hits[['keep','text','Parameter','page','Type']])
         else:
             st.info("🤔 No Tranport  Mitigation paragraph found")
+
+
+def adaptation():
+    if 'key1' in st.session_state:
+        df = st.session_state['key1'].copy()
+        df = np.where((df.ActionLabel==True) | (df.PolicyLabel==True) | (df.PlansLabel==True))
+        df = df.reset_index(drop=True)
+        df = df[(df.AdaptationLabel == True)&(df.Transport == True)].reset_index(drop=True)
+        non_target = ['Action','Policy','Plans'] 
+        df['Type'] = df.apply(lambda x: [i for i in non_target if x[i+'Label']==True],axis=1)
+        df['keep'] = True
+        st.session_state['adaptation_hits'] = df
+
+def mitigation_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['mitigation_hits']
+        if len(hits) !=0:
+            # collecting some statistics
+            count_adaptation = sum(hits['AdaptationLabel'] == True)
+            count_action = sum(hits['ActionLabel'] == True)
+            count_policy = sum(hits['PolicyLabel'] == True)
+            count_plans = sum(hits['PlansLabel'] == True)
+
+            c1, c2 = st.columns([1,1])
+            with c1:
+                st.write('**Adaptation Paragraphs**: `{}`'.format(count_adaptation))
+                st.write('**Transport Action Related Paragraphs**: `{}`'.format(count_action))
+            with c2:
+                st.write('**Transport Policy Related Paragraphs**: `{}`'.format(count_policy))
+                st.write('**Transport Plans Related Paragraphs**: `{}`'.format(count_plans))
+            st.write('----------------')
+            st.dataframe(hits[['keep','text','Parameter','page','Type']])
+        else:
+            st.info("🤔 No Tranport  Adaptation paragraph found")