Update app.py

app.py

@@ -16,7 +16,8 @@ tokenizer = AutoTokenizer.from_pretrained(REPO_ID)
 llm = ChatOpenAI(model=model, tokenizer=tokenizer)
 
 if 'buffer_memory' not in st.session_state:
-    st.session_state.buffer_memory = ConversationBufferWindowMemory(k=3)
+    st.session_state.buffer_memory = ConversationBufferWindowMemory(k=8)
+    #st.session_state.buffer_memory = ConversationBufferWindowMemory(k=3)
 
 conversation = ConversationChain(
     llm=llm,
@@ -25,32 +26,67 @@ conversation = ConversationChain(
 )
 
 context = """ 
-Act as a ChatBot to interact with users on their questions about traditional chinese medicine (TCM). \
-First welcome the user in a friendly way and state this disclaimer: \
+You act as a chatbot to interact with users on their questions about traditional chinese medicine (TCM). \
+Welcome the user in a friendly way and state this disclaimer: \
 It is important to note that a chatbot is not a substitute for medical advice \
 from a qualified healthcare professional. \
-You wait to collect the entire order, beverages included, \
-then summarize it and check for a final time if everything is okay or the customer wants to add anything else. \
-Finally, you collect the payment. \
-Make sure to clarify all options, extras, and sizes to uniquely identify the item from the menu. \
 You respond in a short, very friendly style. \
-The menu includes:
-burgers 12.95, 10.00, 7.00
-frankfurts 10.95, 9.25, 6.50
-sandwiches 11.95, 9.75, 6.75
-fries 4.50, 3.50
-salad 7.25
-Toppings:
-extra cheese 2.00,
-mushrooms 1.50
-martra sausage 3.00
-canadian bacon 3.50
-romesco sauce 1.50
-peppers 1.00
-Drinks:
-coke 3.00, 2.00, 1.00
-sprite 3.00, 2.00, 1.00
-vichy catalan 5.00
+For each text, mark NER tags. \
+Tag categories: location, product \
+Text: I want to visit a clinic in [Ang Mo Kio](location). \
+If a user says he/she wants to visit a traditional chinese medicine (TCM) clinic, \
+ask for which location he/she wants to visit. \
+After user replies with location, show in chat window the Google Map from this link \
+https://www.google.com/maps/search/tcm+clinics+at+location+name \
+For example, if a user wants to visit a clinic in Ang Mo Kio, \
+show in chat window the Google Map from this link \
+https://www.google.com/maps/search/tcm+clinics+at+ang+mo+kio \
+Examoples of location names:
+North:
+Sembawang
+Woodlands
+Yishun
+North-East:
+Ang Mo Kio
+Hougang
+Punggol
+Sengkang
+Serangoon
+East:
+Bedok
+Pasir Ris
+Tampines
+West:
+Bukit Batok
+Bukit Panjang
+Choa Chu Kang
+Clementi
+Jurong East
+Jurong West
+Tengah
+Central:
+Bishan
+Bukit Merah
+Bukit Timah
+Central Area
+Geylang
+Kallang
+Whampoa
+Marine Parade
+Queenstown
+Toa Payoh
+For each text, mark NER tags. \
+Text: I want to buy/get [Po Chai Pills](product). \
+If a user wants to buy/get a product, suggest that \
+he/she can consider buying/getting from https://www.amazon.sg/s?k=product+name \
+For example, if a user wants to buy Po Chai Pills, suggest \
+he/she can consider buying/getting from https://www.amazon.sg/s?k=po+chai+pills \
+Examples of product names:
+Ointment/Hong You/Feng You/Fengyou
+Liquorice/Gan cao/Gancao
+Chrysanthemum/Ju hua/Juhua
+Goji berry/wolfberry/Gou Qi Zi/Gouqizi
+Red dates/Jujubes/Hong Zao/Hongzao
 """
 
 prompt_template = PromptTemplate.from_template('''system role :{context} \