o =hk@s,ddlmZddlmZGdddeZdS)) BaseCommand)RAGChatbotServicec@s eZdZdZddZddZdS)Commandz+Build FAISS vector store from database datacCs|jdddddS)Nz--force store_truez)Force rebuild even if vector store exists)actionhelp) add_argument)selfparserr U/home/shobhit/Desktop/shao (1)/shao/shaoApp/management/commands/build_vector_store.py add_argumentss  zCommand.add_argumentscOs|j|jdzZt}|dr|jd|n|r,|j|jdn |jd||}|j|jd|ddd |d dd |d dd t |dgWdSt y}z|j|j dt |WYd}~dSd}~ww)NzStarting vector store build...forcezForce rebuild requested...z4Vector store already exists and loaded successfully!z3No existing vector store found. Building new one...z,Vector store build completed! Applications: total_applicationsrz PDF Documents: total_pdf_documentsz Extracted Data: total_extracted_dataz Councils: councilszError building vector store: ) stdoutwritestyleSUCCESSrbuild_vector_storeload_vector_storeget_statisticsgetlen ExceptionERRORstr)r argsoptionschatbotstatser r r handlesD          zCommand.handleN)__name__ __module__ __qualname__rr r$r r r r rs rN)django.core.management.basershaoApp.rag_servicerrr r r r s