Update app.py

app.py

@@ -88,8 +88,8 @@ def update_vectors(files, parser):
             logging.info(f"Loaded {len(data)} chunks from {file.name}")
             all_data.extend(data)
             total_chunks += len(data)
-            if not any(doc["name"] == file.name for doc in uploaded_documents):
-                uploaded_documents.append({"name": file.name, "selected": True})
+            if not any(doc["name"] == os.path.basename(file.name) for doc in uploaded_documents):
+                uploaded_documents.append({"name": os.path.basename(file.name), "selected": True})
                 logging.info(f"Added new document to uploaded_documents: {file.name}")
             else:
                 logging.info(f"Document already exists in uploaded_documents: {file.name}")
@@ -322,8 +322,8 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sel
                     return
                 
                 # Filter relevant documents based on user selection
-                filtered_docs = [doc for doc in relevant_docs if doc.metadata["source"] in selected_docs]
-                logging.info(f"Filtered to {len(filtered_docs)} documents based on user selection")
+                filtered_docs = [doc for doc in relevant_docs if os.path.basename(doc.metadata["source"]) in [os.path.basename(doc) for doc in selected_docs]]
+                logging.info(f"Number of filtered documents: {len(filtered_docs)}")
                 
                 if not filtered_docs:
                     logging.warning("No relevant information found in the selected documents")
@@ -353,6 +353,7 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sel
         yield f"An unexpected error occurred: {str(e)}"
 
     logging.info("Finished respond function")
+    logging.info(f"Selected docs: {selected_docs}")
 
 def get_response_from_cloudflare(prompt, context, query, num_calls=3, temperature=0.2, search_type="pdf"):
     headers = {
@@ -511,6 +512,9 @@ Write a detailed and complete response that answers the following user question:
                 logging.error(f"Error in API call {i+1}: {str(e)}")
         
         logging.info("Finished generating response")
+        logging.info(f"Relevant docs: {[doc.metadata['source'] for doc in relevant_docs]}")
+        logging.info(f"Selected docs: {selected_docs}")
+        logging.info(f"Filtered docs: {[doc.metadata['source'] for doc in filtered_docs]}")
 
 def vote(data: gr.LikeData):
     if data.liked:
@@ -534,8 +538,8 @@ uploaded_documents = []
 
 def display_documents():
     return gr.CheckboxGroup(
-        choices=[doc["name"] for doc in uploaded_documents],
-        value=[doc["name"] for doc in uploaded_documents if doc["selected"]],
+        choices=[os.path.basename(doc["name"]) for doc in uploaded_documents],
+        value=[os.path.basename(doc["name"]) for doc in uploaded_documents if doc["selected"]],
         label="Select documents to query"
     )