Update app.py

app.py

@@ -19,6 +19,8 @@ if 'init_time' not in st.session_state:
     st.session_state.init_time = None
 if 'form_key' not in st.session_state:
     st.session_state.form_key = 0  # This will help us reset the form
+if 'processing' not in st.session_state:
+    st.session_state.processing = False
 
 # THEN: Import your modules
 from rag_engine import process_query, load_model
@@ -33,7 +35,6 @@ st.markdown("""
     text-align: center;
     margin-bottom: 1rem;
 }
-
 /* Button styling */
 .stButton>button {
     background-color: #fff0f0 !important;
@@ -46,7 +47,6 @@ st.markdown("""
     overflow: hidden !important;
     text-overflow: ellipsis !important;
 }
-
 /* Form submit button specific styling */
 button[type="submit"], 
 .stFormSubmit>button, 
@@ -60,7 +60,6 @@ button[type="submit"],
     background-color: #fafbff !important;
     border-color: #c5cae9 !important;
 }
-
 /* Input field styling */
 div[data-baseweb="input"] {
     border: 1px solid #fff0f0 !important;
@@ -73,7 +72,6 @@ div[data-baseweb="input"]:focus-within {
 div[data-baseweb="input"]:active {
     border: 1px solid #fff0f0 !important;
 }
-
 /* Style the st.info boxes */
 div.stInfo {
     background-color: #f8faff !important;
@@ -81,7 +79,6 @@ div.stInfo {
     border: 1px solid #e1e4f2 !important;
     border-radius: 8px !important;
 }
-
 /* COMBINED SCROLL CONTAINER */
 .questions-scroll-container {
     width: 100%;
@@ -89,12 +86,10 @@ div.stInfo {
     scrollbar-width: none; /* Firefox */
     -ms-overflow-style: none; /* IE and Edge */
 }
-
 /* Hide scrollbar for Chrome, Safari and Opera */
 .questions-scroll-container::-webkit-scrollbar {
     display: none;
 }
-
 /* Inner content that holds both rows */
 .questions-content {
     display: inline-flex;
@@ -103,20 +98,24 @@ div.stInfo {
     gap: 10px;
     padding: 5px 0;
 }
-
 /* Individual rows */
 .questions-row {
     display: flex;
     flex-direction: row;
     gap: 10px;
 }
-
 /* Placeholder for buttons */
 .button-placeholder {
     min-height: 38px;
     min-width: 120px;
     margin: 0 5px;
 }
+/* Hide "Press Enter to submit form" text */
+.stTextInput label + div + div small,
+[data-testid="stForm"] small,
+form small {
+    display: none !important;
+}
 </style>
 <div class="main-title">Spirituality Q&A</div>
 """, unsafe_allow_html=True)
@@ -229,20 +228,35 @@ with col1:
 with col2:
     word_limit = st.slider("Word limit:", 50, 500, 200)
 
-# Only process the query if explicitly submitted
-if st.session_state.submit_clicked and st.session_state.last_query:
+# Only process the query if explicitly submitted and not already processing
+if st.session_state.submit_clicked and st.session_state.last_query and not st.session_state.processing:
     st.session_state.submit_clicked = False
+    st.session_state.processing = True
 
     with st.spinner("Processing your question..."):
         try:
+            progress_bar = st.progress(0)
+            progress_bar.progress(25, "Generating embeddings...")
+            
+            # Call processing function with progress tracking
+            progress_bar.progress(50, "Retrieving relevant passages...")
+            
             result = process_query(st.session_state.last_query, top_k=top_k, word_limit=word_limit)
+            
+            progress_bar.progress(75, "Generating answer...")
+            time.sleep(0.5)  # Small delay for visual feedback
+            progress_bar.progress(100)
+            
             st.subheader("Answer:")
             st.write(result["answer_with_rag"])
             st.subheader("Sources:")
             for citation in result["citations"].split("\n"):
                 st.write(citation)
+                
         except Exception as e:
             st.error(f"Error processing query: {str(e)}")
+        finally:
+            st.session_state.processing = False
 
 # Add helpful information
 st.markdown("---")