Create app.py

app.py

@@ -0,0 +1,91 @@
+import streamlit as st
+import hashlib
+import qrcode
+from PIL import Image, ImageDraw
+import numpy as np
+import io
+
+# Function to hash image data
+def hash_image(image):
+    image_data = image.tobytes()
+    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:
+        file.write(f"{hash_code}: {description}\n")
+
+# Function to check image authenticity
+def check_authenticity(image):
+    hash_code = hash_image(image)
+    try:
+        with open("hash.txt", "r") as file:
+            hashes = file.readlines()
+            for line in hashes:
+                parts = line.strip().split(': ', 1)
+                if len(parts) == 2:
+                    saved_hash, description = parts
+                    if saved_hash == hash_code:
+                        return f"Image is authentic. Description: {description}"
+                else:
+                    saved_hash = parts[0]
+                    if saved_hash == hash_code:
+                        return "Image is authentic, but no description provided."
+    except FileNotFoundError:
+        return "Hash file not found. Please process an image first."
+    return "Image is new or modified."
+
+def process_image(image, description):
+    image_format = image.format if hasattr(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
+    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)
+
+    buffer = io.BytesIO()
+    image_with_qr.save(buffer, format=image_format)
+    buffer.seek(0)
+    return buffer.getvalue(), "Image processed and hashes stored."
+
+st.title('Image Authenticity and QR Code Embedding')
+
+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'):
+        processed_image_data, message = process_image(image, description)
+        st.image(processed_image_data, caption='Processed Image')
+        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.")