Update app.py

app.py

@@ -160,8 +160,8 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
         #    new_docsearch = pickle.load(f)
         
         docsearch = FAISS.from_texts(texts, embeddings)
-        #query = str(system_prompt + history[-context_length*2:] +  [prompt_msg])
-        query = str(system_prompt +   [prompt_msg])
+        query = str(system_prompt + history[-context_length*2:] +  [prompt_msg])
+    
         docs = docsearch.similarity_search(query)
         #print(docs[0].page_content)
 
@@ -189,9 +189,9 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
 
     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)]
+    chat_messages = [(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)]
     
-    chat_messages = [(history[-2]['content'], history[-1]['content'])]
+    #chat_messages = [(history[-2]['content'], history[-1]['content'])]
     
     return '', chat_messages, total_tokens_used_msg, state