Update app.py

app.py

@@ -48,10 +48,6 @@ def format_elapsed_time(time):
     # Format the elapsed time to two decimal places
     return "{:.2f}".format(time)
 
-def search_knowledgebase(query):
-    results = retriever(query)
-    return results
-
 def process_search_results(search_results):
     """
     Processes search results to extract and organize metadata and other details.
@@ -108,7 +104,7 @@ def lookup_related_chunks(df_chunks, chunk_id):
     return df_chunks[(df_chunks['Title'] == title) & (df_chunks['PageNumber'].isin(page_range))]
 
 
-def search_and_reconstruct(query, df_chunks):
+def search_and_reconstruct(query, df_chunks, k):
     """
     Combines search, lookup of related chunks, and text reconstruction.
 
@@ -118,7 +114,7 @@ def search_and_reconstruct(query, df_chunks):
     :param top_k: Number of top search results to retrieve.
     :return: A list of dictionaries with document title, page number, and reconstructed text.
     """
-    search_results = search_knowledgebase(query)
+    search_results = retriever(query, k)
     processed_results = process_search_results(search_results)
 
     reconstructed_results = []