Update app.py

app.py

@@ -29,7 +29,7 @@ prompt_templates = {"All Needs Experts": "Respond as if you are combination of a
 actor_description = {"All Needs Experts": "<div style='float: left;margin: 0px 5px 0px 5px;'><img src='https://na.weshareresearch.com/wp-content/uploads/2023/04/experts2.jpg' alt='needs expert image' style='width:70px;align:top;'></div>A combiation of all needs assessment experts."}
 
 def get_empty_state():
-    return {"total_tokens": 0, "messages": []}
+    return { "messages": []}
 
 
 def download_prompt_templates():
@@ -74,7 +74,7 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
     history = state['messages']
 
     if not prompt:
-        return gr.update(value=''), [(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)], f"Total tokens used: {state['total_tokens']}", state
+        return gr.update(value=''), [(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)], state
     
     prompt_template = prompt_templates[prompt_template]
 
@@ -105,7 +105,9 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
     #state.append(completion.copy())
     
     completion = { "content": completion }
-    state['total_tokens'] += completion['usage']['total_tokens']
+    
+    
+    #state['total_tokens'] += completion['usage']['total_tokens']
 
 
     
@@ -117,10 +119,10 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
     #    }
     #    history.append(error.copy())
 
-    total_tokens_used_msg = f"Total tokens used: {state['total_tokens']}"
+    #total_tokens_used_msg = f"Total tokens used: {state['total_tokens']}"
 
     chat_messages = [(prompt_msg['content'], completion['content'])]    
-    return '', chat_messages, total_tokens_used_msg, state    
+    return '', chat_messages,  state    # total_tokens_used_msg,
 
 
 def clear_conversation():
@@ -158,7 +160,7 @@ with gr.Blocks(css=css) as demo:
                 input_message = gr.Textbox(show_label=False, placeholder="Enter your needs assessment question", visible=True).style(container=False)
 
                 btn_submit = gr.Button("Submit")
-                total_tokens_str = gr.Markdown(elem_id="total_tokens_str")
+                #total_tokens_str = gr.Markdown(elem_id="total_tokens_str")
                 btn_clear_conversation = gr.Button("Start New Conversation")
             with gr.Column():
                 prompt_template = gr.Dropdown(label="Choose an Expert:", choices=list(prompt_templates.keys()))
@@ -169,9 +171,9 @@ with gr.Blocks(css=css) as demo:
                     context_length = gr.Slider(minimum=1, maximum=5, value=2, step=1, label="Context Length", info="Number of previous questions you have asked.")
 
    
-    btn_submit.click(submit_message, [ input_message, prompt_template, temperature, max_tokens, context_length, state], [input_message, chatbot, total_tokens_str, state])
-    input_message.submit(submit_message, [ input_message, prompt_template, temperature, max_tokens, context_length, state], [input_message, chatbot, total_tokens_str, state])
-    btn_clear_conversation.click(clear_conversation, [], [input_message, chatbot, total_tokens_str, state])
+    btn_submit.click(submit_message, [ input_message, prompt_template, temperature, max_tokens, context_length, state], [input_message, chatbot,  state])
+    input_message.submit(submit_message, [ input_message, prompt_template, temperature, max_tokens, context_length, state], [input_message, chatbot,  state])
+    btn_clear_conversation.click(clear_conversation, [], [input_message, chatbot,  state])
     prompt_template.change(on_prompt_template_change_description, inputs=[prompt_template], outputs=[prompt_template_preview])