Update app.py

app.py

@@ -2095,17 +2095,27 @@ 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 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 convert_pdf_stream_to_images(pdf_stream: io.BytesIO, dpi=300):
+    # Ensure we are at the start of the stream
+    pdf_stream.seek(0)
+    
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_pdf:
+        tmp_pdf.write(pdf_stream.read())
+        tmp_pdf_path = tmp_pdf.name
+    
+    # Now use convert_from_path on the temp file
+    images = convert_from_path(tmp_pdf_path, dpi=dpi)
+    return images
 
 def delay_for_tpm_node(state: GameState):
     logger.info("--- Running DelayForTPMNode ---")
@@ -2347,9 +2357,14 @@ def process_pdf():
         # 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, dpi=250)
-        print("PDF converted to images:", images)
-        
+        # images = convert_bytes_to_image(pdf_stream, dpi=250)
+        # print("PDF converted to images:", images)
+
+        if isinstance(pdf_stream, io.BytesIO):
+            images = convert_pdf_stream_to_images(pdf_stream, dpi=300)
+        else:
+            images = convert_from_path(pdf_stream, dpi=300)
+
         #updating logic here [Dev Patel]
         initial_state_dict = {
             "project_json": project_skeleton,