from rag_system.loaders.pdf_loader import load_pdf
from rag_system.loaders.web_loader import load_web_crawl
from rag_system.vectordb.azure_search import add_documents


def main():
    print("[1/2] Splitting and processing documents...")
    pdf_documents = load_pdf("data/verint-responsible-ethical-ai.pdf")
    # web_documents = load_web_crawl(["https://excalibur.mgmresorts.com/en.html"])
    # web_documents = load_web_crawl(["https://www.verint.com"])
    # web_documents = load_web_crawl("https://firecrawl.dev")
    print("[2/2] Generating and storing embeddings...")
    add_documents(pdf_documents)
    # add_documents(web_documents)
    print("Embeddings stored. You can now run the Streamlit app with:\n")
    print("   streamlit run rag_system/app/streamlit_app.py")


if __name__ == "__main__":
    main()