Spaces:
Running
Running
File size: 3,931 Bytes
c5e57d6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 |
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)
|