EnigmaOfTheWorld commited on
Commit
80953fb
·
1 Parent(s): 74d6cf0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -12,8 +12,14 @@ docsearch = FAISS.load_local("embeddings",embeddings)
12
  chain = load_qa_chain(OpenAI(openai_api_key=os.environ['OPENAI_API_KEY']), chain_type="stuff")
13
 
14
  def ask_anything(chat_history,query):
15
- docs = docsearch.similarity_search(query)
16
- result = chain.run(input_documents=docs, question=query)
 
 
 
 
 
 
17
  return chat_history + [(query,result)]
18
 
19
 
 
12
  chain = load_qa_chain(OpenAI(openai_api_key=os.environ['OPENAI_API_KEY']), chain_type="stuff")
13
 
14
  def ask_anything(chat_history,query):
15
+ greetings = ['hi', 'hello', 'hey', 'good morning', 'good afternoon', 'good evening', 'greetings', 'salutations', 'yo', 'howdy', 'hola', 'bonjour', 'konnichiwa', 'ni hao', 'ciao', 'salaam', 'shalom', 'namaste']
16
+ if query.lower() in greetings:
17
+ result = f"Hello!How are you?"
18
+ elif query.lower() in ["what are you?","who are you?"]:
19
+ result = f'I am an AI chatbot available to assist you.'
20
+ else:
21
+ docs = docsearch.similarity_search(query)
22
+ result = chain.run(input_documents=docs, question=query)
23
  return chat_history + [(query,result)]
24
 
25