Spaces:
Running
Running
import gradio as gr | |
import numpy as np | |
from PIL import Image | |
# Función que ajusta los parámetros según la computadora seleccionada | |
def set_computer_params(computer): | |
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) | |
# Función principal que convierte la imagen según los parámetros seleccionados | |
def convert_to_pixelated_8bit(image, computer): | |
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 | |
# Crear la interfaz gráfica de Gradio | |
iface = gr.Interface( | |
fn=convert_to_pixelated_8bit, | |
inputs=[ | |
gr.Image(type="numpy", label="Subir Imagen"), # Subida de imagen | |
gr.Dropdown( # Selección de computadora retro | |
["Commodore 64", "ZX Spectrum", "Amstrad CPC", "Atari 8-bit (400/800)", "Apple II", "MSX", "NES (Nintendo)", "IBM PC CGA"], | |
label="Elige una computadora retro", | |
value="Commodore 64" # Valor predeterminado | |
) | |
], | |
outputs=gr.Image(type="numpy", label="Imagen convertida"), # Imagen resultante | |
title="Emulador de Gráficos de Computadoras Retro", | |
description="Sube una imagen y emula el estilo gráfico de computadoras de los años 80 con 8 bits y pixelado. Elige entre varias computadoras retro.", | |
# Agregando el mensaje "Hecho con ❤️ por Peláez" | |
article="Hecho con en MDQ❤️ por Peláez" | |
) | |
iface.launch() | |