#!/usr/bin/env python3 """ Generate paginated HTML galleries for folders containing images. Places HTML files in an 'html/' subfolder within each image directory. """ import os from pathlib import Path from collections import defaultdict # Configuration IMAGES_PER_PAGE = 100 IMAGE_EXTENSIONS = {'.jpg', '.jpeg', '.png', '.gif', '.tif', '.tiff'} BASE_DIR = Path('/mnt/vision/data/kaust') def get_image_files(folder): """Get all image files in a folder (non-recursive).""" images = [] try: for item in sorted(os.listdir(folder)): if Path(item).suffix.lower() in IMAGE_EXTENSIONS: images.append(item) except (PermissionError, OSError): pass return images def generate_page(folder, images, page_num, total_pages, html_dir, folder_name): """Generate a single HTML page for a gallery.""" start_idx = (page_num - 1) * IMAGES_PER_PAGE end_idx = start_idx + IMAGES_PER_PAGE page_images = images[start_idx:end_idx] # Navigation prev_link = f'page{page_num-1}.html' if page_num > 1 else None next_link = f'page{page_num+1}.html' if page_num < total_pages else None index_link = '../../index.html' html_content = f""" {folder_name} - Page {page_num} of {total_pages}

{folder_name}

Gallery: {len(images)} total images | Page {page_num} of {total_pages} ({len(page_images)} images on this page)

← Back to Index 📁 Browse Folder {'← Previous' if prev_link else '← Previous'} Page {page_num} / {total_pages} {'Next →' if next_link else 'Next →'}
""" for img in page_images: img_path = f'../{img}' html_content += f"""
{img}
{img}
Download
""" html_content += """
""" html_content += f""" ← Back to Index 📁 Browse Folder {'← Previous' if prev_link else '← Previous'} Page {page_num} / {total_pages} {'Next →' if next_link else 'Next →'}
""" output_file = html_dir / f'page{page_num}.html' with open(output_file, 'w') as f: f.write(html_content) return output_file def create_index_page(folder, total_pages, html_dir, folder_name, images_count): """Create an index page for the gallery.""" html_content = f""" {folder_name} - Gallery Index

{folder_name} - Gallery

Total images: {images_count} | Pages: {total_pages} | Images per page: {IMAGES_PER_PAGE}

← Back to Main Index 📁 Browse Folder Start Gallery →
""" for i in range(1, total_pages + 1): start = (i - 1) * IMAGES_PER_PAGE + 1 end = min(i * IMAGES_PER_PAGE, images_count) html_content += f' Page {i}
Images {start}-{end}\n' html_content += """
""" output_file = html_dir / 'index.html' with open(output_file, 'w') as f: f.write(html_content) return output_file def process_folder(folder_path): """Process a single folder and generate gallery pages.""" folder = Path(folder_path) images = get_image_files(folder) if len(images) < 5: # Skip folders with very few images return None # Create html subfolder html_dir = folder / 'html' html_dir.mkdir(exist_ok=True) # Calculate pagination total_pages = (len(images) + IMAGES_PER_PAGE - 1) // IMAGES_PER_PAGE folder_name = folder.name # Generate pages for page_num in range(1, total_pages + 1): generate_page(folder, images, page_num, total_pages, html_dir, folder_name) # Generate index create_index_page(folder, total_pages, html_dir, folder_name, len(images)) return { 'folder': folder_name, 'images': len(images), 'pages': total_pages } def main(): """Main function to process all folders.""" print("Scanning for image folders...") results = [] # Process all GSxxxxxx folders gs_folders = sorted([d for d in BASE_DIR.iterdir() if d.is_dir() and d.name.startswith('GS')]) print(f"\nFound {len(gs_folders)} GSxxxxxx folders") print("Generating galleries...") for folder in gs_folders: print(f"Processing {folder.name}...", end=' ') result = process_folder(folder) if result: results.append(result) print(f"✓ {result['images']} images, {result['pages']} pages") else: print("✗ skipped (too few images)") # Process subdirectories in GSxxxxxx folders (fixed/, distances-m/) print("\nProcessing subdirectories...") for folder in gs_folders: for subfolder in ['fixed', 'distances-m']: subdir = folder / subfolder if subdir.exists() and subdir.is_dir(): print(f"Processing {folder.name}/{subfolder}...", end=' ') result = process_folder(subdir) if result: results.append(result) print(f"✓ {result['images']} images, {result['pages']} pages") else: print("✗ skipped") # Process other folders with images print("\nProcessing other folders...") for folder_name in ['panos', 'buildings']: folder_path = BASE_DIR / folder_name if folder_path.exists(): # Process subfolders for subfolder in folder_path.iterdir(): if subfolder.is_dir(): print(f"Processing {folder_name}/{subfolder.name}...", end=' ') result = process_folder(subfolder) if result: results.append(result) print(f"✓ {result['images']} images, {result['pages']} pages") else: print("✗ skipped") print(f"\n{'='*60}") print(f"Complete! Generated galleries for {len(results)} folders") print(f"Total images indexed: {sum(r['images'] for r in results)}") print(f"Total pages generated: {sum(r['pages'] for r in results)}") if __name__ == '__main__': main()