Update app.py

app.py

@@ -1,10 +1,6 @@
-import gradio as gr
+import streamlit as st
 import hashlib
-import qrcode
-from PIL import Image, ImageDraw
-import os
 from PIL import Image
-import numpy as np
 import io
 
 # Function to hash image data
@@ -13,27 +9,6 @@ def hash_image(image):
     image_hash = hashlib.sha256(image_data).hexdigest()
     return image_hash
 
-# Function to generate QR code
-def generate_qr_code(data):
-    qr = qrcode.QRCode(
-        version=1,
-        error_correction=qrcode.constants.ERROR_CORRECT_L,
-        box_size=10,
-        border=4,
-    )
-    qr.add_data(data)
-    qr.make(fit=True)
-    qr_img = qr.make_image(fill='black', back_color='white')
-    return qr_img
-
-
-def embed_qr_code(image, qr_img):
-    if isinstance(image, np.ndarray):
-        image = Image.fromarray(image.astype('uint8'))  # Convert NumPy array to PIL Image
-    image.paste(qr_img, (10, 10))  # Adjust position as needed
-    return image
-
-
 # Function to save hash to file
 def save_hash(hash_code, description=""):
     with open("hash.txt", "a") as file:
@@ -52,7 +27,6 @@ def check_authenticity(image):
                     if saved_hash == hash_code:
                         return f"Image is authentic. Description: {description}"
                 else:
-                    # Handle the case where there is no ':' or description
                     saved_hash = parts[0]
                     if saved_hash == hash_code:
                         return "Image is authentic, but no description provided."
@@ -60,56 +34,26 @@ def check_authenticity(image):
         return "Hash file not found. Please process an image first."
     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."
-
-# 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", type="pil")
-            description_input = gr.Textbox(label="Description")
-            submit_button = gr.Button("Process Image")
-        image_output = gr.Image(label="Processed Image", type="pil")
-        message_output = gr.Textbox(label="Status Message")
-        submit_button.click(
-            process_image, 
-            inputs=[image_input, description_input], 
-            outputs=[image_output, message_output]
-        )
-
-    with gr.Tab("Check Image Authenticity"):
-        with gr.Row():
-            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()
-
+    image_format = image.format if hasattr(image, 'format') else 'PNG'  # Default to PNG if format cannot be detected
+    hash_code = hash_image(image)
+    save_hash(hash_code, description)
+    return "Image hash computed and stored."
+
+st.title('Image Authenticity Checker')
+
+uploaded_file = st.file_uploader("Choose an image...", type=['png', 'jpg', 'jpeg'])
+description = st.text_input("Enter a description for the image:")
+
+if uploaded_file is not None:
+    image = Image.open(uploaded_file)
+    if st.button('Process Image'):
+        message = process_image(image, description)
+        st.success(message)
+
+if st.button('Check Authenticity'):
+    if uploaded_file is not None:
+        result = check_authenticity(image)
+        st.info(result)
+    else:
+        st.error("Please upload an image first.")