Update app.py

app.py

@@ -123,7 +123,7 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
     embeddings = OpenAIEmbeddings()
         
     
-    history = state['question']
+    history = state['messages']
 
     if not prompt:
         return gr.update(value=''), [(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)], f"Total tokens used: {state['total_tokens']}", state
@@ -161,7 +161,7 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
         
         docsearch = FAISS.from_texts(texts, embeddings)
         #query = str(system_prompt + history[-context_length*2:] +  [prompt_msg])
-        query = str(system_prompt + history[0] +  [prompt_msg])
+        query = str(system_prompt + history +  [prompt_msg])
         docs = docsearch.similarity_search(query)
         #print(docs[0].page_content)
 
@@ -171,9 +171,12 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
         
 # VectorDBQA.from_chain_type(llm=OpenAI(), chain_type="stuff", vectorstore=docsearch, return_source_documents=True)
 # https://colab.research.google.com/drive/1dzdNDZyofRB0f2KIB4gHXmIza7ehMX30?usp=sharing#scrollTo=b-ejDn_JfpWW
+
+        get_empty_state()
+        state.append(completion.copy())
         
-        history.append(prompt_msg.copy())
-        history.append(completion.copy())
+        #history.append(prompt_msg.copy())
+        #history.append(completion.copy())
         #history.append(completion.choices[0].message.to_dict())
         #history.append(completion["result"].choices[0].message.to_dict())
 
@@ -189,6 +192,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'])]
     #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'])]