ingest.py

#!/usr/bin/env python3
"""
ingest.py – Run this once to index all documents into Endee.
Usage:
    python ingest.py
    python ingest.py --host http://localhost:8080 --recreate
"""

import argparse
import sys
import os
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "src"))

from indexer import DocumentIndexer


def main():
    parser = argparse.ArgumentParser(description="Index industrial documents into Endee.")
    parser.add_argument("--host", default="http://localhost:8080", help="Endee server URL")
    parser.add_argument("--token", default="", help="Auth token (leave empty if none)")
    parser.add_argument("--recreate", action="store_true", help="Drop and recreate the index")
    args = parser.parse_args()

    print("=" * 55)
    print("  IndustrialDocSearch – Endee Document Ingestion")
    print("=" * 55)
    print(f"  Host    : {args.host}")
    print(f"  Recreate: {args.recreate}")
    print("=" * 55)

    indexer = DocumentIndexer(host=args.host, auth_token=args.token)
    indexer.setup(recreate=args.recreate)

    print("\n✅ All done! You can now run the app:")
    print("   streamlit run app.py")


if __name__ == "__main__":
    main()