functionality for user feedback

app.py

@@ -38,6 +38,8 @@ related_summaries = [[ex1]]
 related_chosen = []
 related_attn = []
 related_clr_bts = []
+correct_drg_text_list = []
+correct_salient_words_list = []
 
 rule_df, drg2idx, i2d, d2mdc, d2w = load_rule('MSDRG_RULE13.csv')
 
@@ -116,7 +118,9 @@ def run(text, related_discharges=False):
         gr.ClearButton.update(visible=True),
         gr.TextArea.update(visible=True),
         gr.Button.update(visible=True),
-        gr.Button.update(visible=True)
+        gr.Button.update(visible=True),
+        gr.TextArea.update(visible=True),
+        gr.TextArea.update(visible=True)
     )
 
 
@@ -125,18 +129,24 @@ def run_related():
     global related_chosen
     attn_list = []
     clr_bts = []
+    correct_drg_list = []
+    correct_salient_list = []
     for related in related_chosen:
         text = related[0]
         attn_html = run(text=text, related_discharges=True)
         attn_list.append(gr.HTML.update(value=attn_html))
         clr_bts.append(gr.ClearButton.update(visible=True))
+        correct_drg_list.append(gr.Textbox.update(visible=True))
+        correct_salient_list.append(gr.Textbox.update(visible=True))
     if len(attn_list) != 3:
         # find difference
         diff = 3 - len(attn_list)
         for i in range(diff):
             attn_list.append(gr.HTML.update(value=''))
             clr_bts.append(gr.ClearButton.update(visible=False))
-    return attn_list + clr_bts
+            correct_drg_list.append(gr.Textbox.update(visible=False))
+            correct_salient_list.append(gr.Textbox.update(visible=False))
+    return attn_list + clr_bts + correct_drg_list + correct_salient_list
         
 
 
@@ -155,6 +165,9 @@ def load_df_example(df, event: gr.SelectData):
     related_chosen.append([discharge_summary])
     return prettify_text(related_chosen)
 
+def save_results(text):
+    return
+
 def prettify_text(nested_list):
     string = ''
     for li in nested_list:
@@ -171,7 +184,7 @@ def remove_most_recent():
     return prettify_text(related_chosen)
 
 def clr_btn():
-    return gr.ClearButton.update(visible=False)
+    return gr.ClearButton.update(visible=False), gr.Textbox.update(visible=False), gr.Textbox.update(visible=False)
 
 def main():
     with gr.Blocks() as demo:
@@ -191,35 +204,67 @@ def main():
         with gr.Row() as row:
             attn_viz = gr.HTML() 
         with gr.Row() as row:
-            attn_clr_btn = gr.ClearButton(value='Remove output', visible=False, components=[attn_viz]) 
-            attn_clr_btn.click(clr_btn, outputs=[attn_clr_btn])                    
+            with gr.Column() as col:
+                correct_drg_text = gr.Textbox(visible=False, label="Input Correct DRG", interactive=True)
+                correct_drg_text.submit(save_results, inputs=correct_drg_text)            
+            with gr.Column() as col:
+                salient_words_box = gr.Textbox(visible=False, label="Input Salient Words (comma separated)", interactive=True)
+                salient_words_box.submit(save_results, inputs=salient_words_box)
+            attn_clr_btn = gr.ClearButton(value='Remove DRG Results', visible=False, components=[attn_viz]) 
+            attn_clr_btn.click(clr_btn, outputs=[attn_clr_btn, correct_drg_text, salient_words_box])        
         
         # related row 1
         with gr.Row() as row:
+            attn_1 = gr.HTML()
+            related_attn.append(attn_1)
+        with gr.Row() as row:
+            with gr.Column() as col:
+                correct_drg_text_1 = gr.Textbox(visible=False, label="Input Correct DRG", interactive=True)
+                correct_drg_text_1.submit(save_results, inputs=correct_drg_text_1)
+                correct_drg_text_list.append(correct_drg_text_1)            
             with gr.Column() as col:
-                attn = gr.HTML()
-                related_attn.append(attn)
-                attn_clr = gr.ClearButton(value='Remove output', visible=False, components=[attn])    
-                related_clr_bts.append(attn_clr)
-                attn_clr.click(clr_btn, outputs=[attn_clr]) 
+                salient_words_box_1 = gr.Textbox(visible=False, label="Input Salient Words (comma separated)", interactive=True)
+                salient_words_box_1.submit(save_results, inputs=salient_words_box_1)
+                correct_salient_words_list.append(salient_words_box_1)
+            attn_clr_1 = gr.ClearButton(value='Remove DRG Results', visible=False, components=[attn_1])    
+            related_clr_bts.append(attn_clr_1)
+            attn_clr_1.click(clr_btn, outputs=[attn_clr_1, correct_drg_text_1, salient_words_box_1]) 
 
         # related row 2
         with gr.Row() as row:
