Update app.py

app.py

@@ -65,9 +65,11 @@ def cstr(s, color='black'):
     return "<text style=color:{}>{}</text>".format(color, s)
 def cstr_bold(s, color='black'):
     return "<text style=color:{}><b>{}</b></text>".format(color, s)
+def cstr_break(s, color='black'):
+    return "<text style=color:{}>{}<br></text>".format(color, s)
 
-def print_colored(text, start_idx, end_idx):
-    a = cstr(' '.join([text[:start_idx], cstr_bold(text[start_idx:end_idx], color='red'), text[end_idx:]]), color='black')
+def print_colored(text, start_idx, end_idx, confidence):
+    a = cstr_break(' '.join(cstr(' '.join([text[:start_idx], cstr_bold(text[start_idx:end_idx], color='red'), text[end_idx:]]), color='black'), 'Confidence: {}'.format(confidence), color='black)
     #a = colored(text[:start_idx]) + colored(text[start_idx:end_idx], 'red', 'on_yellow') + colored(text[end_idx:])
     return a
        
@@ -85,7 +87,8 @@ def final_qa_pipeline(file, query):
     if len(lvl1) > 0:
         fnl_rank = qa_ranker(query, [l["docs"] for l in lvl1], top_k_ranker)
         #return (fnl_rank[0]["answer"], str(np.round(100*fnl_rank[0]["score"],2))+"%" , fnl_rank[0]['doc'])
-        return (print_colored(fnl_rank[0]['doc'], fnl_rank[0]['start'], fnl_rank[0]['end']), str(np.round(100*fnl_rank[0]["score"],2))+"%" )
+        #return (print_colored(fnl_rank[0]['doc'], fnl_rank[0]['start'], fnl_rank[0]['end']), str(np.round(100*fnl_rank[0]["score"],2))+"%"
+        return (print_colored(fnl_rank[0]['doc'], fnl_rank[0]['start'], fnl_rank[0]['end'], str(np.round(100*fnl_rank[0]["score"],2))+"%")  )
         #for fnl_ in fnl_rank:
         #    print("\n")
         #    print_colored(fnl_['doc'], fnl_['start'], fnl_['end'])