documenter.py

import os
import json

# Define a set of multimedia file extensions
MULTIMEDIA_EXTENSIONS = {'.jpg', '.jpeg', '.png', '.gif', '.bmp', 
 '.mp3', '.wav', '.aac', '.flac',
 '.mp4', '.avi', '.mov', '.mkv', '.wmv'}

def get_directory_structure(root_dir):
 def build_structure(current_dir):
 structure = []
 for item in os.listdir(current_dir):
 item_path = os.path.join(current_dir, item)
 if os.path.isdir(item_path):
 structure.append({
 "name": item,
 "type": "directory",
 "content": None,
 "children": build_structure(item_path)
 })
 else:
 file_extension = os.path.splitext(item)[1].lower()
 if file_extension in MULTIMEDIA_EXTENSIONS:
 # For multimedia files, don't capture the content
 structure.append({
 "name": item,
 "type": "file",
 "content": None, # No content for multimedia files
 "children": []
 })
 else:
 with open(item_path, 'r', encoding='utf-8', errors='ignore') as file:
 file_content = file.read()
 structure.append({
 "name": item,
 "type": "file",
 "content": file_content,
 "children": []
 })
 return structure
 
 return build_structure(root_dir)

def generate_json(output_file):
 current_directory = os.getcwd()
 structure = get_directory_structure(current_directory)
 with open(output_file, 'w', encoding='utf-8') as json_file:
 json.dump(structure, json_file, indent=4)

if __name__ == "__main__":
 output_json_file = "output.json"
 generate_json(output_json_file)
 print(f"Directory structure saved to {output_json_file}")