Update app.py

app.py

@@ -1,16 +1,14 @@
 import numpy as np
-import gradio as gr
-from PIL import Image
 from skimage.metrics import structural_similarity as ssim
-import cv2
-import pytesseract
+from PIL import Image
+import gradio as gr
 from difflib import SequenceMatcher
 
 # Function to calculate SSIM between two images
 def calculate_ssim(img1, img2):
-    img1_gray = cv2.cvtColor(img1, cv2.COLOR_RGB2GRAY)
-    img2_gray = cv2.cvtColor(img2, cv2.COLOR_RGB2GRAY)
-    return ssim(img1_gray, img2_gray)
+    img1_gray = img1.convert("L")
+    img2_gray = img2.convert("L")
+    return ssim(np.array(img1_gray), np.array(img2_gray))
 
 # Function to compare trademarks based on text similarity
 def compare_text(trademark1, trademark2):
@@ -38,12 +36,7 @@ def compare_colors(trademark1, trademark2):
 
 # Function to compare trademarks based on multiple aspects
 def compare_trademarks(trademark1, trademark2):
-    # Resize images to a consistent size
-    size = (300, 300)
-    trademark1_resized = trademark1.resize(size)
-    trademark2_resized = trademark2.resize(size)
-
-    ssim_score = calculate_ssim(np.array(trademark1_resized), np.array(trademark2_resized))
+    ssim_score = calculate_ssim(trademark1, trademark2)
     text_similarity = compare_text(trademark1, trademark2)
     color_similarity = compare_colors(trademark1, trademark2)
 
@@ -64,13 +57,12 @@ def prevent_trademark_conflict(trademark1, trademark2):
 iface = gr.Interface(
     fn=prevent_trademark_conflict,
     inputs=[
-        gr.inputs.Image(type="pil", label="Trademark Image 1"),
-        gr.inputs.Image(type="pil", label="Trademark Image 2")
+        gr.inputs.Image(type="file", label="Trademark Image 1"),
+        gr.inputs.Image(type="file", label="Trademark Image 2")
     ],
     outputs="text",
     title="Trademark Conflict Prevention",
-  
-    description="Upload two trademark images to prevent conflict."
+    description="Upload two trademark images. Get the trademark similarity score."
 )
 
 iface.launch()