app.py

import sys
import os
import json
import PyPDF2
from PIL import Image
import io

def extract_images_and_structure_from_pdf(pdf_path, output_path):
    try:
        with open(pdf_path, 'rb') as pdf_file:
            pdf_reader = PyPDF2.PdfReader(pdf_file)
            structured_data = []
            image_count = 0

            for page_num in range(len(pdf_reader.pages)):
                page = pdf_reader.pages[page_num]

                page_content = {"section": f"Page {page_num + 1}", "content": []}

                # Extract text (basic approach, might need refinement)
                text = page.extract_text()
                if text:
                    page_content["content"].append({"type": "text", "text": text})

                # Extract images
                # Iterate through the objects on the page
                for img_file_index, img in enumerate(page.images):
                  image_count += 1
                  image_filename = f"image_{page_num+1}_{image_count}_{img_file_index}.jpg"
                  image_filepath = os.path.join(output_path, image_filename)
                  try:
                      image = Image.open(io.BytesIO(img.data))
                      image.save(image_filepath)
                      page_content["content"].append({"type": "image", "image_path": image_filename})
                  except Exception as e:
                      print(f"Error processing image: {e}")
                      page_content["content"].append({"type": "image", "image_path": None, "error": str(e)})

                structured_data.append(page_content)

            # Save to JSON file
            json_filepath = os.path.join(output_path, "output.json")
            with open(json_filepath, 'w', encoding='utf-8') as f:
                json.dump(structured_data, f, indent=4, ensure_ascii=False)

            print(f"Extracted structured data and images to {json_filepath}")

    except Exception as e:
        print(f"An error occurred: {str(e)}")


if __name__ == "__main__":
    if len(sys.argv) < 3:
        print("Usage: python app.py <pdf_file_path> <output_path>")
        sys.exit(1)

    pdf_file_path = sys.argv[1]
    output_path = sys.argv[2]

    if not os.path.exists(output_path):
        os.makedirs(output_path)

    extract_images_and_structure_from_pdf(pdf_file_path, output_path)