pre_message_style

main.py

@@ -22,7 +22,7 @@ cache = {
     'original_document_image_1': None,
     'original_document_image_2': None
 }
-pre_message_style = 'overflow:auto;border:2px solid pink;padding:4px;border-radius:4px;'
+pre_message_style = 'border:2px solid pink;padding:4px;border-radius:4px;font-size: 16px;font-weight: 700;background-image: linear-gradient(to bottom right, #e0e619, #ffffff, #FF77CC, rgb(255, 122, 89));'
 visualize_bboxes_on_image_kwargs = {
     'label_text_color': 'white',
     'label_fill_color': 'black',
@@ -45,7 +45,8 @@ def similarity_fn(model: lp.Detectron2LayoutModel, document_image_1: Image.Image
     show_vectors_type = False
     try:
         if document_image_1 is None or document_image_2 is None:
-            message = f'<pre style="{pre_message_style}">Please load both the documents to compare.<pre>'
+            message = 'Please load both the documents to compare.'
+            gr.Info(message)
         else:
             input_document_image_1_hash = str(average_hash(document_image_1))
             input_document_image_2_hash = str(average_hash(document_image_2))
@@ -54,6 +55,7 @@ def similarity_fn(model: lp.Detectron2LayoutModel, document_image_1: Image.Image
                 document_image_1_features = cache['document_image_1_features']
                 document_image_1 = cache['original_document_image_1']
             else:
+                gr.Info('Generating features for document 1')
                 document_image_1_features = get_features(
                     document_image_1, model, label_names)
                 cache['document_image_1_features'] = document_image_1_features
@@ -63,11 +65,13 @@ def similarity_fn(model: lp.Detectron2LayoutModel, document_image_1: Image.Image
                 document_image_2_features = cache['document_image_2_features']
                 document_image_2 = cache['original_document_image_2']
             else:
+                gr.Info('Generating features for document 2')
                 document_image_2_features = get_features(
                     document_image_2, model, label_names)
                 cache['document_image_2_features'] = document_image_2_features
                 cache['original_document_image_2'] = document_image_2
 
+            gr.Info('Calculating similarity')
             [[similarity]] = cosine_similarity(
                 [
                     cache['document_image_1_features'][vectors_type]
@@ -75,7 +79,9 @@ def similarity_fn(model: lp.Detectron2LayoutModel, document_image_1: Image.Image
                 [
                     cache['document_image_2_features'][vectors_type]
                 ])
-            message = f'<pre style="{pre_message_style}">Similarity between the two documents is: {round(similarity, 4)}<pre>'
+            message = f'Similarity between the two documents is: {round(similarity, 4)}'
+            gr.Info(message)
+            gr.Info('Visualizing the bounding boxes for the predicted layout elements on the documents.')
             document_image_1 = visualize_bboxes_on_image(
                 image=document_image_1,
                 bboxes=cache['document_image_1_features'][annotations['predicted_bboxes']],
@@ -106,9 +112,10 @@ def similarity_fn(model: lp.Detectron2LayoutModel, document_image_1: Image.Image
 
             show_vectors_type = True
     except Exception as e:
-        message = f'<pre style="{pre_message_style}">{traceback.format_exc()}<pre>'
+        message = f'<pre style="overflow:auto;">{traceback.format_exc()}</pre>'
+        gr.Info(message)
     return [
-        gr.HTML(message, visible=True),
+        gr.HTML(f'<div style="{pre_message_style}">{message}</div>', visible=True),
         document_image_1,
         document_image_2,
         gr.Dropdown(visible=show_vectors_type)