Update app.py

app.py

@@ -15,31 +15,36 @@ def calculate_similarity(img1, img2):
 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 in image_list:
+    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):
+def image_similarity(target_image, image_folder):
     target_image = target_image.astype(np.uint8)
-    image = image.astype(np.uint8)
-    scores = compute_similarity(target_image, [image])
-    result = f"Image: {image.tolist()}\nScore: {scores[0]:.4f}"
-    return [result]
+    image_paths = gr.Interface.get_filepaths(image_folder)
+    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)
 
 # Prepare Gradio interface
 iface = gr.Interface(
     fn=image_similarity,
     inputs=[
         gr.inputs.Image(type="numpy", label="Target Image"),
-        gr.inputs.Image(type="numpy", label="Image")
+        gr.inputs.ImageFolder(label="Image Folder")
     ],
     outputs="text",
     title="Image Similarity Calculator",
-    description="Upload a target image and another image. Get the similarity score."
+    description="Upload a target image and select a folder containing images. Get similarity scores."
 )
 
 # Launch the interface
 iface.launch()
+