Update app.py

app.py

@@ -165,7 +165,13 @@ def search(query: str, ds, images, k, api_key):
     # Get the top k images
     results = []
     for idx in top_k_indices:
-        results.append((images[idx], f"Page {idx}"))
+        img = images[idx]
+        # buffered = BytesIO()
+        # img.save(buffered, format="JPEG")
+        # img_processed = Image.open(BytesIO(buffered.getvalue()))
+        img_copy = img.copy()
+        results.append((img_copy, f"Page {idx}"))
+        # results.append((images[idx], f"Page {idx}"))
 
     # Generate response from Gemini
     ai_response = query_gemini(query, results, api_key)