New App.py with the retrieval documents

app.py

@@ -0,0 +1,121 @@
+import streamlit as st
+from rag_query import query_rag
+
+# Custom CSS for styling
+st.markdown("""
+    <style>
+    .main {background-color: #f5f5f5;}
+    .stTextInput > div > div > input {border: 2px solid #4CAF50; border-radius: 5px;}
+    .stSelectbox > div > div > select {border: 2px solid #4CAF50; border-radius: 5px;}
+    .stButton > button {
+        background-color: #333333; /* Dark grey */
+        color: white;
+        border-radius: 5px;
+        border: none;
+        padding: 10px 20px;
+        font-size: 16px;
+    }
+    .stButton > button:hover {
+        background-color: #4CAF50; /* Green on hover */
+        color: white;
+    }
+    .umc-header {
+        font-size: 24px; /* Adjusted font size for sidebar */
+        font-weight: bold;
+        margin-top: 10px; /* Space between logo and text */
+        margin-bottom: 20px; /* Space before Instructions */
+    }
+    .spacer {
+        margin-top: 30px; /* Space between text and Instructions */
+    }
+    .chunk-dropdown {
+        margin-top: 20px;
+        margin-bottom: 20px;
+    }
+    </style>
+""", unsafe_allow_html=True)
+
+# Initialize session state for password and query history
+if "authenticated" not in st.session_state:
+    st.session_state.authenticated = False
+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
+
+# 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!")
+
+# Password form (shown only if not authenticated)
+if not st.session_state.authenticated:
+    with st.form(key="password_form"):
+        password = st.text_input("Enter password:", type="password", key="password")
+        submit_password = st.form_submit_button("Login")
+        if submit_password:
+            if password == "umc2024":  # Password as set
+                st.session_state.authenticated = True
+                st.experimental_rerun()  # Refresh to show main app
+            else:
+                st.error("Incorrect password. Please try again.")
+
+# Main app content (shown only if authenticated)
+if st.session_state.authenticated:
+    # Interactive input with sample questions
+    sample_questions = [
+        "Can you tell me the history of United Methodist Church",
+        "What does the Constitution say about inclusiveness?",
+        "Tell me about John Wesley."
+    ]
+    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..."):
+            # 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
+
+    # Display query history
+    for i, (q, r) in enumerate(st.session_state.query_history):
+        st.markdown(f"**Query {i+1}: {q}**")
+        st.markdown(r)
+        # Display retrieved chunks in a dropdown
+        if st.session_state.retrieved_chunks:
+            chunk_options = [f"Chunk {j+1}: {chunk.page_content[:100]}..." for j, chunk in enumerate(st.session_state.retrieved_chunks)]
+            selected_chunk = st.selectbox("View Retrieved Chunks:", ["Select a chunk"] + chunk_options, key=f"chunk_select_{i}")
+            if selected_chunk != "Select a chunk":
+                chunk_index = int(selected_chunk.split(":")[0].split(" ")[1]) - 1
+                chunk = st.session_state.retrieved_chunks[chunk_index]
+                st.markdown("**Retrieved Chunk Details:**")
+                st.markdown(f"**Source:** {chunk.metadata['source']}")
+                if chunk.metadata["part"]:
+                    st.markdown(f"**Part:** {chunk.metadata['part']}")
+                st.markdown(f"**Heading:** {chunk.metadata['heading']}")
+                if chunk.metadata["paragraph_number"]:
+                    st.markdown(f"**Paragraph {chunk.metadata['paragraph_number']}:** {chunk.metadata['paragraph_title']}")
+                st.markdown(f"**Text:** {chunk.page_content}")
+        st.markdown("---")
+
+    # New prompt window with sample questions
+    if st.session_state.query_history:  # Show only after at least one query
+        st.markdown("### Ask Another Question")
+        new_query = st.selectbox("Pick another question or type your own:", [""] + sample_questions, key="new_query_select") or st.text_input("Your new question:", key="new_query_input")
+        if st.button("Submit New Query"):
+            with st.spinner("Fetching response..."):
+                results = vector_store.similarity_search(new_query, k=5)
+                st.session_state.retrieved_chunks = results
+                response = query_rag(new_query)
+                st.session_state.query_history.append((new_query, response))
+                st.experimental_rerun()  # Refresh to show new response
+
+# Sidebar with logo, text, and instructions (always shown)
+st.sidebar.image("UMC_Logo.png", width=200)  # Adjusted logo size for sidebar
+st.sidebar.markdown('<p class="umc-header">United Methodist Communications</p>', unsafe_allow_html=True)
+st.sidebar.markdown('<div class="spacer"></div>', unsafe_allow_html=True)  # Add space
+st.sidebar.header("Instructions")
+st.sidebar.write("Enter a question about The United Methodist Church's Book Of Discipline 2020-2024. The assistant will retrieve relevant sections and explain them in simple language, citing parts, paragraphs, and titles as needed.")
+st.sidebar.header("About")
+st.sidebar.write("Powered by United Methodist Communications.")