update streamlit, add dotted border with the most relevant item

document_qa/grobid_processors.py

@@ -554,18 +554,21 @@ class GrobidAggregationProcessor(GrobidQuantitiesProcessor, GrobidMaterialsProce
             return []
 
     @staticmethod
-    def box_to_dict(box, color=None, type=None):
+    def box_to_dict(box, color=None, type=None, border=None):
 
         if box is None or box == "" or len(box) < 5:
             return {}
 
         item = {"page": box[0], "x": box[1], "y": box[2], "width": box[3], "height": box[4]}
-        if color is not None:
+        if color:
             item['color'] = color
 
         if type:
             item['type'] = type
 
+        if border:
+            item['border'] = border
+
         return item
 
     @staticmethod

requirements.txt

@@ -27,6 +27,6 @@ typing-inspect==0.9.0
 typing_extensions==4.11.0
 pydantic==2.6.4
 sentence_transformers==2.6.1
-streamlit-pdf-viewer==0.0.21
+streamlit-pdf-viewer==0.0.22
 umap-learn
 plotly

streamlit_app.py

@@ -470,6 +470,9 @@ with right_column:
         for i, color in enumerate(gradients):
             for annotation in annotations[i]:
                 annotation['color'] = color
+                if i == 0:
+                    annotation['border'] = "dotted"
+
         st.session_state['annotations'] = [annotation for annotation_doc in annotations for annotation in
                                            annotation_doc]