import re import argparse import os import pandas as pd SCRIPT_DIR = os.path.abspath(os.path.dirname(__file__)) IN_DATASET = os.path.join(SCRIPT_DIR, "data.csv") def search(query): df = pd.read_csv(IN_DATASET) def main(): parser = argparse.ArgumentParser() parser.add_argument("query", help="the query to search the corpus with", type=str) args = parser.parse_args() search(query) if __name__ == "__main__": main()