Update app.py

app.py

@@ -65,11 +65,8 @@ class FastPDFSearch:
                 img = Image.open(io.BytesIO(img_data))
                 if img.mode != 'RGB':
                     img = img.convert('RGB')
-                img_byte_arr = io.BytesIO()
-                img.save(img_byte_arr, format='PNG')
-                img_byte_arr.seek(0)
                 doc.close()
-                return img_byte_arr
+                return img
             doc.close()
             return None
         except Exception as e:
@@ -85,9 +82,9 @@ class FastPDFSearch:
         filename = result.metadata["filename"]
         page_number = result.metadata["page_number"]
         text = result.page_content
-        img_bytes = self.get_pdf_page_image(self.pdf_files[filename], page_number)
-        if img_bytes:
-            return text, img_bytes, f"{filename} - Page {page_number}"
+        img = self.get_pdf_page_image(self.pdf_files[filename], page_number)
+        if img:
+            return text, img, f"{filename} - Page {page_number}"
         else:
             return text, None, f"{filename} - Page {page_number}"
 
@@ -95,9 +92,9 @@ class FastPDFSearch:
 searcher = FastPDFSearch()
 
 def gradio_search(query):
-    text, img_bytes, label = searcher.search(query)
-    if img_bytes:
-        return text, [(img_bytes, label)]
+    text, img, label = searcher.search(query)
+    if img:
+        return text, [(img, label)]
     else:
         return text, []