Update demo_generate/halftone_method.py

demo_generate/halftone_method.py

@@ -1,66 +1,65 @@
-from PIL import Image, ImageDraw
 import io
 import logging
 from datetime import datetime
+from PIL import Image, ImageDraw, ImageOps
+import numpy as np
 
 logger = logging.getLogger(__name__)
 
-def halftone_effect(img, dot_size=10):
-    """
-    Apply a halftone effect to the input image.
-    """
-    grayscale = img.convert('L')
-    width, height = grayscale.size
-
-    halftone_image = Image.new('L', (width, height), color=255)
-    draw = ImageDraw.Draw(halftone_image)
-
-    for x in range(0, width, dot_size):
-        for y in range(0, height, dot_size):
-            block_width = min(dot_size, width - x)
-            block_height = min(dot_size, height - y)
-
-            region = grayscale.crop((x, y, x + block_width, y + block_height))
-            avg_brightness = sum(region.getdata()) / (block_width * block_height)
-
-            max_radius = dot_size // 2
-            radius = int((255 - avg_brightness) / 255 * max_radius)
-            radius = max(0, min(radius, max_radius))
-
-            center_x = x + dot_size // 2
-            center_y = y + dot_size // 2
-            draw.ellipse(
-                (
-                    center_x - radius,
-                    center_y - radius,
-                    center_x + radius,
-                    center_y + radius
-                ),
-                fill=0
-            )
-
-    return halftone_image
-
-def apply_halftone(image_data, dot_size=10):
+def apply_halftone(image, dot_size=10):
+    """Görüntüye halftone efekti uygular"""
     start_time = datetime.now()
     logger.info(f"Applying halftone effect - {start_time}")
     
     try:
-        input_image = Image.open(io.BytesIO(image_data))
-        logger.info(f"Image opened. Mode: {input_image.mode}, Size: {input_image.size}")
+        # Eğer image bir dosya verisi (bytes) ise Image'a dönüştür
+        if isinstance(image, bytes):
+            image = Image.open(io.BytesIO(image))
+        
+        # Görüntüyü gri tonlamalı yap
+        if image.mode != 'L':
+            image = image.convert('L')
+        
+        # Orijinal boyutları sakla
+        orig_width, orig_height = image.size
         
-        dot_size = max(5, min(20, dot_size))
-        logger.info(f"Applying halftone effect with dot size: {dot_size}")
+        # Yeniden boyutlandırma faktörü
+        resize_factor = dot_size / 5
         
-        output_image = halftone_effect(input_image, dot_size)
+        # Geçici olarak küçült
+        small_width = int(orig_width / resize_factor)
+        small_height = int(orig_height / resize_factor)
+        small_image = image.resize((small_width, small_height), Image.BILINEAR)
         
-        img_io = io.BytesIO()
-        output_image.save(img_io, 'PNG')
-        img_io.seek(0)
+        # Yeni görüntü oluştur
+        output_image = Image.new('L', (orig_width, orig_height), 255)
+        draw = ImageDraw.Draw(output_image)
+        
+        # Halftone noktalarını çiz
+        for y in range(small_height):
+            for x in range(small_width):
+                # Piksel parlaklığını al
+                brightness = 255 - small_image.getpixel((x, y))
+                
+                # Nokta boyutunu hesapla
+                radius = int((brightness / 255) * (dot_size / 2))
+                
+                # Noktayı çiz
+                if radius > 0:
+                    center_x = int((x * resize_factor) + (resize_factor / 2))
+                    center_y = int((y * resize_factor) + (resize_factor / 2))
+                    bbox = [
+                        center_x - radius,
+                        center_y - radius,
+                        center_x + radius,
+                        center_y + radius
+                    ]
+                    draw.ellipse(bbox, fill=0)
         
         duration = (datetime.now() - start_time).total_seconds()
         logger.info(f"Successfully processed. Duration: {duration} seconds")
-        return img_io
+        return output_image
+        
     except Exception as e:
         logger.error(f"Halftone processing error: {str(e)}", exc_info=True)
         raise