Update app.py

app.py

@@ -74,15 +74,17 @@ def store_document_data(PDF_FILE):
 def retrieve_document(query):
     print(f"Retrieving document based on:\n{query}")
 
-    embeddings_file = requests.get(API_URL_EMBEDDINGS)
-    metadata_file = requests.get(API_URL_METADATA)
+    embeddings_ = requests.get(API_URL_EMBEDDINGS)
+    metadata_ = requests.get(API_URL_METADATA)
 
-    print(embeddings_file.json(), metadata_file.json()['metadata_file'])
+    embeddings_file = embeddings_.json()
+    metadata_dile = metadata_.json()
+    print(embeddings_file, metadata_file)
     # Generate query embedding
     query_embedding = embedding_model.encode([query]).astype(np.float32)
 
     # Search for the closest document in FAISS index
-    index = faiss.read_index(embeddings_file.json()['index_file'])
+    index = faiss.read_index(embeddings_file)
     _, closest_idx = index.search(query_embedding, 1)
 
     with open(metadata_file, "r") as f: