Update app.py

app.py

@@ -183,6 +183,8 @@ def search(query: str, k: int):
     k = max(1, min(int(k), len(images)))
     device = _ensure_model_device()
 
+    print(query)
+
     # Encode query
     qs = []
     with torch.no_grad():
@@ -194,6 +196,8 @@ def search(query: str, k: int):
     scores = processor.score(qs, ds, device=device)
     top_k_indices = scores[0].topk(k).indices.tolist()
 
+    print(top_k_indices)
+
     # Build gallery results with 1-based page numbering
     results = []
     for idx in top_k_indices:
@@ -202,6 +206,7 @@ def search(query: str, k: int):
 
     # Generate grounded response
     ai_response = query_gpt(query, results)
+    print(ai_response)
     return ai_response