from PIL import Image, ImageTk
import gradio as gr

def convert_to_ascii(image, text_size):
    width, height = image.size
    aspect_ratio = height / width
    new_width = int(text_size)
    new_height = int(aspect_ratio * text_size * 0.5)

    resized_image = image.resize((new_width, new_height))
    grayscale_image = resized_image.convert('L')

    ascii_chars = '█@&%#*░+=-:,.\/|][}{)(´„‟‚‛‘ '  # add more characters if you want

    ascii_image = ''
    for y in range(new_height):
        for x in range(new_width):
            pixel_value = grayscale_image.getpixel((x, y))
            if pixel_value == 255:  # make sure to use the GOD DAM transparency
                ascii_image += ' '
            else:
                ascii_image += ascii_chars[int(pixel_value / 255 * (len(ascii_chars) - 1))]
        ascii_image += '\n'

    return ascii_image


def image_to_ascii(file, text_size):
    image = Image.open(file.name)
    return convert_to_ascii(image, int(text_size))


iface = gr.Interface(
    fn=image_to_ascii,
    inputs=["image", "number"],
    outputs="text",
    title="Image to ASCII Art",
    description="Convert an image to ASCII art."
)

iface.launch()