Update route/app.py

route/app.py

@@ -11,6 +11,22 @@ import traceback
 from Rag.rag import rag_query, embeddings, initialize_hotels_vectorstore
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
+global_hotels_vectorstore = initialize_hotels_vectorstore()
+
+# Hàm để lấy hotels vectorstore
+def get_hotels_vectorstore():
+    global global_hotels_vectorstore
+    if global_hotels_vectorstore is None:
+        global_hotels_vectorstore = initialize_hotels_vectorstore()
+    return global_hotels_vectorstore
+
+# Hàm để truy vấn RAG với hotels vectorstore
+def query_hotels(question):
+    vs = get_hotels_vectorstore()
+    if vs is None:
+        return "Không thể khởi tạo hotels vector store. Vui lòng kiểm tra API key và thử lại."
+    return rag_query(question, vs)
+
 try:
     import main
     RAG_AVAILABLE = True