Update app.py

app.py

@@ -98,6 +98,7 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
         chain = load_qa_chain(ChatOpenAI(temperature=temperature, max_tokens=max_tokens, model_name="gpt-3.5-turbo"), chain_type="stuff")
         completion = chain.run(input_documents=docs, question=query)
         completion = { "content": completion }
+        chat_messages = [(prompt_msg['content'], completion['content'])]
         
         get_empty_state()
         state.append(completion.copy())
@@ -116,7 +117,7 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
 
     
     
-    chat_messages = [(prompt_msg['content'], completion['content'])]
+    
     
     return '', chat_messages, total_tokens_used_msg, state