Update app.py

app.py

@@ -1,4 +1,4 @@
-import io
+from io import BytesIO
 import os
 import sys
 
@@ -92,12 +92,12 @@ async def search(query: str, k: int):
         results.append({"image": images[idx], "page": f"Page {idx}"})
 
     # Generate PDF
-    pdf_buffer = io.BytesIO()
+    pdf_buffer = BytesIO()
     c = canvas.Canvas(pdf_buffer, pagesize=letter)
     width, height = letter
     for result in results:
         img = result["image"]
-        img_byte_arr = io.BytesIO()
+        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')