Update app.py

app.py

@@ -2095,6 +2095,18 @@ def similarity_matching(sprites_data: str, project_folder: str) -> str:
     # # logger.info(f"🎉 Final project saved: {project_json_path}")
     # return project_json_path
 
+def convert_bytes_to_image(pdf_bytes: bytes, dpi: int):
+    images = convert_from_bytes(pdf_bytes, dpi=dpi, poppler_path=poppler_path)
+    # Save each page to an in-memory BytesIO and return a list of BytesIOs
+    buffers = []
+    for img in images:
+        buf = BytesIO()
+        img.save(buf, format="PNG")
+        buf.seek(0)
+        buffers.append(buf)
+    return buffers
+
+
 def delay_for_tpm_node(state: GameState):
     logger.info("--- Running DelayForTPMNode ---")
     time.sleep(60)  # Adjust the delay as needed
@@ -2294,7 +2306,6 @@ def process_pdf():
         pdf_stream = io.BytesIO(pdf_bytes)
         logger.info(f"Saved uploaded PDF to: {pdf_stream}")
 
-
         # pdf= save_pdf_to_generated_dir(saved_pdf_path, project_id)
         pdf= save_pdf_to_generated_dir(pdf_stream, project_id)
         # logger.info(f"Created project folder: {project_folder}")
@@ -2326,15 +2337,18 @@ def process_pdf():
         with open(project_output, 'r') as f:
             project_skeleton = json.load(f)
             
-        images = convert_from_path(pdf_stream, dpi=300)  
-        print(type)
-        page = images[0]
-        # img_base64 = base64.b64encode(images).decode("utf-8") 
-        buf = BytesIO()
-        page.save(buf, format="PNG")
-        img_bytes = buf.getvalue()
-        img_b64 = base64.b64encode(img_bytes).decode("utf-8")
+        # images = convert_from_path(pdf_stream, dpi=300)  
+        # print(type)
+        # page = images[0]
+        # # img_base64 = base64.b64encode(images).decode("utf-8") 
+        # buf = BytesIO()
+        # page.save(buf, format="PNG")
+        # img_bytes = buf.getvalue()
+        # img_b64 = base64.b64encode(img_bytes).decode("utf-8")
         #image_paths = await convert_pdf_to_images_async(saved_pdf_path)
+
+        images = convert_bytes_to_image(pdf_stream)
+        print("PDF converted to images:", images)
         
         #updating logic here [Dev Patel]
         initial_state_dict = {