Update app.py

app.py

@@ -31,16 +31,18 @@ def calculate_similarity(img1, img2):
 def image_similarity(img1, img2):
     img1 = img1.astype(np.uint8)
     img2 = img2.astype(np.uint8)
-    ssim_score = calculate_similarity(img1, img2)
     
-    # Remove text from images (placeholder implementation)
+    # Remove text from images
     img1_processed = remove_text(img1)
     img2_processed = remove_text(img2)
     
-    # Calculate similarity score after text removal (placeholder implementation)
-    additional_score = 0.75  # Placeholder value, replace with your own calculation
+    # Calculate similarity score based on SSIM
+    ssim_score = calculate_similarity(img1, img2)
+    
+    # Calculate similarity score after text removal
+    additional_score = calculate_similarity(img1_processed, img2_processed)
     
-    result = f"SSIM Score: {ssim_score:.4f}\nAdditional Score: {additional_score:.4f}"
+    result = f"SSIM Score: {ssim_score:.4f}\nAdditional Score (after text removal): {additional_score:.4f}"
     return result
 
 iface = gr.Interface(