+            attn_2 = gr.HTML()
+            related_attn.append(attn_2)
+        with gr.Row() as row:
+            with gr.Column() as col:
+                correct_drg_text_2 = gr.Textbox(visible=False, label="Input Correct DRG", interactive=True)
+                correct_drg_text_2.submit(save_results, inputs=correct_drg_text_2)
+                correct_drg_text_list.append(correct_drg_text_2)            
             with gr.Column() as col:
-                attn = gr.HTML()
-                related_attn.append(attn)
-                attn_clr = gr.ClearButton(value='Remove output', visible=False, components=[attn])    
-                related_clr_bts.append(attn_clr)
-                attn_clr.click(clr_btn, outputs=[attn_clr]) 
+                salient_words_box_2 = gr.Textbox(visible=False, label="Input Salient Words (comma separated)", interactive=True)
+                salient_words_box_2.submit(save_results, inputs=salient_words_box_2)
+                correct_salient_words_list.append(salient_words_box_2)
+            attn_clr_2 = gr.ClearButton(value='Remove DRG Results', visible=False, components=[attn_2])    
+            related_clr_bts.append(attn_clr_2)
+            attn_clr_2.click(clr_btn, outputs=[attn_clr_2, correct_drg_text_2, salient_words_box_2]) 
+
 
         # related row 3
         with gr.Row() as row:
+            attn_3 = gr.HTML()
+            related_attn.append(attn_3)
+        with gr.Row() as row:
+            with gr.Column() as col:
+                correct_drg_text_3 = gr.Textbox(visible=False, label="Input Correct DRG", interactive=True)
+                correct_drg_text_3.submit(save_results, inputs=correct_drg_text_3)
+                correct_drg_text_list.append(correct_drg_text_3)            
             with gr.Column() as col:
-                attn = gr.HTML()
-                related_attn.append(attn)
-                attn_clr = gr.ClearButton(value='Remove output', visible=False, components=[attn])    
-                related_clr_bts.append(attn_clr)
-                attn_clr.click(clr_btn, outputs=[attn_clr]) 
+                salient_words_box_3 = gr.Textbox(visible=False, label="Input Salient Words (comma separated)", interactive=True)
+                salient_words_box_3.submit(save_results, inputs=salient_words_box_3)
+                correct_salient_words_list.append(salient_words_box_3)
+            attn_clr_3 = gr.ClearButton(value='Remove DRG Results', visible=False, components=[attn_3])    
+            related_clr_bts.append(attn_clr_3)
+            attn_clr_3.click(clr_btn, outputs=[attn_clr_3, correct_drg_text_3, salient_words_box_3]) 
+
 
         # input to related summaries
         with gr.Row() as row:
@@ -229,13 +274,15 @@ def main():
             sbm_btn = gr.Button(value="Submit Related Summaries", components=[input_related], visible=False)                              
         
         with gr.Row() as row:
-            # related = gr.Dataset(samples=[], components=[input_related], visible=False, type='index', headers=['AAAAA', 'BBBB', 'CCCCC', 'DDDDD', 'RRRRR'])
             related = gr.DataFrame(
                 value=None, headers=['Similarity Score', 'Related Discharge Summary'], max_rows=5,
                 datatype=['number', 'str'], col_count=(2, 'fixed'), visible=False
             )
         # initial run
-        btn.click(run, inputs=[input], outputs=[attn_viz, related, attn_clr_btn, input_related, sbm_btn, rmv_related_btn])
+        btn.click(run, inputs=[input], outputs=[
+            attn_viz, related, attn_clr_btn, input_related,
+            sbm_btn, rmv_related_btn, correct_drg_text, salient_words_box
+        ])
         # find related summaries
         # related.click(load_example, inputs=[related], outputs=[input_related])
         related.select(load_df_example, inputs=[related], outputs=[input_related])
@@ -243,7 +290,7 @@ def main():
         rmv_related_btn.click(remove_most_recent, outputs=[input_related])
 
         # perform attribution on related summaries
-        sbm_btn.click(run_related, outputs=related_attn + related_clr_bts)
+        sbm_btn.click(run_related, outputs=related_attn + related_clr_bts + correct_drg_text_list + correct_salient_words_list)
 
         
     demo.launch()