Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 import langchain
 import chromadb
 import glob
-import logging 
+
 
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.vectorstores import Chroma
@@ -26,7 +26,7 @@ from langchain.chains.question_answering import load_qa_chain
 #    persist_directory="./embeddings" # Optional, defaults to .chromadb/ in the current directory
 #))
 
-logger=logging.getLogger() 
+
 
 def get_empty_state():
     return {"total_tokens": 0, "messages": []}
@@ -117,8 +117,7 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
         query = str(system_prompt + history[-context_length*2:] +  [prompt_msg])
         #completion = completion({"query": query})
         completion = completion.run(query)
-        logger.info("test") 
-        logger.info(completion) 
+
 #        completion = completion({"question": query, "chat_history": history[-context_length*2:]})
 
 
@@ -126,8 +125,8 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
 # https://colab.research.google.com/drive/1dzdNDZyofRB0f2KIB4gHXmIza7ehMX30?usp=sharing#scrollTo=b-ejDn_JfpWW
         
         history.append(prompt_msg)
-        #history.append(completion.choices[0].message.to_dict())
-        history.append(completion["result"].choices[0].message.to_dict())
+        history.append(completion.choices[0].message.to_dict())
+        #history.append(completion["result"].choices[0].message.to_dict())
 
         state['total_tokens'] += completion['usage']['total_tokens']