UI fix

appStore/target.py

@@ -85,6 +85,12 @@ def target_display():
     if  'key1' in st.session_state:
         df = st.session_state.key1
         hits  = df[df['Target Label'] == 'TARGET']
+        _lab_dict = {
+                      'LABEL_0':'NEGATIVE',
+                      'LABEL_1':'NOT GHG',
+                      'LABEL_2':'GHG',
+                      }
+        hits['GHG Label'] = hits['GHG Label'].apply(lambda i: _lab_dict[i])
         range_val = min(5,len(hits))
         if range_val !=0:
             count_target = sum(hits['Target Label'] == 'TARGET')
@@ -117,7 +123,7 @@ def target_display():
             hits = hits.sort_values(by=['Relevancy'], ascending=False)
             netzerohit = hits[hits['Netzero Label'] == 'NETZERO']
             if not netzerohit.empty:
-                netzero = netzero.sort_values(by = ['Netzero Score'], ascending = False)
+                netzerohit = netzerohit.sort_values(by = ['Netzero Score'], ascending = False)
                 st.markdown("###### Netzero paragraph ######")
                 st.write('** Text `page {}`: {}'.format(netzerohit.iloc[i]['page'],
                                 netzerohit.iloc[i]['text'].replace("\n", " ")))                        
@@ -132,10 +138,12 @@ def target_display():
                 # the page number reflects the page that contains the main paragraph 
                 # according to split limit, the overlapping part can be on a separate page
                 st.write('**Result {}** (Relevancy Score: {:.2f}): `page {}`, `Sector: {}`,\
-                            `GHG:`'\
+                            `GHG: {}`, `Adapt-Mitig :{}`'\
                     .format(i+1,hits.iloc[i]['Relevancy'],
-                            hits.iloc[i]['page'], hits.iloc[i]['Sector Label']
-                            ))                        
+                            hits.iloc[i]['page'], hits.iloc[i]['Sector Label'],
+                            hits.iloc[i]['GHG Label'],hits.iloc[i]['Adapt-Mitig Label']))                        
                 st.write("\t Text: \t{}".format(hits.iloc[i]['text'].replace("\n", " ")))
+            hits = hits.reset_index(drop =True)
+            st.write(hits)
         else:
             st.info("🤔 No Targets found")