colpali fix

RAG/colpali.py

@@ -4,6 +4,7 @@ import random
 import pprint
 import os
 from io import BytesIO
+import glob
 from pathlib import Path
 from typing import Optional, cast
 import numpy as np
@@ -154,6 +155,23 @@ def generate_ans(top_result,query):
     return content_text
 
 def img_highlight(img,batch_queries,query_tokens):
+    
+    img_name = os.path.basename(img)  # e.g., "my_image.png"
+
+    # Construct the search pattern
+    search_pattern = f"/home/user/app/similarity_maps/similarity_map_{img_name}_token_*"
+
+    # Search for matching files
+    matching_files = glob.glob(search_pattern)
+
+    # Check if any match exists
+    map_images = []
+    if matching_files:
+        print("✅ Matching similarity map exists:")
+        for file_path in matching_files:
+            print(f" - {file_path}")
+            map_images.append({'file':file_path})
+        return map_images
     # Reference from : https://github.com/tonywu71/colpali-cookbooks/blob/main/examples/gen_colpali_similarity_maps.ipynb
     with open(img, "rb") as f:
         img_b64 = base64.b64encode(f.read()).decode("utf-8")