image-to-ascii / app.py
peasoup's picture
Update app.py
2cae52b
from PIL import Image
import gradio as gr
import numpy as np
def convert_to_ascii(image, text_size):
if isinstance(image, str):
gimage = Image.open(image)
elif isinstance(image, np.ndarray):
gimage = Image.fromarray(image)
else:
raise ValueError("Unsupported input type. Please provide a file path or a NumPy array.")
width, height = gimage.size
# Calculate a new aspect ratio based on text_size and original image dimensions
aspect_ratio = height / (width * 2) # Adjusted aspect ratio
new_width = int(text_size)
new_height = int(aspect_ratio * text_size * 0.5)
resized_image = gimage.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:
ascii_image += 'β€Ž '
else:
ascii_image += ascii_chars[int(pixel_value / 255 * (len(ascii_chars) - 1))]
ascii_image += '\n'
return ascii_image
# Create an Interface with a title
iface = gr.Interface(
fn=convert_to_ascii, # Function to run
inputs=["image","number"], # Input component (in this case, an image)
outputs=gr.Textbox(show_copy_button=True), # Output component (in this case, text)
title="image-to-ascii (by peasoup)",
description="so i made this for mobile people and mac users (you're welcome!!), look at this omg: https://github.com/itspeasoup/image-to-ascii"
)
# Launch the interface
iface.launch()