fixed answer issue

app.py

@@ -271,19 +271,18 @@ ANSWER:"""
             st.error("OpenAI client not initialized. Please provide a valid API key.")
             return None, []
         
-        # Step 1: Retrieve relevant chunks
-        with st.spinner("🔍 Searching relevant documentation..."):
+        # Use a single spinner for the entire process to prevent flickering
+        with st.spinner("🤖 Generating answer..."):
+            # Step 1: Retrieve relevant chunks
             retrieved_chunks = self.retrieve_relevant_chunks(user_question, n_chunks)
-        
-        if not retrieved_chunks:
-            return "I couldn't find relevant information in the Scikit-learn documentation to answer your question. Please try rephrasing your question or ask about a different topic.", []
-        
-        # Step 2: Create augmented prompt
-        with st.spinner("📝 Preparing context..."):
+            
+            if not retrieved_chunks:
+                return "I couldn't find relevant information in the Scikit-learn documentation to answer your question. Please try rephrasing your question or ask about a different topic.", []
+            
+            # Step 2: Create augmented prompt
             rag_prompt = self.create_rag_prompt(user_question, retrieved_chunks)
-        
-        # Step 3: Generate answer
-        with st.spinner("🤖 Generating answer..."):
+            
+            # Step 3: Generate answer
             answer = self.generate_answer(rag_prompt, model)
         
         # Extract source URLs
@@ -394,8 +393,13 @@ def main():
     with col1:
         st.header("💬 Ask Your Question")
         
-        # Question input
-        default_question = st.session_state.get('selected_question', '')
+        # Question input with better state management
+        default_question = ''
+        if 'selected_question' in st.session_state:
+            default_question = st.session_state['selected_question']
+            # Clear after getting the value to prevent re-triggering
+            del st.session_state['selected_question']
+            
         user_question = st.text_input(
             "Enter your question about Scikit-learn:",
             value=default_question,
@@ -403,10 +407,6 @@ def main():
             key="question_input"
         )
         
-        # Clear selected question after using it
-        if 'selected_question' in st.session_state:
-            del st.session_state['selected_question']
-        
         # Submit button
         submit_button = st.button("🚀 Get Answer", type="primary")
         
@@ -428,8 +428,8 @@ def main():
                         'sources': sources
                     })
                     
-                    # Clear input
-                    st.rerun()
+                    # Success message instead of rerun to prevent flickering
+                    st.success("✅ Answer generated successfully! Check the chat history below.")
         
         # Display chat history
         if st.session_state.chat_history:
@@ -464,7 +464,7 @@ def main():
         
         for question in example_questions:
             if st.button(question, key=f"example_{hash(question)}"):
-                # Use a different approach to set the question
+                # Set the question and rerun only once
                 st.session_state['selected_question'] = question
                 st.rerun()
         

chroma_db/chroma.sqlite3

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7da052d90807d731532e1195c8fbd4478b133ff07654b8ed5798dfd2d30b7a90
+oid sha256:29dd849ea2bda952a956fc1ae9039fcc202dd536fba5f36beb87b2da0bb04c99
 size 13279232