Update app.py

app.py

@@ -98,14 +98,14 @@ 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())
 
         state['total_tokens'] += completion['usage']['total_tokens']
 
-        return '', chat_messages, total_tokens_used_msg, state
+
     
     except Exception as e:
         history.append(prompt_msg.copy())
@@ -117,13 +117,8 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
 
     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    
 
 
 def clear_conversation():