main.py 461 B

1234567891011121314151617181920
  1. from pathlib import Path
  2. import sys
  3. ROOT = Path(__file__).resolve().parent
  4. SRC = ROOT / "src"
  5. DATA_ROOT = ROOT.parents[1] / "data" / "rss_digest"
  6. if str(SRC) not in sys.path:
  7. sys.path.insert(0, str(SRC))
  8. from rss_digest.pipeline import run_pipeline
  9. from rss_digest.ui_server import serve_ui
  10. if __name__ == "__main__":
  11. if len(sys.argv) > 1 and sys.argv[1] == "run":
  12. run_pipeline(ROOT, DATA_ROOT)
  13. else:
  14. serve_ui(ROOT, DATA_ROOT)