Example/rag_example.py

import chromadb
transcripts_folder_path = '/home/nightwing/Codes/Xyzbot/Data/transcripts'
chromadb_path = "/home/nightwing/Codes/Xyzbot/Rag/chromadb.db"
client = chromadb.PersistentClient(path=chromadb_path)
collection = client.get_or_create_collection(name="yt_transcript_collection")
from Rag.rag_pipeline import main_workflow

# Run the application
if __name__ == "__main__":
    main_workflow(transcripts_folder_path, collection)