Update app.py

app.py

@@ -9,6 +9,7 @@ from pydantic import BaseModel, Field
 from tempfile import NamedTemporaryFile
 from langchain_community.vectorstores import FAISS
 from langchain_core.vectorstores import VectorStore
+from langchain_core.documents import Document
 from langchain_community.document_loaders import PyPDFLoader
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from llama_parse import LlamaParse
@@ -462,7 +463,10 @@ def get_response_from_pdf(query, model, selected_docs, num_calls=3, temperature=
         return
 
     # Pre-filter the documents
-    filtered_docs = [doc for doc in database.docstore.values() if doc.metadata["source"] in selected_docs]
+    filtered_docs = []
+    for doc_id, doc in database.docstore._dict.items():
+        if isinstance(doc, Document) and doc.metadata.get("source") in selected_docs:
+            filtered_docs.append(doc)
     
     logging.info(f"Number of documents after pre-filtering: {len(filtered_docs)}")