Update app.py

app.py

@@ -294,7 +294,7 @@ def handle_chat(user_prompt, user_input, teapot_ai):
         st.sidebar.write("---")
 
     context = "\n".join(documents)
-    prompt = """You are Teapot, an open-source AI assistant optimized for low-end devices, providing short, accurate responses without hallucinating while excelling at information extraction and text summarization.If a user asks who you are reply "I am Teapot"."""
+    prompt = """You are Teapot, an open-source AI assistant optimized for low-end devices, providing short, accurate responses without hallucinating while excelling at information extraction and text summarization. If a user asks who you are reply "I am Teapot"."""
     response = query_teapot(prompt, context+user_prompt, user_input, teapot_ai)
 
     with st.chat_message("assistant"):
@@ -325,20 +325,23 @@ def main():
             st.markdown(message["content"])
     
     user_input = st.chat_input("Ask me anything")
-    list1 = ["Tell me about teapotllm", "What is Teapot AI?", "How does teapotllm work?"]
+    list1 = ["Tell me about teapotllm", "What is Teapot AI?","What devices can Teapot run on?","Who are you?"]
     list2 = ["Who invented quantum mechanics?", "Who are the authors of attention is all you need", "Tell me about popular places to travel in France","Tell me about Isaac Asimov", "Explain artificial intelligence"]
     list3 = ["Extract the year Google was founded", "Extract the last name of the father of artificial intelligence", "Output the capital of New York","Extarct the city where the louvre is located","Find the chemical symbol for gold"]
 
     # Randomly select one from each list
     random_selection = [random.choice(list1), random.choice(list2), random.choice(list3)]
 
+    choice1 = random.choice(list1)
+    choice2 = random.choice(list2)
+    choice3 = random.choice(list3)
     s1, s2, s3 = st.columns([1, 2, 3])
     with s1:
-        user_input = suggestion_button(random.choice(list1)) or user_input
+        user_input = suggestion_button(choice1) or user_input
     with s2:
-        user_input = suggestion_button(random.choice(list2)) or user_input
+        user_input = suggestion_button(choice2) or user_input
     with s3:
-        user_input = suggestion_button(random.choice(list3)) or user_input
+        user_input = suggestion_button(choice3) or user_input
     
     if user_input:
         with st.spinner('Generating Response...'):