Update app.py

app.py

@@ -109,7 +109,7 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
         history.append(prompt_msg)
         history.append(completion.choices[0].message.to_dict())
 
-        state['total_tokens'] += completion['usage']['total_tokens']
+        #state['total_tokens'] += completion['usage']['total_tokens']
     
     except Exception as e:
         history.append(prompt_msg)
@@ -118,10 +118,10 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
             "content": f"Error: {e}"
         })
 
-    total_tokens_used_msg = f"Total tokens used: {state['total_tokens']}"
-    chat_messages = str([(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)])
+    #total_tokens_used_msg = f"Total tokens used: {state['total_tokens']}"
+    chat_messages = [(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)]
 
-    return '', chat_messages, total_tokens_used_msg, state
+    return '', chat_messages #, total_tokens_used_msg, state
 
 def clear_conversation():
     return gr.update(value=None, visible=True), None, "", get_empty_state()