Update app.py

app.py

@@ -321,7 +321,6 @@ def main():
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
     
-    user_input = st.chat_input("Ask me anything")
     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"]
@@ -335,16 +334,23 @@ def main():
     
     s1, s2, s3 = st.columns([1, 1, 1])
     
+    user_input = None
+    
     with s1:
         if st.button(choice1, use_container_width=True):
             user_input = choice1
-        
+    
     with s2:
-         if st.button(choice2, use_container_width=True):
+        if st.button(choice2, use_container_width=True):
             user_input = choice2
+    
     with s3:
         if st.button(choice3, use_container_width=True):
             user_input = choice3
+    
+    # Also allow user input through text box
+    if not user_input:
+        user_input = st.chat_input("Ask me anything")
         
     if user_input:
         with st.spinner('Generating Response...'):