Update rag_routerv2.py

rag_routerv2.py

@@ -234,42 +234,39 @@ async def health_check():
 
 @router.on_event("startup")
 async def startup():
-   init_db()
-   print("RAG Router started")
-   
-   table_name = "digiyatra"
-   user_id = "digiyatra"
-   
-   # Create vector store and index
-   vector_store = LanceDBVectorStore(
-       uri="./lancedb/dev",
-       table_name=table_name,
-       mode="overwrite",
-       query_type="hybrid"
-   )
-
-   # Load CSV and create nodes
-   with open('combined_digi_yatra.csv', newline='') as f:
-       nodes = [
-           TextNode(text=str(row), id_=str(uuid.uuid4()))
-           for row in list(csv.reader(f))[1:]
-       ]
+    init_db()
+    print("RAG Router started")
+    
+    table_name = "digiyatra"
+    user_id = "digiyatra"
+    
+    db = get_db()
+    # Check if table already exists
+    existing = db.execute('SELECT id FROM tables WHERE table_id = ?', (table_name,)).fetchone()
+    if not existing:
+        vector_store = LanceDBVectorStore(
+            uri="./lancedb/dev",
+            table_name=table_name,
+            mode="overwrite", 
+            query_type="hybrid"
+        )
 
-   # Create and persist index
-   index = VectorStoreIndex(nodes, vector_store=vector_store)
-   index.storage_context.persist(persist_dir=f"./lancedb/index/{table_name}")
+        with open('combined_digi_yatra.csv', newline='') as f:
+            nodes = [TextNode(text=str(row), id_=str(uuid.uuid4())) 
+                    for row in list(csv.reader(f))[1:]]
 
-   # Store in SQLite
-   db = get_db()
-   db.execute(
-       'INSERT INTO tables (user_id, table_id, table_name) VALUES (?, ?, ?)',
-       (user_id, table_name, table_name)
-   )
-   db.execute(
-       'INSERT INTO table_files (table_id, filename, file_path) VALUES (?, ?, ?)',
-       (table_name, 'combined_digi_yatra.csv', 'combined_digi_yatra.csv')
-   )
-   db.commit()
+        index = VectorStoreIndex(nodes, vector_store=vector_store)
+        index.storage_context.persist(persist_dir=f"./lancedb/index/{table_name}")
+
+        db.execute(
+            'INSERT INTO tables (user_id, table_id, table_name) VALUES (?, ?, ?)',
+            (user_id, table_name, table_name)
+        )
+        db.execute(
+            'INSERT INTO table_files (table_id, filename, file_path) VALUES (?, ?, ?)',
+            (table_name, 'combined_digi_yatra.csv', 'combined_digi_yatra.csv')
+        )
+        db.commit()
 
 @router.on_event("shutdown")
 async def shutdown():