keyword search update

src/chatbot.py

@@ -70,13 +70,13 @@ def keyword_search(query, db=db, embeddings=embeddings):
     # Format vector store query results into dataframe
     #print(results[0][0].metadata.keys())
     
-    df_res = pd.DataFrame(columns=['Speech Content', 'Relevance']) # Add Date/Party/Politician
+    df_res = pd.DataFrame(columns=['Speech Content','Date','Relevance']) # Add Date/Party/Politician
     for doc in results:
         speech_content = doc[0].page_content
-        #speech_date = doc[0].metadata["date"]
+        speech_date = doc[0].metadata["date"]
         score = doc[1] # Relevance based on relevance search
         df_res = pd.concat([df_res, pd.DataFrame({'Speech Content': [speech_content],
-                                                          #'Date': [speech_date],
+                                                          'Date': [speech_date],
                                                           'Relevance': [score]})], ignore_index=True)
     
     df_res.sort_values('Relevance', inplace=True)