Diff of /src/master.py [000000] .. [8c54ae]

Switch to side-by-side view

--- a
+++ b/src/master.py
@@ -0,0 +1,40 @@
+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)