Upload files to "image/script"

image/script/generate_rows.py

@@ -0,0 +1,73 @@
+# gpt
+
+import os
+from PIL import Image, ImageOps, ImageEnhance
+
+def combine_images(images):
+    """Combine multiple Image objects horizontally."""
+    widths, heights = zip(*(i.size for i in images))
+
+    total_width = sum(widths)
+    max_height = max(heights)
+
+    new_im = Image.new('RGB', (total_width, max_height))
+
+    x_offset = 0
+    for im in images:
+        new_im.paste(im, (x_offset, 0))
+        x_offset += im.size[0]
+
+    return new_im
+
+def find_image_path(base_path, file_name):
+    """Find the image path, checking for both PNG and JPEG extensions."""
+    for ext in ['.png', '.jpg', '.jpeg']:
+        path = os.path.join(base_path, file_name + ext)
+        if os.path.exists(path):
+            return path
+    return None
+
+def process_and_colorize_image(image_path, red, green, blue):
+    """Process and colorize the image."""
+    image = Image.open(image_path).convert("L")  # Convert to grayscale
+
+    # Colorize the image
+    colormap = ImageOps.colorize(image, black=(0, 0, 0), white=(red, green, blue))
+
+    # Enhance brightness
+    enhancer = ImageEnhance.Brightness(colormap)
+    colormap = enhancer.enhance(1.5)
+
+    return colormap
+
+def process_file(file_path, images_directory):
+    images_by_row = {}
+
+    with open(file_path, 'r') as file:
+        for line in file:
+            parts = line.split()
+            if len(parts) < 6:
+                continue
+
+            file_name, x, y, red, green, blue = parts[0], int(parts[1]), int(parts[2]), int(parts[3]), int(parts[4]), int(parts[5])
+            image_path = find_image_path(images_directory, file_name)
+
+            if image_path is None:
+                print(f"Image not found: {file_name}")
+                continue
+
+            processed_image = process_and_colorize_image(image_path, red, green, blue)
+
+            if y not in images_by_row:
+                images_by_row[y] = []
+
+            images_by_row[y].append(processed_image)
+
+    for y, images in images_by_row.items():
+        combined_image = combine_images(images)
+        output_filename = f"image_row_{y}.png"
+        combined_image.save("../rows/" + output_filename)
+        print(f"Saved {output_filename}")
+
+# Replace 'your_file.txt' with the path to your text file
+process_file('../image.txt', '../original')