Update app.py

app.py

@@ -140,7 +140,7 @@ def rag_workflow(query):
     formatted_doc_snippets = rag_chain.format_documents(kadiAPY_doc_documents)
     formatted_code_snippets = rag_chain.format_documents(kadiAPY_code_documents)
     print("FORMATTED Retrieved Document Contexts:", formatted_doc_snippets)
-    print("FORMATTED Retrieved Code Contexts:" , formatted_doc_snippets)
+    print("FORMATTED Retrieved Code Contexts:" , formatted_code_snippets)
     print(formatted_code_snippets)
     # Step 4: Generate the final response
     response = rag_chain.generate_response(query, formatted_doc_snippets, formatted_code_snippets)
@@ -160,13 +160,13 @@ def initialize():
     )
     
     code_texts, code_references = extract_files_and_filepath_from_dir(DATA_DIR, ['kadi_apy'], [])
-    doc_texts, kadiAPY_doc_references = extract_files_and_filepath_from_dir(DATA_DIR, ['docs/source/'], [])
+    doc_texts, doc_references = extract_files_and_filepath_from_dir(DATA_DIR, ['docs/source/'], [])
     
     print("LEEEEEEEEEEEENGTH of code_texts: ", len(code_texts))
     print("LEEEEEEEEEEEENGTH of doc_files: ", len(doc_texts))
     
     code_chunks = split_python_code_into_chunks(code_texts, code_references)
-    doc_chunks = split_into_chunks(doc_texts, kadiAPY_doc_references, CHUNK_SIZE, CHUNK_OVERLAP)
+    doc_chunks = split_into_chunks(doc_texts, doc_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)}")