timothyafolami
25 lines
730 B
Python
25 lines
730 B
Python
from data_ingestion.utils import search, load_embedded_data
|
|
import sys, os
|
|
|
|
# Add the root directory to sys.path
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
|
from loggings.logging_config import logger
|
|
|
|
|
|
|
|
# loading the embedded data
|
|
embed_db = load_embedded_data()
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logger.info("Receiving the search query")
|
|
query = input("Enter the search query: ")
|
|
logger.info(f"Searching for {query}")
|
|
page_content, all, pages = search(embed_db, query)
|
|
logger.info("Search completed")
|
|
logger.info(f"Page content: {page_content}")
|
|
print(f"Page content: {page_content}")
|
|
print(f"Pages: {pages}")
|
|
print(f"All: {all}")
|
|
print("Search completed") |