Update app.py

app.py

@@ -1,9 +1,15 @@
 import streamlit as st
-from PIL import Image
+from PIL import Image, ImageChops
 import io
-import cv2
 import numpy as np
 
+# Check if OpenCV is installed
+try:
+    import cv2
+    OPENCV_AVAILABLE = True
+except ImportError:
+    OPENCV_AVAILABLE = False
+
 # Function to perform image search (placeholder)
 def search_image(image):
     # Placeholder function for image search
@@ -20,8 +26,8 @@ def save_to_profile(results, user_profile):
         user_profile["search_history"] = []
     user_profile["search_history"].append(results)
 
-# Function to compare two images and calculate similarity percentage
-def compare_images(image1, image2):
+# Function to compare two images using OpenCV (if available)
+def compare_images_opencv(image1, image2):
     # Convert images to grayscale
     img1 = cv2.cvtColor(np.array(image1), cv2.COLOR_RGB2GRAY)
     img2 = cv2.cvtColor(np.array(image2), cv2.COLOR_RGB2GRAY)
@@ -44,6 +50,21 @@ def compare_images(image1, image2):
     similarity = (len(matches) / min(len(kp1), len(kp2))) * 100
     return similarity, matches
 
+# Function to compare two images using pixel-based method (fallback)
+def compare_images_pixel(image1, image2):
+    # Resize images to the same size
+    image1 = image1.resize((256, 256))
+    image2 = image2.resize((256, 256))
+
+    # Calculate the difference between the two images
+    diff = ImageChops.difference(image1, image2)
+
+    # Calculate the similarity percentage
+    diff_pixels = sum(diff.convert("L").point(lambda x: 255 if x > 0 else 0).getdata())
+    total_pixels = 256 * 256
+    similarity = 100 - (diff_pixels / total_pixels) * 100
+    return similarity
+
 # Streamlit app
 def main():
     # Sidebar with navigation
@@ -106,7 +127,10 @@ def main():
         if uploaded_file1 is not None and uploaded_file2 is not None:
             if st.button("Compare"):
                 with st.spinner("Comparing images..."):
-                    similarity, matches = compare_images(image1, image2)
+                    if OPENCV_AVAILABLE:
+                        similarity, _ = compare_images_opencv(image1, image2)
+                    else:
+                        similarity = compare_images_pixel(image1, image2)
                     st.success(f"Similarity: {similarity:.2f}%")
 
     elif page == "About Us":
@@ -146,4 +170,4 @@ def main():
                 st.sidebar.write(f"- {match['source']}: {match['url']}")
 
 if __name__ == "__main__":
-    main()
+    main()