Update app.py

app.py

@@ -5,6 +5,7 @@ from PIL import Image, ImageDraw
 import os
 from PIL import Image
 import numpy as np
+import io
 
 # Function to hash image data
 def hash_image(image):
@@ -60,24 +61,41 @@ def check_authenticity(image):
     return "Image is new or modified."
 
 
+
+
 def process_image(image, description):
+    # Determine the image format
+    image_format = image.format if image.format else 'PNG'  # Default to PNG if format cannot be detected
+
     hash_code1 = hash_image(image)
     qr_img = generate_qr_code(hash_code1)
     qr_img = qr_img.resize((100, 100))  # Resize QR code as needed
+
+    # Convert image to PIL Image if it's not one already (necessary if the image input comes as a numpy array)
+    if not isinstance(image, Image.Image):
+        image = Image.open(io.BytesIO(image))
+
     image_with_qr = embed_qr_code(image, qr_img)
     save_hash(hash_code1, description)
     hash_code2 = hash_image(image_with_qr)
     save_hash(hash_code2)
+
+    # Save the processed image in the original format
+    buffer = io.BytesIO()
+    image_with_qr.save(buffer, format=image_format)
+    buffer.seek(0)
+    image_with_qr = Image.open(buffer)  # Reopen the image from buffer to return it properly through Gradio
+
     return image_with_qr, "Image processed and hashes stored."
 
-# Update Gradio interface setup to handle the tuple output properly
+# Adjust the Gradio block as needed
 with gr.Blocks() as app:
     with gr.Tab("Upload and Process Image"):
         with gr.Row():
-            image_input = gr.Image(label="Upload Image")
+            image_input = gr.Image(label="Upload Image", type="pil")
             description_input = gr.Textbox(label="Description")
             submit_button = gr.Button("Process Image")
-        image_output = gr.Image(label="Processed Image")
+        image_output = gr.Image(label="Processed Image", type="pil")
         message_output = gr.Textbox(label="Status Message")
         submit_button.click(
             process_image, 
@@ -87,10 +105,11 @@ with gr.Blocks() as app:
 
     with gr.Tab("Check Image Authenticity"):
         with gr.Row():
-            image_check_input = gr.Image(label="Upload Image to Verify")
+            image_check_input = gr.Image(label="Upload Image to Verify", type="pil")
             check_button = gr.Button("Check Authenticity")
         authenticity_output = gr.Textbox(label="Result")
         check_button.click(check_authenticity, inputs=[image_check_input], outputs=authenticity_output)
 
 # Launch the application
 app.launch()
+