Update app.py

app.py

@@ -371,6 +371,11 @@ class RAGPipeline:
 
     def process_query(self, query: str, placeholder) -> str:
         try:
+            # Verify this is the current query being processed
+            if hasattr(st.session_state, 'current_query') and query != st.session_state.current_query:
+                logging.warning(f"Skipping outdated query: {query}")
+                return ""
+                
             query = self.preprocess_query(query)
             status = placeholder.empty()
             status.write("🔍 Finding relevant information...")
@@ -385,7 +390,7 @@ class RAGPipeline:
                 cleaned_text = self.postprocess_response(doc)
                 if cleaned_text:
                     cleaned_docs.append(cleaned_text)
-        
+    
             status.write("💭 Generating response...")
             
             prompt = f"""Context information is below:
@@ -393,7 +398,7 @@ class RAGPipeline:
             
             Given the context above, please answer the following question:
             {query}
-        
+    
             Guidelines for your response:
             - Structure your response in clear, logical paragraphs
             - Start a new paragraph for each new main point or aspect
@@ -407,7 +412,7 @@ class RAGPipeline:
             - Exclude any dates, timestamps, or user comments
             - Focus on factual sports information
             - If you cannot answer based on the context, say so politely
-        
+    
             Format your response with proper paragraph breaks where appropriate.
             
             Answer:"""
@@ -421,7 +426,7 @@ class RAGPipeline:
                     final_response = self.postprocess_response(response_text)
                     
                     # Convert cleaned response to markdown with proper paragraph spacing
-                    markdown_response = final_response.replace('\n\n', '\n\n \n\n')  # Add visual spacing between paragraphs
+                    markdown_response = final_response.replace('\n\n', '\n\n \n\n')
                     
                     response_placeholder.markdown(markdown_response)
                     return final_response
@@ -440,7 +445,7 @@ class RAGPipeline:
             logging.error(f"Process error: {str(e)}")
             message = "Something went wrong. Please try again with a different question."
             placeholder.warning(message)
-            return messag
+            return message
 
     def query_model(self, prompt: str) -> str:
         """Query the local Llama model"""
@@ -790,6 +795,12 @@ def main():
         if not check_environment():
             return
 
+        # Initialize session state variables
+        if 'current_query' not in st.session_state:
+            st.session_state.current_query = None
+        if 'processing' not in st.session_state:
+            st.session_state.processing = False
+
         # Improved CSS styling
         st.markdown("""
             <style>
@@ -868,18 +879,12 @@ def main():
                 Hey there! 👋 I can help you with information on Ice Hockey, Baseball, American Football, Soccer, and Basketball. 
                 With access to the ESPN API, I'm up to date with the latest details for these sports up until October 2024.
             </p>
-        """, unsafe_allow_html=True)
-        
-        st.markdown("""
             <p class='description'>
                 Got any general questions? Feel free to ask—I'll do my best to provide answers based on the information I've been trained on!
             </p>
         """, unsafe_allow_html=True)
 
-        # Add spacing
-        st.markdown("<br>", unsafe_allow_html=True)
-
-        # Initialize the pipeline with better error handling
+        # Initialize the pipeline
         if 'rag' not in st.session_state:
             try:
                 with st.spinner("Loading resources..."):
@@ -904,8 +909,18 @@ def main():
             # Centered button with unique key
             if st.button("Get Answer", key="answer_button"):
                 if query:
+                    # Clear any previous response
+                    if 'response_placeholder' in st.session_state:
+                        st.session_state.response_placeholder.empty()
+                        
                     response_placeholder = st.empty()
+                    st.session_state.response_placeholder = response_placeholder
+                    
                     try:
+                        # Update current query and processing state
+                        st.session_state.current_query = query
+                        st.session_state.processing = True
+                        
                         # Log query processing start
                         logging.info(f"Processing query: {query}")
                         
@@ -916,11 +931,15 @@ def main():
                             # Log successful response
                             logging.info(f"Generated response: {response}")
                             
+                        # Reset processing state
+                        st.session_state.processing = False
+                            
                     except Exception as e:
                         # Log error details
                         logging.error(f"Query processing error: {str(e)}")
                         logging.error("Full error details: ", exc_info=True)
                         response_placeholder.warning("Unable to process your question. Please try again.")
+                        st.session_state.processing = False
                 else:
                     st.warning("Please enter a question!")