Update app.py

app.py

@@ -11,7 +11,6 @@ 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 = []
@@ -22,47 +21,25 @@ def compute_similarity(target_image, image_list):
         scores.append(similarity_score)
     return scores
 
-
-# Function to handle the Gradio interface
 # Function to handle the Gradio interface
-def image_similarity(target_image, image_list):
+def image_similarity(target_image, image):
     target_image = target_image.astype(np.uint8)
-    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}],\n"
-    return formatted_image[:-2]
+    image = image.astype(np.uint8)
+    scores = compute_similarity(target_image, [image])
+    result = f"Image: {image.tolist()}\nScore: {scores[0]:.4f}"
+    return [result]
 
 # 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 List", multiple=True)
+        gr.inputs.Image(type="numpy", label="Image")
     ],
     outputs="text",
     title="Image Similarity Calculator",
-    description="Upload a target image and a list of images. Get similarity scores."
+    description="Upload a target image and another image. Get the similarity score."
 )
 
 # Launch the interface
 iface.launch()
-
-
-
-
-
-
-
-