import gradio as gr
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)

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)

def process_image(image, computer):
    if image is None:
        return None
    return convert_to_pixelated_8bit(Image.fromarray(image), computer)

# Crear la interfaz gráfica de Gradio
with gr.Blocks() as iface:
    gr.Markdown("# Emulador de Gráficos de Computadoras Retro")
    gr.Markdown("Sube una imagen, usa la cámara o emula el estilo gráfico de computadoras de los años 80 con 8 bits y pixelado. Elige entre varias computadoras retro.")
    
    with gr.Row():
        with gr.Column():
            input_image = gr.Image(source="upload", type="numpy", label="Subir Imagen")
            camera_input = gr.Image(source="webcam", type="numpy", label="Usar Cámara")
            capture_button = gr.Button("Capturar Imagen")
        
        computer_dropdown = gr.Dropdown(
            ["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"
        )
    
    output_image = gr.Image(type="numpy", label="Imagen convertida")
    
    convert_button = gr.Button("Convertir Imagen")
    
    # Conexiones
    convert_button.click(
        process_image,
        inputs=[input_image, computer_dropdown],
        outputs=output_image
    )
    
    capture_button.click(
        lambda x: x,
        inputs=camera_input,
        outputs=input_image
    )
    
    gr.Markdown("Hecho con ❤️ en MDQ por Peláez")

iface.launch()