Few enhancements

app.py

@@ -19,6 +19,18 @@ st.markdown("""
         background-color: #4CAF50; /* Green on hover */
         color: white;
     }
+    .new-chat-button > button {
+        background-color: #FF0000; /* Red */
+        color: white;
+        border-radius: 5px;
+        border: none;
+        padding: 10px 20px;
+        font-size: 16px;
+    }
+    .new-chat-button > button:hover {
+        background-color: #CC0000; /* Darker red on hover */
+        color: white;
+    }
     .umc-header {
         font-size: 24px; /* Adjusted font size for sidebar */
         font-weight: bold;
@@ -42,10 +54,22 @@ if "query_history" not in st.session_state:
     st.session_state.query_history = []  # Store queries and responses
 if "retrieved_chunks" not in st.session_state:
     st.session_state.retrieved_chunks = []  # Store retrieved chunks
+if "new_query_input" not in st.session_state:
+    st.session_state.new_query_input = ""  # Store new query input
 
 # Main app title (shown on all screens)
-st.title("United Methodist Church Discipline Assistant")
-st.write("Ask questions about The Book of Discipline 2020-2024!")
+st.title("United Methodist Church Assistant")
+
+# Header with description and New Chat button
+col1, col2 = st.columns([3, 1])
+with col1:
+    st.write("Ask questions about The Book of Discipline 2020-2024!")
+with col2:
+    if st.button("New Chat", key="new_chat"):
+        st.session_state.query_history = []
+        st.session_state.retrieved_chunks = []
+        st.session_state.new_query_input = ""
+        st.rerun()
 
 # Password form (shown only if not authenticated)
 if not st.session_state.authenticated:
@@ -66,18 +90,21 @@ if st.session_state.authenticated:
         "Can you tell me the history of United Methodist Church",
         "What does the Constitution say about inclusiveness?",
         "Tell me about John Wesley."
+        " What are special provisions about Sunday prayers"
     ]
     query = st.selectbox("Pick a question or type your own:", [""] + sample_questions, key="query_select") or st.text_input("Your question:", key="query_input")
 
     # Process query and display response
     if st.button("Submit"):
-        with st.spinner("Fetching response..."):
+        with st.spinner("UMC AI Bot is thinking..."):
             # Retrieve chunks and response
             results = vector_store.similarity_search(query, k=5)  # Assuming vector_store is accessible
             st.session_state.retrieved_chunks = results  # Store retrieved chunks
             response = query_rag(query)
             st.session_state.query_history.append((query, response))  # Store query and response
-
+            st.session_state.new_query_input = ""  # Clear the new query input
+            st.rerun()  # Refresh to show response
+            
     # Display query history
     for i, (q, r) in enumerate(st.session_state.query_history):
         st.markdown(f"**Query {i+1}: {q}**")
@@ -101,18 +128,19 @@ if st.session_state.authenticated:
 
     # New prompt window with sample questions
     if st.session_state.query_history:  # Show only after at least one query
-        st.markdown("### Ask Another Question")
+        st.markdown("### Explore More Insights")
         selected_query = st.selectbox("Pick another question or type your own:", [""] + sample_questions, key="new_query_select")
-        custom_query = st.text_input("Your new question:", key="new_query_input")
+        custom_query = st.text_input("What else would you like to know?", value=st.session_state.new_query_input, key=f"new_query_input_{len(st.session_state.query_history)}")
         # Use custom query if provided, otherwise use selected query
         new_query = custom_query if custom_query else selected_query
         if st.button("Submit New Query"):
             if new_query:  # Ensure a query is provided
                 with st.spinner("UMC AI Bot is thinking..."):
-                    results = vector_store.similarity_search(new_query, k=5)
+                    results = vector_store.similarity_search(new_query, k=8)
                     st.session_state.retrieved_chunks = results
                     response = query_rag(new_query)
                     st.session_state.query_history.append((new_query, response))
+                    st.session_state.new_query_input = ""  # Clear the input for the next query
                     st.rerun()  # Refresh to show new response
             else:
                 st.warning("Please select a question or enter a new one.")