Update app.py

app.py

@@ -5,7 +5,7 @@ import os
 import requests
 import time
 from langsmith import traceable
-
+import random
 
 ##### Begin Library Code
 from transformers import pipeline
@@ -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."
+    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,14 +325,20 @@ 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?"]
+    list2 = ["Who invented quantum mechanics?", "Who are the authors of attention is all you need", "Tell me about popular places to travel in France"]
+    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"]
+
+# Randomly select one from each list
+random_selection = [random.choice(list1), random.choice(list2), random.choice(list3)]
+
     s1, s2, s3 = st.columns([1, 2, 3])
     with s1:
-        user_input = suggestion_button("Tell me about teapotllm") or user_input
+        user_input = suggestion_button(random.choice(list1)) or user_input
     with s2:
-        user_input = suggestion_button("Who invented quantum mechanics?") or user_input
+        user_input = suggestion_button(random.choice(list2)) or user_input
     with s3:
-        user_input = suggestion_button("Extract the year Google was founded") or user_input
+        user_input = suggestion_button(random.choice(list3)) or user_input
     
     if user_input:
         with st.spinner('Generating Response...'):