From efaf4654fa04c1003609b428064856282919ec5c Mon Sep 17 00:00:00 2001 From: Kirill Markin Date: Sat, 8 Jun 2024 11:21:11 +0200 Subject: [PATCH] cleanup --- repo_to_text/main.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/repo_to_text/main.py b/repo_to_text/main.py index 5a35053..da633ec 100644 --- a/repo_to_text/main.py +++ b/repo_to_text/main.py @@ -84,7 +84,11 @@ def save_repo_to_text(path='.') -> None: output_file = 'repo_structure.txt' with open(output_file, 'w') as file: - file.write(tree_structure + '\n') + project_name = os.path.basename(os.path.abspath(path)) + file.write(f'Project: {project_name}\n\n') + file.write('Directory Structure:\n') + file.write('```\n.\n') + file.write(tree_structure + '\n' + '```\n') logging.debug('Tree structure written to file') for root, _, files in os.walk(path): @@ -97,7 +101,7 @@ def save_repo_to_text(path='.') -> None: relative_path = relative_path.replace('./', '', 1) - file.write(f'\n{relative_path}\n') + file.write(f'\nContents of {relative_path}:\n') file.write('```\n') try: with open(file_path, 'r', encoding='utf-8') as f: