Update app.py

app.py

@@ -83,6 +83,7 @@ with st.sidebar:
     if verbose == True:
         retv_cnt = st.slider('Display N retrived Doc', min_value=0, max_value=32, value=0, step=1)
 
+    show_input = st.checkbox('Show Input of LLM', value=False)
     
       
 
@@ -134,6 +135,8 @@ def main():
             return
         
         response = resp['data']['responses'][0]['content']
+        reasoning = resp['data']['logs']['content']['llm']['reasoning']
+        llm_input = resp['data']['logs']['content']['llm']['llm_reasoning']
         context = resp['data']['logs']['content']['retrival_model']['matched_doc']
         context_prob = resp['data']['logs']['content']['retrival_model']['matched_prob']
 
@@ -146,8 +149,10 @@ def main():
 
             context = [str(round(b, 3)) + ':  ' + 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:\n{retrived}\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:\n{retrived}\n\n##### Response:\n{response}'    # ##### Matched Context:{clen}\n{context}\n\n
 
+        if show_input:
+            response += '\n\n### LLM Input:\n' + llm_input
         
         # Display assistant response in chat message container
         with st.chat_message("assistant", avatar=None):