Update app.py

app.py

@@ -3,6 +3,7 @@ from skimage.metrics import structural_similarity as ssim
 import gradio as gr
 import cv2
 
+# Function to calculate SSIM between two images
 def calculate_similarity(img1, img2):
     if len(img1.shape) == 2:
         img1 = cv2.cvtColor(img1, cv2.COLOR_GRAY2RGB)
@@ -12,14 +13,19 @@ def calculate_similarity(img1, img2):
     max_width = max(img1.shape[1], img2.shape[1])
     img1_resized = cv2.resize(img1, (max_width, max_height))
     img2_resized = cv2.resize(img2, (max_width, max_height))
-    similarity_score = ssim(img1_resized, img2_resized, multichannel=True)
-    return similarity_score
+    
+    # Calculate similarity score based on SSIM
+    ssim_score = ssim(img1_resized, img2_resized, win_size=3, multichannel=True)
+    
+    return ssim_score
 
+# Rest of the code remains the same
 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}"
+    ssim_score = calculate_similarity(img1, img2)
+    additional_score = 0.75  # Placeholder value, replace with your own calculation
+    result = f"SSIM Score: {ssim_score:.4f}\nAdditional Score: {additional_score:.4f}"
     return result
 
 iface = gr.Interface(