Update app.py

app.py

@@ -319,11 +319,13 @@ async def search_by_image(file: UploadFile = File(...), k: int = 5):
     exact_row = find_exact_dataset_match(feats)
 
     if exact_row is not None:
+        is_dataset_image = True
         # Use ground-truth caption from CSV
         query_caption = str(exact_row.get("caption", "")).strip()
         query_caption = clean_caption(query_caption)
         query_id = str(exact_row["ID"])
     else:
+        is_dataset_image = False
         # Not a known dataset image -> use BLIP1 model
         query_caption = generate_caption_with_blip(image)
         query_id = None
@@ -342,5 +344,6 @@ async def search_by_image(file: UploadFile = File(...), k: int = 5):
             "modality": modality,
             "scores": scores,
             "results": results,
+            "is_dataset_image": is_dataset_image,
         }
     )