Update app.py

.gitignore

@@ -0,0 +1 @@
+venv

app.py

@@ -5,8 +5,10 @@ import time
 import json
 import base64
 import os
-from PIL import Image
 from io import BytesIO
+import PIL
+from PIL.ExifTags import TAGS
+
 
 class Prodia:
     def __init__(self, api_key, base=None):
@@ -98,6 +100,13 @@ def flip_text(prompt, negative_prompt, model, steps, sampler, cfg_scale, width,
 
     return job["imageUrl"]
 
+def get_exif_data(image):
+    print(image)
+    exif = image.getexif()
+    print(exif)
+
+    return image.info
+
 css = """
 #generate {
     height: 100%;
@@ -170,5 +179,16 @@ with gr.Blocks(css=css) as demo:
 
         text_button.click(flip_text, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed], outputs=image_output)
     
-demo.queue(concurrency_count=16)
+    with gr.Tab("PNG Info"):
+        with gr.Row():
+            with gr.Column():
+                image_input = gr.Image(type="pil")
+            with gr.Column():
+                exif_button = gr.Button("Get input data")
+                exif_output = gr.Textbox(label="EXIF Data")
+
+                exif_button.click(get_exif_data, inputs=[image_input], outputs=exif_output)
+
+
+demo.queue(concurrency_count=24)
 demo.launch()

requirements.txt

@@ -1,3 +1,5 @@
 numpy
 gradio
-requests
+requests
+pillow
+pyexif