Update app.py

app.py

@@ -104,6 +104,8 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
         state.append(completion.copy())
 
         state['total_tokens'] += completion['usage']['total_tokens']
+
+        return '', chat_messages, total_tokens_used_msg, state
     
     except Exception as e:
         history.append(prompt_msg.copy())
@@ -119,7 +121,7 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
     
     
     
-    return '', chat_messages, total_tokens_used_msg, state
+