.

app.py

@@ -122,6 +122,12 @@ def show_avg(base_fn, model_name, patent_claim_num, show_pick=False):
   lst = result['output']
   result = ''
   sum_all = {}
+  colors = [
+    ['00ff00', '000000', '1'], 
+    ['008800', 'ffffff', '2-10'], 
+    ['ff0000', 'ffffff', 'out of top 10'], 
+  ]
+
   for i, tk in enumerate(lst):
     token_text = handle_char_return(tk['actual_next_token_text'])
     if token_text == '<|end_of_claim|>': 
@@ -133,12 +139,6 @@ def show_avg(base_fn, model_name, patent_claim_num, show_pick=False):
     pick = int(tk['actual_next_token_top_seq'])
     prob = float(tk['actual_next_token_top_prob'])
 
-    colors = [
-      ['00ff00', '000000', '1'], 
-      ['008800', 'ffffff', '2-10'], 
-      ['ff0000', 'ffffff', 'out of top 10'], 
-    ]
-
     for i, item in enumerate(colors):
       sum_all[item[2]] = 0
 
@@ -159,21 +159,20 @@ def show_avg(base_fn, model_name, patent_claim_num, show_pick=False):
       tag = colors[2][2]
       sum_all[tag] += 1
 
-    if show_pick:
-      pick = '[%s]' % pick
-    else:
-      pick = ''
+    result += "<span style=background-color:#%s;color:#%s;border-radius:5px;>%s%s</span>  " % (bg_color, fg_color, token_text, pick)
 
-    result += "<span style=background-color:#%s;color:#%s;border-radius:5px;>%s%s</span>  " % (bg_color, fg_color, token_text, pick) #&nbsp;
+  st.text('debug 2')  
 
   color_msg = ''
   for i, v in enumerate(colors):
     color_msg += "<span style=background-color:#%s;color:#%s;border-radius:5px;>&nbsp;%s&nbsp;</span> " % (v[0], v[1], v[2]) 
 
+
   # sum_pick as top 1~10
   keys_with_auto = (sum_pick+sum_outscope_len)
   keys_without_auto = len(full_text)
   saved_ratio = float(keys_without_auto-keys_with_auto)/keys_without_auto * 100
+
   s = 'model: %s\n' \
     'Autocomplete Effectiveness: %.1f%% (keystrokes saved)\n' \
     'Total keystrokes: %s (with autocomplete), %s (without autocomplete)\n' \
@@ -248,8 +247,8 @@ def main():
     page_title="Patent-GPT-J demo",  # String or None. Strings get appended with "• Streamlit".
     page_icon=None,  # String, anything supported by st.image, or None.
   )
-  st.subheader("PatentGPT-J Demo 1 (Autocomplete Effectiveness)")
-  st.text("Data coverage: ipg22 (in 2022)")
+  st.subheader("PatentGPT-J Demo 3 (Autocomplete Effectiveness)")
+  st.text("Data coverage: unicorn text")
 
   num_set = set()
   fn_lst = glob.glob(os.path.join(folder, '*'))
@@ -283,28 +282,11 @@ def main():
       select_lst.append(num)
   select_lst.sort()
 
-  if len(select_lst) == 0:
-    st.text('select_lst is empty')
-    return 
-
-  if enable_summary_button:
-    if st.button('Show Summary'):
-      st.text('len(select_lst) = %s' % len(select_lst))
-      show_overall_summary(prefix_lst, select_lst)
-
   show_patent_lst = [ s.replace('_', ' (claim ') + ')' for s in select_lst]
-
-  #pick = 0
   pick = random.randrange(len(select_lst))
-  if st.button('Random pick'):
-    pick = random.randrange(len(select_lst))
-    num = select_lst[pick]
-    #num = random.choice(select_lst)
-
-  selected = st.selectbox("Choose a patent claim", show_patent_lst, index=pick)
-  num = selected.replace(')', '').replace(' (claim ', '_')
+  num = select_lst[pick]
 
-  #st.text('Selected: %s' % num)
+  st.text('debug')
 
   avgs = []
   for prefix in prefix_lst: