Update rag_engine.py

rag_engine.py

@@ -118,16 +118,15 @@ def load_data_files():
         st.error("Failed to initialize required services")
         return None, None, None
     
-    # Download necessary files with a spinner to show progress
-    with st.spinner("Downloading necessary files..."):
-        success = True
-        success &= download_file_from_gcs(bucket, faiss_index_file_gcs, local_faiss_index_file)
-        success &= download_file_from_gcs(bucket, text_chunks_file_gcs, local_text_chunks_file)
-        success &= download_file_from_gcs(bucket, metadata_file_gcs, local_metadata_file)
-        
-        if not success:
-            st.error("Failed to download required files")
-            return None, None, None
+    # Download necessary files - remove the spinner from here
+    success = True
+    success &= download_file_from_gcs(bucket, faiss_index_file_gcs, local_faiss_index_file)
+    success &= download_file_from_gcs(bucket, text_chunks_file_gcs, local_text_chunks_file)
+    success &= download_file_from_gcs(bucket, metadata_file_gcs, local_metadata_file)
+    
+    if not success:
+        st.error("Failed to download required files")
+        return None, None, None
     
     # Load FAISS index
     try:
@@ -299,7 +298,6 @@ def answer_with_llm(query, context=None, word_limit=100):
         user_message = f"""
         Context:
         {formatted_context}
-
         Question:
         {query}
         """
@@ -331,8 +329,8 @@ def answer_with_llm(query, context=None, word_limit=100):
         return "I apologize, but I'm unable to answer at the moment."
 
 def format_citations(sources):
-    """Format citations to display each one on a new line."""
-    return "\n".join([f"📚 {title} by {author}, Published by {publisher}" for title, author, publisher in sources])
+    """Format citations to display each one on a new line with a full stop."""
+    return "\n".join([f"📚 {title} by {author}, Published by {publisher}." for title, author, publisher in sources])
 
 def process_query(query, top_k=5, word_limit=100):
     """Process a query through the RAG pipeline with proper formatting."""