Update app.py

app.py

@@ -447,31 +447,31 @@ def rag_workflow(query):
 def initialize():
     global docstore, codestore, chunks, llm
     #download_gitlab_repo()
-    #code_partial_paths = ['kadi_apy/lib/resources/']
+    code_partial_paths = ['kadi_apy/lib/resources/']
     code_file_path = []
     #doc_partial_paths = []
-    #doc_partial_paths = ['docs/source/setup/']
-    #doc_file_paths = ['docs/source/usage/lib.rst']
+    doc_partial_paths = ['docs/source/setup/']
+    doc_file_paths = ['docs/source/usage/lib.rst']
 
 
     code_partial_paths = ['kadi_apy/lib/resources/']
     
     process_directory(DATA_DIR,code_partial_paths, code_file_path)
     
-    #code_files, code_file_references = process_directory5(DATA_DIR, code_partial_paths, code_file_path)
+    code_files, code_file_references = process_directory5(DATA_DIR, code_partial_paths, code_file_path)
     
-    #doc_files, doc_file_references = process_directory(DATA_DIR, doc_partial_paths, doc_file_paths)
+    doc_files, doc_file_references = process_directory5(DATA_DIR, doc_partial_paths, doc_file_paths)
     
-    #code_chunks = split_pythoncode_into_chunks(code_files, code_file_references, 1500, 0)
-    #doc_chunks = split_into_chunks(doc_files, doc_file_references, CHUNK_SIZE, CHUNK_OVERLAP)
+    code_chunks = split_pythoncode_into_chunks(code_files, code_file_references, 1500, 0)
+    doc_chunks = split_into_chunks(doc_files, doc_file_references, CHUNK_SIZE, CHUNK_OVERLAP)
 
-    #print(f"Total number of code_chunks: {len(code_chunks)}")
-    #print(f"Total number of doc_chunks: {len(doc_chunks)}")
+    print(f"Total number of code_chunks: {len(code_chunks)}")
+    print(f"Total number of doc_chunks: {len(doc_chunks)}")
 
-    #docstore = setup_vectorstore(doc_chunks, EMBEDDING_MODEL_NAME, PERSIST_DOC_DIRECTORY)
-    #codestore = setup_vectorstore(code_chunks, EMBEDDING_MODEL_NAME, PERSIST_CODE_DIRECTORY)
+    docstore = setup_vectorstore(doc_chunks, EMBEDDING_MODEL_NAME, PERSIST_DOC_DIRECTORY)
+    codestore = setup_vectorstore(code_chunks, EMBEDDING_MODEL_NAME, PERSIST_CODE_DIRECTORY)
     
-    #llm = setup_llm(LLM_MODEL_NAME, LLM_TEMPERATURE, GROQ_API_KEY)
+    llm = setup_llm(LLM_MODEL_NAME, LLM_TEMPERATURE, GROQ_API_KEY)
 
 
 initialize()