Update app.py

app.py

@@ -89,22 +89,19 @@ async def search(query: str, k: int):
 
     results = []
     for idx in top_k_indices:
-        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}"})
+        results.append({"image": images[idx], "page": f"Page {idx}"})
 
     # Generate PDF
     pdf_buffer = BytesIO()
     c = canvas.Canvas(pdf_buffer, pagesize=letter)
     width, height = letter
-
     for result in results:
-        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()
+        img = result["image"]
+        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_file:
+            img.save(temp_file.name, format='PNG')
+            temp_file.seek(0)
+            c.drawImage(temp_file.name, 0, 0, width, height)
+            c.showPage()
 
     c.save()
     pdf_buffer.seek(0)