EnigmaOfTheWorld commited on
Commit
f991c0b
·
1 Parent(s): 128592a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -16,8 +16,12 @@ docsearch = FAISS.load_local(folder_path = 'chanakyaNeeti',embeddings=embeddings
16
  chain = load_qa_chain(OpenAI(openai_api_key=os.environ['OPENAI_API_KEY']), chain_type="stuff")
17
 
18
  def learn_from_chanakya_neeti(chat_history,query):
19
- docs = docsearch.similarity_search(query)
20
- result = result if "i don’t know" not in (result:=chain.run(input_documents=docs, question=query)).lower().strip() else "Please only ask questions from chanakya neeti."
 
 
 
 
21
 
22
  return chat_history + [(query,result)]
23
 
 
16
  chain = load_qa_chain(OpenAI(openai_api_key=os.environ['OPENAI_API_KEY']), chain_type="stuff")
17
 
18
  def learn_from_chanakya_neeti(chat_history,query):
19
+ greetings = ['hi', 'hello', 'hey', 'good morning', 'good afternoon', 'good evening', 'greetings', 'salutations', 'yo', 'howdy', 'hola', 'bonjour', 'konnichiwa', 'ni hao', 'ciao', 'salaam', 'shalom', 'namaste']
20
+ if query.lower() in greetings:
21
+ result = f"{query}."
22
+ else:
23
+ docs = docsearch.similarity_search(query)
24
+ result = result if "i don’t know" not in (result:=chain.run(input_documents=docs, question=query)).lower().strip() else "Please only ask questions from chanakya neeti."
25
 
26
  return chat_history + [(query,result)]
27