Update app.py

app.py

@@ -89,19 +89,21 @@ async def search(query: str, k: int):
 
     results = []
     for idx in top_k_indices:
-        results.append({"image": images[idx], "page": f"Page {idx}"})
+        img_byte_arr = BytesIO()
+        images[idx].save(img_byte_arr, format='PNG')
+        img_base64 = base64.b64encode(img_byte_arr.getvalue()).decode('utf-8')
+        results.append({"image": img_base64, "page": f"Page {idx}"})
 
     # Generate PDF
     pdf_buffer = BytesIO()
     c = canvas.Canvas(pdf_buffer, pagesize=letter)
     width, height = letter
+
     for result in results:
-        img = result["image"]
-        img_byte_arr = BytesIO()
-        img.save(img_byte_arr, format='PNG')
-        img_byte_arr = img_byte_arr.getvalue()
-        img_base64 = base64.b64encode(img_byte_arr).decode('utf-8')
-        c.drawImage(BytesIO(base64.b64decode(img_base64)), 0, 0, width, height)
+        img_base64 = result["image"]
+        img_data = base64.b64decode(img_base64)
+        img_byte_arr = BytesIO(img_data)
+        c.drawImage(img_byte_arr, 0, 0, width, height)
         c.showPage()
 
     c.save()