From 561ebaa074a4edb1450e6c79711a29d6144801a2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=AD=D0=BC=D0=B8=D0=BB=D1=8C=20=D0=90=D0=BC=D0=B8=D1=80?=
 =?UTF-8?q?=D0=BE=D0=B2?= <emillamirov555777@mail.ru>
Date: Thu, 28 May 2026 16:43:54 +0000
Subject: [PATCH] add main.py

---
 main.py | 64 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 64 insertions(+)
 create mode 100644 main.py

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..4dd43d8
--- /dev/null
+++ b/main.py
@@ -0,0 +1,64 @@
+import os
+from dotenv import load_dotenv
+from agent import create_agent
+from vectorstore import create_vectorstore, load_documents
+
+# Load environment variables
+load_dotenv()
+
+
+def initialize_knowledge_base(documents_dir: str = "./documents"):
+    """Initialize the vectorstore and load documents if not already loaded."""
+    vectorstore = create_vectorstore()
+    
+    # Check if vectorstore is empty
+    collection_count = vectorstore._collection.count()
+    if collection_count == 0 and os.path.exists(documents_dir):
+        print(f"Loading documents from {documents_dir}...")
+        load_documents(documents_dir, vectorstore)
+    else:
+        print(f"Vectorstore already contains {collection_count} documents")
+    
+    return vectorstore
+
+
+def main():
+    """Main CLI chat loop."""
+    print("=" * 50)
+    print("RAG Agent with ChromaDB and Web Search")
+    print("=" * 50)
+    
+    # Initialize knowledge base
+    initialize_knowledge_base()
+    
+    # Create agent
+    agent = create_agent()
+    
+    print("\nAgent ready. Type 'exit' to quit.\n")
+    
+    while True:
+        try:
+            query = input("Запрос: ").strip()
+            
+            if query.lower() == "exit":
+                print("Goodbye!")
+                break
+            
+            if not query:
+                continue
+            
+            # Run agent
+            result = agent.invoke({"input": query})
+            answer = result.get("output", "No answer generated")
+            
+            print(f"\n{answer}\n")
+            
+        except KeyboardInterrupt:
+            print("\nGoodbye!")
+            break
+        except Exception as e:
+            print(f"Error: {e}")
+
+
+if __name__ == "__main__":
+    main()
\ No newline at end of file