Update app.py

app.py

@@ -133,7 +133,7 @@ def process_directory(directory):
                 continue
 
             with open(file_path, 'rb') as f:
-                if file_ext in ['.rst', '.md', '.txt', '.html', '.json', '.yaml']:
+                if file_ext in ['.rst', '.md', '.txt', '.html', '.json', '.yaml', '.py']:
                     text = f.read().decode('utf-8')
                 elif file_ext == '.pdf':
                     reader = PdfReader(f)
@@ -204,6 +204,8 @@ def initialize():
 
     download_gitlab_repo()
     all_texts, file_references = process_directory(REPOSITORY_DIRECTORY)   
+    for text in all_texts:
+        print(text)
     chunks = split_into_chunks(all_texts, file_references, CHUNK_SIZE, CHUNK_OVERLAP)
     vectorstore = setup_chroma(chunks, EMBEDDING_MODEL_NAME, PERSIST_DIRECTORY)
     llm = setup_llm(LLM_MODEL_NAME, LLM_TEMPERATURE, GROQ_API_KEY)