Update app.py

add cot display into functions

app.py

@@ -38,10 +38,10 @@ import requests
 import os
 import gradio as gr
 from sentence_transformers import SentenceTransformer, CrossEncoder, util
-from torch import tensor as torch_tensor
-from datasets import load_dataset
+#from torch import tensor as torch_tensor
+#from datasets import load_dataset
 
-from greg_funcs import mrkl_rspnd
+from greg_funcs import mrkl_rspnd, get_cot
 
 
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")  
@@ -397,6 +397,7 @@ class ChatWrapper:
 
 
             response = mrkl_rspnd(inp)
+            cot = get_cot(response)
             output = response['output']
 
             """
@@ -463,7 +464,7 @@ class ChatWrapper:
             raise e
         finally:
             self.lock.release()
-        return history, history, html_video, temp_file, html_audio, temp_aud_file, ""
+        return history, history, html_video, temp_file, html_audio, temp_aud_file, cot
         # return history, history, html_audio, temp_aud_file, ""
 
 
@@ -655,7 +656,8 @@ with gr.Blocks(css=CSS) as block:
 
             with gr.Column(scale=7):
                 chatbot = gr.Chatbot()
-
+        with gr.Accordion(label='Show AI chain of thought: ', open=False,):
+            ai_cot = gr.HTML(show_label=False)
         with gr.Row():
             message = gr.Textbox(label="What's on your mind??",
                                  placeholder=PLACEHOLDER,
@@ -852,7 +854,7 @@ with gr.Blocks(css=CSS) as block:
                                  anticipation_level_state, joy_level_state, trust_level_state, fear_level_state,
                                  surprise_level_state, sadness_level_state, disgust_level_state, anger_level_state,
                                  lang_level_state, translate_to_state, literary_style_state],
-                   outputs=[chatbot, history_state, video_html, my_file, audio_html, tmp_aud_file, message])
+                   outputs=[chatbot, history_state, video_html, my_file, audio_html, tmp_aud_file, ai_cot])
     # outputs=[chatbot, history_state, audio_html, tmp_aud_file, message])
 
     submit.click(chat, inputs=[message, history_state, chain_state, trace_chain_state,
@@ -861,8 +863,8 @@ with gr.Blocks(css=CSS) as block:
                                anticipation_level_state, joy_level_state, trust_level_state, fear_level_state,
                                surprise_level_state, sadness_level_state, disgust_level_state, anger_level_state,
                                lang_level_state, translate_to_state, literary_style_state],
-                 outputs=[chatbot, history_state, video_html, my_file, audio_html, tmp_aud_file, message])
-    # outputs=[chatbot, history_state, audio_html, tmp_aud_file, message])
+                 outputs=[chatbot, history_state, video_html, my_file, audio_html, tmp_aud_file, ai_cot])
+        # history, history, html_video, temp_file, html_audio, temp_aud_file, cot
 
 
 block.launch(debug=True)