Update app.py

app.py

@@ -1,7 +1,10 @@
-from PIL import Image, ImageTk
 import gradio as gr
+from PIL import Image
+import numpy as np
 
 def convert_to_ascii(image, text_size):
+    image = Image.fromarray(image.astype('uint8'))
+    
     width, height = image.size
     aspect_ratio = height / width
     new_width = int(text_size)
@@ -10,13 +13,13 @@ def convert_to_ascii(image, text_size):
     resized_image = image.resize((new_width, new_height))
     grayscale_image = resized_image.convert('L')
 
-    ascii_chars = '█@&%#*░+=-:,.\/|][}{)(´„‟‚‛‘ '  # add more characters if you want
+    ascii_chars = '█@&%#*░+=-:,.\/|][}{)(´„‟‚‛‘ '
 
     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
+            if pixel_value == 255:
                 ascii_image += ' '
             else:
                 ascii_image += ascii_chars[int(pixel_value / 255 * (len(ascii_chars) - 1))]
@@ -24,19 +27,10 @@ def convert_to_ascii(image, text_size):
 
     return ascii_image
 
+def image_to_ascii(image, text_size):
+    ascii_text = convert_to_ascii(image, text_size)
+    return ascii_text
 
-def image_to_ascii(file, text_size):
-    image = Image.open(file)  # Modify this line to handle the file object correctly
-    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 = gr.Interface(fn=image_to_ascii, inputs=["image", "number"], outputs="text", title="Image to ASCII")
 iface.launch()
-    
+