Update app.py

app.py

@@ -1,19 +1,25 @@
-import numpy as np
-from skimage.metrics import structural_similarity as ssim
 from PIL import Image
-import gradio as gr
 from difflib import SequenceMatcher
+from skimage.metrics import structural_similarity as ssim
+import gradio as gr
+import pytesseract
 
 # Function to calculate SSIM between two images
 def calculate_ssim(img1, img2):
     img1_gray = img1.convert("L")
     img2_gray = img2.convert("L")
-    return ssim(np.array(img1_gray), np.array(img2_gray))
+    return ssim(img1_gray, img2_gray)
+
+# Function to extract text from an image using OCR
+def extract_text(image):
+    image = Image.open(image)
+    text = pytesseract.image_to_string(image).lower()
+    return text
 
 # Function to compare trademarks based on text similarity
 def compare_text(trademark1, trademark2):
-    text1 = pytesseract.image_to_string(trademark1).lower()
-    text2 = pytesseract.image_to_string(trademark2).lower()
+    text1 = extract_text(trademark1)
+    text2 = extract_text(trademark2)
     similarity_ratio = SequenceMatcher(None, text1, text2).ratio()
     return similarity_ratio
 
@@ -23,14 +29,8 @@ def compare_colors(trademark1, trademark2):
     trademark2 = trademark2.convert("RGB")
     colors1 = trademark1.getcolors(trademark1.size[0] * trademark1.size[1])
     colors2 = trademark2.getcolors(trademark2.size[0] * trademark2.size[1])
-    color_vector1 = np.zeros(3)
-    color_vector2 = np.zeros(3)
-    for count, color in colors1:
-        color_vector1 += np.array(color) * count
-    for count, color in colors2:
-        color_vector2 += np.array(color) * count
-    color_vector1 /= trademark1.size[0] * trademark1.size[1]
-    color_vector2 /= trademark2.size[0] * trademark2.size[1]
+    color_vector1 = sum([(count * np.array(color)) for count, color in colors1]) / (trademark1.size[0] * trademark1.size[1])
+    color_vector2 = sum([(count * np.array(color)) for count, color in colors2]) / (trademark2.size[0] * trademark2.size[1])
     color_similarity = 1 - np.linalg.norm(color_vector1 - color_vector2)
     return color_similarity
 
@@ -57,8 +57,8 @@ def prevent_trademark_conflict(trademark1, trademark2):
 iface = gr.Interface(
     fn=prevent_trademark_conflict,
     inputs=[
-        gr.inputs.Image(type="file", label="Trademark Image 1"),
-        gr.inputs.Image(type="file", label="Trademark Image 2")
+        gr.inputs.ImagePicker(label="Trademark Image 1"),
+        gr.inputs.ImagePicker(label="Trademark Image 2")
     ],
     outputs="text",
     title="Trademark Conflict Prevention",