Update app.py

app.py

@@ -77,7 +77,8 @@ with st.sidebar:
     ctx_checker_tmp = 0.008
     lm_tmp = st.slider('Language Model Sensitivity', min_value=0.001, max_value=1.0, value=0.1, step=0.001)
     verbose = st.checkbox('Show Detailed Response', value=False)
-    
+    if verbose == True:
+        retv_cnt = st.slider('Display N retrived Doc', min_value=0, max_value=32, value=0, step=1)
 
     
       
@@ -128,12 +129,14 @@ def main():
             return
         
         response = resp['data']['responses'][0]['content']
+        retrived = resp['data']['logs']['content']['retrival_model']['retrived_docs'][:retv_cnt]
         context = resp['data']['logs']['content']['retrival_model']['matched_doc']
 
         if verbose:
             clen = len(context)
+            retrived = '\n\n===============================\n\n'.join(retrived)
             context = '\n\n===============================\n\n'.join(context)
-            response = f'###### Config: Context Checker Value: {ctx_checker_tmp}, LM Value: {lm_tmp}\n\n##### Matched Context: {clen}\n{context}\n\n##### Response:\n{response}'
+            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}'
 
         
         # Display assistant response in chat message container