Update app.py

app.py

@@ -3,12 +3,58 @@ import numpy as np
 from PIL import Image
 
 def set_computer_params(computer):
-    # Esta función se mantiene igual que en el código original
-    # ... (código de set_computer_params)
+    if computer == "Commodore 64":
+        return 16, 4  # 16 niveles, pixelado moderado
+    elif computer == "ZX Spectrum":
+        return 8, 5  # 8 niveles, pixelado medio
+    elif computer == "Amstrad CPC":
+        return 16, 5  # 16 niveles, pixelado intermedio
+    elif computer == "Atari 8-bit (400/800)":
+        return 32, 6  # 32 niveles, pixelado más alto
+    elif computer == "Apple II":
+        return 6, 6  # 6 niveles, pixelado considerable
+    elif computer == "MSX":
+        return 16, 5  # 16 niveles, pixelado medio
+    elif computer == "NES (Nintendo)":
+        return 32, 4  # 32 niveles, pixelado moderado
+    elif computer == "IBM PC CGA":
+        return 4, 6  # 4 niveles, alto pixelado
+    return 16, 4  # Valores predeterminados (Commodore 64)
 
 def convert_to_pixelated_8bit(image, computer):
-    # Esta función se mantiene igual que en el código original
-    # ... (código de convert_to_pixelated_8bit)
+    if image is None:
+        return None
+    
+    # Obtener los parámetros según la computadora retro
+    intensity, pixelation_level = set_computer_params(computer)
+    
+    # Convertir la imagen a numpy array
+    img_array = np.array(image)
+    
+    # Calcular los niveles por canal según la intensidad seleccionada
+    levels = int(256 // intensity)
+    
+    # Reducir la cantidad de colores basada en la intensidad
+    img_array = (img_array // levels) * levels
+    
+    # Ajustar ligeramente el contraste
+    img_array = np.clip(img_array * 1.1, 0, 255).astype(np.uint8)
+    
+    # Aplicar el efecto de pixelado reduciendo y restaurando la resolución
+    img_pil = Image.fromarray(img_array)
+    
+    # Obtener las dimensiones originales
+    original_width, original_height = img_pil.size
+    
+    # Calcular el tamaño reducido para el pixelado
+    pixel_width = max(1, original_width // pixelation_level)
+    pixel_height = max(1, original_height // pixelation_level)
+    
+    # Redimensionar al tamaño más pequeño (pixelado) y luego de vuelta al tamaño original
+    img_pil = img_pil.resize((pixel_width, pixel_height), Image.NEAREST)  # Reducir
+    img_pil = img_pil.resize((original_width, original_height), Image.NEAREST)  # Volver al tamaño original
+    
+    return img_pil
 
 def process_image(image, computer):
     if image is None: