Pixel8Bits / app.py
Marcepelaez's picture
Update app.py
cdfe4f9 verified
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 en MDQ con ❤️ por Peláez"
)
iface.launch()