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()