Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+# import gradio as gr
 import numpy as np
 import tensorflow as tf
 import logging
@@ -82,6 +82,7 @@ def extract_clip_features(image_path, model, preprocess):
 def compare_images(image1, image2, method):
     similarity = None
     start_time = time.time()
+    
     if method == 'pHash':
         img1 = Image.open(image1)
         img2 = Image.open(image2)
@@ -89,21 +90,26 @@ def compare_images(image1, image2, method):
         hash2 = phashstr(img2)
         distance = hamming_distance(hash1, hash2)
         similarity = hamming_to_similarity(distance, len(hash1) * 4)
+        
     elif method == 'ResNet50':
         features1 = extract_features(image1, resnet_model, resnet_preprocess)
         features2 = extract_features(image2, resnet_model, resnet_preprocess)
         similarity = cosine_similarity(features1, features2)
+        
     elif method == 'VGG16':
         features1 = extract_features(image1, vgg_model, vgg_preprocess)
         features2 = extract_features(image2, vgg_model, vgg_preprocess)
         similarity = cosine_similarity(features1, features2)
+        
     elif method == 'CLIP':
         features1 = extract_clip_features(image1, clip_model, preprocess_clip)
         features2 = extract_clip_features(image2, clip_model, preprocess_clip)
         similarity = cosine_similarity(features1, features2)
-    
+
     logging.info(f"AI based Supporting Documents comparison using {method} completed in {time.time() - start_time:.4f} seconds")
-    return similarity
+    
+    # Return similarity with HTML formatting for bold and colorful text
+    return f"<span style='font-weight:bold; color:#4CAF50;'>Similarity Score: {similarity:.2f}%</span>"
 
 # Gradio interface
 demo = gr.Interface(
@@ -113,9 +119,14 @@ demo = gr.Interface(
         gr.Image(type="filepath", label="Upload Second Image"),
         gr.Radio(["pHash", "ResNet50", "VGG16", "CLIP"], label="Select Comparison Method")
     ],
-    outputs=gr.Textbox(label="Similarity"),
-    title="AI based Customs Supporting Documents comparison",
-    description="Upload two images of Suppporting documents and select the comparison method.",
+    outputs=gr.HTML(label="Similarity"),  # Use HTML for bold and colorful text
+    title="AI Based Customs Supporting Documents Comparison",
+    description=(
+        "Upload two images of supporting documents and select the comparison method.\n"
+        "Fraud documents like invoices are used by custom brokers with the same templates. "
+        "This tool helps identify similar document templates used in two different consignments.\n"
+        "Developed by NCTC."
+    ),
     examples=[
         ["Snipaste_2024-05-31_16-18-31.jpg", "Snipaste_2024-05-31_16-18-52.jpg"],
         ["example1.png", "example2.png"]
@@ -123,3 +134,4 @@ demo = gr.Interface(
 )
 
 demo.launch()
+