Update app.py

app.py

@@ -302,13 +302,7 @@ def main():
     
     teapot_ai = TeapotAI(documents=[], settings=TeapotAISettings(rag_num_results=3, log_level="debug"))
 
-    
-    if "messages" not in st.session_state:
-        st.session_state.messages = [{"role": "assistant", "content": "Hi, I am Teapot AI, how can I help you?"}]
-    
-    for message in st.session_state.messages:
-        with st.chat_message(message["role"]):
-            st.markdown(message["content"])
+
     
     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","Summarize the book irobot", "Explain artificial intelligence","what are the key ingredients of bouillabaisse"]
@@ -321,40 +315,35 @@ def main():
     choice2 = random.choice(list2)
     choice3 = random.choice(list3)
     
-
-    # Initialize user_suggested_input in session_state if it doesn't exist
-    if 'user_suggested_input' not in st.session_state:
-        st.session_state.user_suggested_input = None
-    
-    # Define columns
     s1, s2, s3 = st.columns([1, 1, 1])
     
-    # Button click handling
+    user_suggested_input = None
+    
     with s1:
         if st.button(choice1, use_container_width=True):
-            st.session_state.user_suggested_input = choice1
+            user_suggested_input = choice1
     
     with s2:
         if st.button(choice2, use_container_width=True):
-            st.session_state.user_suggested_input = choice2
+            user_suggested_input = choice2
     
     with s3:
         if st.button(choice3, use_container_width=True):
-            st.session_state.user_suggested_input = choice3
+            user_suggested_input = choice3
+
+    if "messages" not in st.session_state:
+        st.session_state.messages = [{"role": "assistant", "content": "Hi, I am Teapot AI, how can I help you?"}]
     
-    # Input field for chat
-    user_input = st.chat_input("Ask me anything")
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
     
-    # Handle chat input and button selection
-    if user_input or st.session_state.user_suggested_input:
+    user_input = st.chat_input("Ask me anything")
+        
+    if user_input:
         with st.spinner('Generating Response...'):
-            response = handle_chat(user_prompt, st.session_state.user_suggested_input or user_input, teapot_ai)
-            st.write(response)
-    
-    # Reset the input field after generating the response if needed
-    if user_input or st.session_state.user_suggested_input:
-        st.session_state.user_suggested_input = None  # Reset suggested input after processing
-
+            response = handle_chat(user_prompt, user_suggested_input or user_input, teapot_ai)
+           
 
 if __name__ == "__main__":
     main()