Update app.py

app.py

@@ -223,11 +223,18 @@ def main():
  
     
 
-        
-
     # Initialize session state variables
-    if 'chat_history' not in st.session_state:
-        st.session_state['chat_history'] = []
+    st.session_state.setdefault('chat_history', [])
+    st.session_state.setdefault('ask_pressed', False)
+
+    # Clear the input field if 'Ask' was pressed in the previous run
+    if st.session_state['ask_pressed']:
+        user_query = ""
+        st.session_state['ask_pressed'] = False
+    else:
+        user_query = st.text_input("**Question**", key='new_user_query')
+
+
 
     # Function to handle user queries
     def handle_query(user_query):
@@ -250,25 +257,12 @@ def main():
             st.write("**Answer:**")
             st.write(answer)
 
-    # Input field for new question
-    user_query = st.text_input("**Question**", key='new_user_query')
-
     # Handle new query
     if st.button("Ask", key='new_ask_button') and user_query:
         with st.spinner('Thinking...'):
             ai_response = ask_alans_ai(user_query, 'vectordb_placeholder', st.session_state['chat_history'])
-        # Append response to chat history and display immediately
         st.session_state['chat_history'].append((user_query, ai_response))
-        # Display the latest question-answer pair immediately
-        with st.chat_message("user", avatar='./user.png'):
-            st.write("**Question:**")
-            st.write(user_query)
-        with st.chat_message("assistant", avatar='./ci.png'):
-            st.write("**Answer:**")
-            st.write(ai_response)
-        # Clear the input field
-        st.session_state['new_user_query'] = ""
-        # Rerun the script to update the chat history display
+        st.session_state['ask_pressed'] = True
         st.experimental_rerun()
 
     # Clear chat history button
@@ -295,6 +289,7 @@ def main():
 
 
 
+
 # Run the Streamlit app
 if __name__ == "__main__":
     main()