import argparse

import subprocess

import os

import sys

def run_scraper():

    print("Running the scraper...")

    subprocess.run([sys.executable, os.path.join('scraping', 'scraper.py')])

def run_preprocess():

    print("Running the preprocessor...")

    subprocess.run([sys.executable, os.path.join('ai', 'preprocess.py')])

def run_model():

    print("Running the model...")

    subprocess.run([sys.executable, os.path.join('ai', 'model.py')])

def run_tests():

    print("Running unit tests...")

    subprocess.run([sys.executable, '-m', 'unittest', 'discover', '-s', 'ai', '-p', '*_test.py'])

def main(scrape, preprocess, model, test):

    if scrape:

        run_scraper()

    if preprocess:

        run_preprocess()

    if model:

        run_model()  # Only run the model if specified

    if test:

        run_tests()

if __name__ == "__main__":

    parser = argparse.ArgumentParser(description="Master program to run scraping, preprocessing, and modeling.")

    parser.add_argument('--scrape', action='store_true', help="Run the scraper.")

    parser.add_argument('--preprocess', action='store_true', help="Run the preprocessor.")

    parser.add_argument('--test', action='store_true', help="Run unit tests.")

    args = parser.parse_args()

    main(args.scrape, args.preprocess, args.test)