Update app.py

app.py

@@ -24,19 +24,24 @@ def compute_similarity(target_image, image_list):
 
 
 # Function to handle the Gradio interface
-def image_similarity(target_image, image):
+# Function to handle the Gradio interface
+def image_similarity(target_image, image_list):
     target_image = target_image.astype(np.uint8)
-    image = image.astype(np.uint8)
-    scores = compute_similarity(target_image, [image])
-    result = (format_image(image), f"Score: {scores[0]:.4f}")
-    return [result]
+    image_list = [image.astype(np.uint8) for image in image_list]
+    scores = compute_similarity(target_image, image_list)
+    results = []
+    for image, score in zip(image_list, scores):
+        formatted_image = format_image(image)
+        result = f"Image: {formatted_image}\nScore: {score:.4f}\n"
+        results.append(result)
+    return "".join(results)
 
 # Function to format the image representation as a string
 def format_image(image):
     formatted_image = ""
     for row in image:
-        formatted_row = ", ".join([f"({', '.join(map(str, pixel))})" for pixel in row])
-        formatted_image += f"[{formatted_row}], "
+        formatted_row = ", ".join([f"[{', '.join(map(str, pixel))}]" for pixel in row])
+        formatted_image += f"[{formatted_row}],\n"
     return formatted_image[:-2]
 
 # Prepare Gradio interface
@@ -44,11 +49,11 @@ iface = gr.Interface(
     fn=image_similarity,
     inputs=[
         gr.inputs.Image(type="numpy", label="Target Image"),
-        gr.inputs.Image(type="numpy", label="Image")
+        gr.inputs.Image(type="numpy", label="Image List", multiple=True)
     ],
     outputs="text",
     title="Image Similarity Calculator",
-    description="Upload a target image and another image. Get the similarity score."
+    description="Upload a target image and a list of images. Get similarity scores."
 )
 
 # Launch the interface
@@ -60,3 +65,4 @@ iface.launch()
 
 
 
+