Update app.py

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 from PIL import Image
 import io
+import cv2
+import numpy as np
 
 # Function to perform image search (placeholder)
 def search_image(image):
@@ -18,11 +20,35 @@ 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):
+    # Convert images to grayscale
+    img1 = cv2.cvtColor(np.array(image1), cv2.COLOR_RGB2GRAY)
+    img2 = cv2.cvtColor(np.array(image2), cv2.COLOR_RGB2GRAY)
+
+    # Initialize ORB detector
+    orb = cv2.ORB_create()
+
+    # Detect keypoints and descriptors
+    kp1, des1 = orb.detectAndCompute(img1, None)
+    kp2, des2 = orb.detectAndCompute(img2, None)
+
+    # Use BFMatcher to match descriptors
+    bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True)
+    matches = bf.match(des1, des2)
+
+    # Sort matches by distance (lower is better)
+    matches = sorted(matches, key=lambda x: x.distance)
+
+    # Calculate similarity percentage
+    similarity = (len(matches) / min(len(kp1), len(kp2))) * 100
+    return similarity, matches
+
 # Streamlit app
 def main():
     # Sidebar with navigation
     st.sidebar.title("Navigation")
-    page = st.sidebar.radio("Go to", ["Home", "About Us", "Help"])
+    page = st.sidebar.radio("Go to", ["Home", "Compare Photos", "About Us", "Help"])
 
     if page == "Home":
         st.title("Image Cross-Check App")
@@ -60,6 +86,29 @@ def main():
                 else:
                     st.warning("No matches found.")
 
+    elif page == "Compare Photos":
+        st.title("Compare Photos")
+        st.write("Upload two images to compare their similarity.")
+
+        # Upload first image
+        uploaded_file1 = st.file_uploader("Choose the first image...", type=["jpg", "jpeg", "png"])
+        if uploaded_file1 is not None:
+            image1 = Image.open(uploaded_file1)
+            st.image(image1, caption='First Image.', use_container_width=True)
+
+        # Upload second image
+        uploaded_file2 = st.file_uploader("Choose the second image...", type=["jpg", "jpeg", "png"])
+        if uploaded_file2 is not None:
+            image2 = Image.open(uploaded_file2)
+            st.image(image2, caption='Second Image.', use_container_width=True)
+
+        # Compare images
+        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)
+                    st.success(f"Similarity: {similarity:.2f}%")
+
     elif page == "About Us":
         st.title("About Us")
         st.write("""
@@ -78,6 +127,9 @@ def main():
         3. **View Results**: If matches are found, they will be displayed on the screen.
         4. **Save Results**: The results are automatically saved to your profile for future reference.
 
+        ### Compare Photos:
+        - Upload two images and click "Compare" to calculate their similarity percentage.
+
         ### Troubleshooting:
         - If the app is not working, ensure you have a stable internet connection.
         - Supported image formats are JPG, JPEG, and PNG.