Update app.py

app.py

@@ -95,17 +95,26 @@ def extract_text_from_pdf(pdf_filepath):
 
 def query_rag(query_text: str, chat_history):
     try:
+        # Ensure the database is initialized
+        initialize_chroma()
+
         embedding_function = HuggingFaceEmbeddings()
         db = Chroma(persist_directory=CHROMA_PATH, embedding_function=embedding_function)
         
+        # Perform similarity search
         results = db.similarity_search_with_score(query_text, k=5)
-        
+
         if not results:
+            print("No relevant information found.")
             return "Sorry, I couldn't find any relevant information."
         
+        # Combine document contents for context
         context_text = "\n\n---\n\n".join([doc.page_content for doc, _score in results])
         
+        # Prepare the prompt for the Falcon model
         prompt = f"Context:\n{context_text}\n\nQuestion:\n{query_text}"
+        
+        # Generate a response using the Falcon model
         response = generate_response(prompt, chat_history)
         
         return response
@@ -114,6 +123,7 @@ def query_rag(query_text: str, chat_history):
         return "An error occurred while querying the RAG system."
 
 
+
 def save_pdf_and_update_database(pdf_filepath):
     try:
         text = extract_text_from_pdf(pdf_filepath)