Update app.py

app.py

@@ -2,7 +2,6 @@ import numpy as np
 from skimage.metrics import structural_similarity as ssim
 import gradio as gr
 import cv2
-import os
 
 # Function to calculate SSIM between two images
 def calculate_similarity(img1, img2):
@@ -12,38 +11,24 @@ def calculate_similarity(img1, img2):
         img2 = cv2.cvtColor(img2, cv2.COLOR_GRAY2RGB)
     return ssim(img1, img2, win_size=3)
 
-# Function to compute similarity scores for all images
-def compute_similarity(target_image, image_list):
-    scores = []
-    target_image_resized = cv2.resize(target_image, (target_image.shape[1], target_image.shape[0]))
-    for image_path in image_list:
-        image = cv2.imread(image_path)
-        image_resized = cv2.resize(image, (target_image.shape[1], target_image.shape[0]))
-        similarity_score = calculate_similarity(target_image_resized, image_resized)
-        scores.append(similarity_score)
-    return scores
-
 # Function to handle the Gradio interface
-def image_similarity(target_image, image_folder):
-    target_image = target_image.astype(np.uint8)
-    image_paths = [os.path.join(image_folder, file) for file in os.listdir(image_folder) if os.path.isfile(os.path.join(image_folder, file))]
-    scores = compute_similarity(target_image, image_paths)
-    results = []
-    for image_path, score in zip(image_paths, scores):
-        result = f"Image: {image_path}\nScore: {score:.4f}\n"
-        results.append(result)
-    return "".join(results)
+def image_similarity(img1, img2):
+    img1 = img1.astype(np.uint8)
+    img2 = img2.astype(np.uint8)
+    similarity_score = calculate_similarity(img1, img2)
+    result = f"Similarity Score: {similarity_score:.4f}"
+    return result
 
 # Prepare Gradio interface
 iface = gr.Interface(
     fn=image_similarity,
     inputs=[
-        gr.inputs.Image(type="numpy", label="Target Image"),
-        gr.inputs.Textbox(label="Image Folder", lines=1)
+        gr.inputs.Image(type="numpy", label="Image 1"),
+        gr.inputs.Image(type="numpy", label="Image 2")
     ],
     outputs="text",
     title="Image Similarity Calculator",
-    description="Upload a target image and specify the path to the folder containing images. Get similarity scores."
+    description="Upload two images to compute their similarity."
 )
 
 # Launch the interface