
    =hk                     2    S SK Jr  S SKJr   " S S\5      rg)    )BaseCommand)RAGChatbotServicec                   $    \ rS rSrSrS rS rSrg)Command   z+Build FAISS vector store from database datac                 &    UR                  SSSS9  g )Nz--force
store_truez)Force rebuild even if vector store exists)actionhelp)add_argument)selfparsers     S/home/developer/Desktop/shao/shao/shaoApp/management/commands/build_vector_store.pyadd_argumentsCommand.add_arguments   s    < 	 	
    c                    U R                   R                  U R                  R                  S5      5         [	        5       nUS   (       a,  U R                   R                  S5        UR                  5         OuUR                  5       (       a5  U R                   R                  U R                  R                  S5      5        O+U R                   R                  S5        UR                  5         UR                  5       nU R                   R                  U R                  R                  SUR                  SS5       S	UR                  S
S5       SUR                  SS5       S[        UR                  S/ 5      5       35      5        g ! [         aJ  nU R                   R                  U R                  R                  S[        U5       35      5         S nAg S nAff = f)NzStarting vector store build...forcezForce rebuild requested...z4Vector store already exists and loaded successfully!z3No existing vector store found. Building new one...z,Vector store build completed!
Applications: total_applicationsr   z
PDF Documents: total_pdf_documentsz
Extracted Data: total_extracted_dataz
Councils: councilszError building vector store: )stdoutwritestyleSUCCESSr   build_vector_storeload_vector_storeget_statisticsgetlen	ExceptionERRORstr)r   argsoptionschatbotstatses         r   handleCommand.handle   s   JJ?@	
	')Gw!!">?**, ,,..KK%%

**+ab KK%%&[\..0 **,EKK

""%%*YY/CQ%G$H I&&+ii0Eq&I%J K'',yy1G'K&L M!!$UYYz2%>!? @	B  	KK

  #@Q!IJ 	s   EF 
GA GG N)__name__
__module____qualname____firstlineno__r   r   r*   __static_attributes__r,   r   r   r   r      s    8D
$r   r   N)django.core.management.baser   shaoApp.rag_servicer   r   r,   r   r   <module>r4      s    3 1.k .r   