Update app.py

app.py

@@ -135,11 +135,16 @@ def main():
         
         response = resp['data']['responses'][0]['content']
         context = resp['data']['logs']['content']['retrival_model']['matched_doc']
+        context_prob = resp['data']['logs']['content']['retrival_model']['matched_prob']
 
         if verbose:
             clen = len(context)
             retrived = resp['data']['logs']['content']['retrival_model']['retrived_doc'][:retv_cnt]
+            retrived_prob = resp['data']['logs']['content']['retrival_model']['retrived_prob'][:retv_cnt]
+            retrived = [str(b) + ':  ' + a for a, b in zip (retrived, retrived_prob)]
             retrived = '\n\n===============================\n\n'.join(retrived)
+
+            context = [str(b) + ':  ' + a for a, b in zip (context, context_prob)]
             context = '\n\n===============================\n\n'.join(context)
             response = f'###### Config: Context Checker Value: {ctx_checker_tmp}, LM Value: {lm_tmp}\n\n##### Retrived Context: {retrived}\n\n##### Matched Context: {clen}\n{context}\n\n##### Response:\n{response}'