Add radio for ground truth

app.py

@@ -10,7 +10,7 @@ from models import create_nest_sentences, load_summary_model, summarizer_gen, lo
 from utils import plot_result, plot_dual_bar_chart, examples_load, example_long_text_load
 import json
 
-ex_text, ex_license, ex_labels = examples_load()
+ex_text, ex_license, ex_labels, ex_glabels = examples_load()
 ex_long_text = example_long_text_load()
 
 
@@ -28,8 +28,17 @@ with st.form(key='my_form'):
     if text_input == display_text:
         text_input = example_text
 
-    labels = st.text_input('Possible labels (comma-separated):',ex_labels, max_chars=1000)
+    labels = st.text_input('Enter possible labels (comma-separated):',ex_labels, max_chars=1000)
     labels = list(set([x.strip() for x in labels.strip().split(',') if len(x.strip()) > 0]))
+    
+    radio = st.radio(
+        "Are ground truth labels available?",
+        ('Yes','No'))
+
+    if radio == 'Yes':
+        glabels = st.text_input('Enter known labels (comma-separated):',ex_glabels, max_chars=1000)
+        glabels = list(set([x.strip() for x in glabels.strip().split(',') if len(x.strip()) > 0]))
+
     submit_button = st.form_submit_button(label='Submit')
 
 

examples.json

@@ -1,5 +1,6 @@
 {
 "text": "Such were the professor’s words—rather let me say such the words of the fate—enounced to destroy me. As he went on I felt as if my soul were grappling with a palpable enemy; one by one the various keys were touched which formed the mechanism of my being; chord after chord was sounded, and soon my mind was filled with one thought, one conception, one purpose. So much has been done, exclaimed the soul of Frankenstein—more, far more, will I achieve; treading in the steps already marked, I will pioneer a new way, explore unknown powers, and unfold to the world the deepest mysteries of creation.",
 "long_text_license": "[This eBook is for the use of anyone anywhere in the United States and most other parts of the world at no cost and with almost no restrictions whatsoever. You may copy it, give it away or re-use it under the terms of the Project Gutenberg License included with this eBook or online at www.gutenberg.org. If you are not located in the United States, you will have to check the laws of the country where you are located before using this eBook.]",
-"labels":"Batman,Science,Sound,Light,Creation,Optics,Eyes,Engineering,Color,Communication,Death"
+"labels":"Batman,Science,Sound,Light,Creation,Optics,Eyes,Engineering,Color,Communication,Death",
+"ground_labels":"Science,Sound,Light,Creation,,Engineering,Communication,Death"
 }

utils.py

@@ -77,7 +77,7 @@ def plot_dual_bar_chart(topics_summary, scores_summary, topics_text, scores_text
 def examples_load():
     with open("examples.json") as f:
         data=json.load(f)
-    return data['text'], data['long_text_license'], data['labels']
+    return data['text'], data['long_text_license'], data['labels'], data['ground_labels']
 
 def example_long_text_load():
     with open("example_long_text.txt", "r") as f: