Spaces:
Running
Running
| import requests | |
| from bs4 import BeautifulSoup | |
| from PIL import Image | |
| from io import BytesIO | |
| import time | |
| # Global variables | |
| H = 200 | |
| ROWS = 6 | |
| PADDING = 30 | |
| # Step 1: Fetch the Images from the URL Folder | |
| def fetch_image_urls(url): | |
| response = requests.get(url + '?t=' + str(time.time())) | |
| soup = BeautifulSoup(response.content, 'html.parser') | |
| images = {} | |
| for node in soup.find_all('a'): | |
| href = node.get('href') | |
| if href.endswith(('.png', '.jpg', '.jpeg')): | |
| try: | |
| image_index = int(href.split('__')[0]) | |
| images[image_index] = url + '/' + href + '?t=' + str(time.time()) | |
| except ValueError: | |
| print(f"Skipping invalid image: {href}") | |
| return images | |
| # Step 2: Resize Images to Height H | |
| def fetch_image(url): | |
| response = requests.get(url) | |
| return Image.open(BytesIO(response.content)) | |
| def resize_images(images, target_height): | |
| resized_images = {} | |
| for index, img in images.items(): | |
| ratio = target_height / img.height | |
| new_width = int(img.width * ratio) | |
| resized_img = img.resize((new_width, target_height), Image.BICUBIC) | |
| resized_images[index] = resized_img | |
| return resized_images | |
| # Step 3: Create a Collage with Efficient Placement Algorithm | |
| def create_collage(image_urls, collage_path, H, ROWS, PADDING): | |
| images = {index: fetch_image(url) for index, url in image_urls.items()} | |
| resized_images = resize_images(images, H) # Resize to H pixels height | |
| center_image = resized_images.pop(0) | |
| other_images = list(resized_images.items()) | |
| # Calculate collage size based on the number of rows | |
| collage_width = 3000 # 16:9 aspect ratio width | |
| collage_height = (H + PADDING) * ROWS + 2 * PADDING # Adjust height based on number of rows, add padding to top and bottom | |
| collage = Image.new('RGB', (collage_width, collage_height), (255, 255, 255)) | |
| # Sort images by width and height | |
| sorted_images = sorted(other_images, key=lambda x: x[1].width * x[1].height, reverse=True) | |
| # Create alternate placement list and insert the center image in the middle | |
| alternate_images = [] | |
| i, j = 0, len(sorted_images) - 1 | |
| halfway_point = (len(sorted_images) + 1) // 2 | |
| count = 0 | |
| while i <= j: | |
| if count == halfway_point: | |
| alternate_images.append((0, center_image)) | |
| if i == j: | |
| alternate_images.append(sorted_images[i]) | |
| else: | |
| alternate_images.append(sorted_images[i]) | |
| alternate_images.append(sorted_images[j]) | |
| i += 1 | |
| j -= 1 | |
| count += 2 | |
| # Calculate number of images per row | |
| images_per_row = len(alternate_images) // ROWS | |
| extra_images = len(alternate_images) % ROWS | |
| # Place images in rows with only padding space between them | |
| def place_images_in_rows(images, collage, max_width, padding, row_height, rows, images_per_row, extra_images): | |
| y = padding | |
| for current_row in range(rows): | |
| row_images_count = images_per_row + (1 if extra_images > 0 else 0) | |
| extra_images -= 1 if extra_images > 0 else 0 | |
| row_images = images[:row_images_count] | |
| row_width = sum(img.width for idx, img in row_images) + padding * (row_images_count - 1) | |
| x = (max_width - row_width) // 2 | |
| for idx, img in row_images: | |
| collage.paste(img, (x, y)) | |
| x += img.width + padding | |
| y += row_height + padding | |
| images = images[row_images_count:] | |
| place_images_in_rows(alternate_images, collage, collage_width, PADDING, H, ROWS, images_per_row, extra_images) | |
| collage.save(collage_path) | |
| # Define the URL folder and other constants | |
| url_folder = 'https://leafmachine.org/partners/' | |
| collage_path = 'img/collage.jpg' | |
| # Fetch, Create, and Update | |
| image_urls = fetch_image_urls(url_folder) | |
| create_collage(image_urls, collage_path, H, ROWS, PADDING) | |