Upload 2 files

Example/rag_example.py

@@ -0,0 +1,18 @@
+import sys
+import chromadb
+from pathlib import Path
+PROJECT_ROOT = Path(__file__).resolve().parent.parent
+# transcripts_folder_path = '/home/nightwing/Codes/Xyzbot/Data/transcripts'
+# transcripts_folder_path = 'Data/transcripts'
+transcripts_folder_path = PROJECT_ROOT / "Data" / "transcripts"
+chromadb_path = PROJECT_ROOT / "Rag" / "chromadb.db"
+client = chromadb.PersistentClient(path=str(chromadb_path))
+collection = client.get_or_create_collection(name="yt_transcript_collection")
+sys.path.append(str(PROJECT_ROOT))
+sys.path.append(str(PROJECT_ROOT / "Rag"))
+# print("Python path:", sys.path)
+from Rag.rag_pipeline import main_workflow
+
+# Run the application
+if __name__ == "__main__":
+    main_workflow(transcripts_folder_path, collection)