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,9 +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, win_size=3, multichannel=True)
-    return similarity_score
+    
+    # Calculate similarity score based on SSIM
+    ssim_score = ssim(img1_resized, img2_resized, win_size=3, multichannel=True)
+    
+    # Additional score calculation after text removal (modify this as per your requirement)
+    text_removed_score = 0.75  # Placeholder value, replace with your own calculation
+    
+    # Combine the scores or choose the desired score
+    combined_score = (ssim_score + text_removed_score) / 2
+    
+    return combined_score
 
+# Rest of the code remains the same
 def image_similarity(img1, img2):
     img1 = img1.astype(np.uint8)
     img2 = img2.astype(np.uint8)