Update app.py

app.py

@@ -27,6 +27,7 @@ def launch_bot():
         })
         st.session_state.cfg = cfg
         st.session_state.vq = VectaraQuery(cfg.api_key, cfg.customer_id, cfg.corpus_ids)
+        st.session_state.first_round = True
 
     cfg = st.session_state.cfg
     vq = st.session_state.vq
@@ -53,10 +54,6 @@ def launch_bot():
 
     if "messages" not in st.session_state.keys():
         st.session_state.messages = [{"role": "assistant", "content": "How may I help you?"}]
-
-    if "messages" not in st.session_state:
-        st.session_state.messages = [{"role": "assistant", "content": "How may I help you?"}]
-        st.session_state.first_round = True
     
     # Display chat messages
     for message in st.session_state.messages:
@@ -74,28 +71,6 @@ def launch_bot():
         if prompt := st.chat_input("Ask me anything..."):
             process_prompt(prompt)  # Process the user-provided prompt
             
-        
-    # Display chat messages
-#    for message in st.session_state.messages:
-#        with st.chat_message(message["role"]):
-#            st.write(message["content"])
-
-    # User-provided prompt
-#    if prompt := st.chat_input():
-#        st.session_state.messages.append({"role": "user", "content": prompt})
-#        with st.chat_message("user"):
-#            st.write(prompt)
-    
-    # Generate a new response if last message is not from assistant
-#    if st.session_state.messages[-1]["role"] != "assistant":
-#        with st.chat_message("assistant"):
-#            with st.spinner("Thinking..."):
-#                response = generate_response(prompt) 
-#                st.write(response) 
-#        message = {"role": "assistant", "content": response}
-#        st.session_state.messages.append(message)
-
-
 def process_prompt(prompt):
     # Your logic to process the prompt and generate response
     st.session_state.messages.append({"role": "user", "content": prompt})