Skip to content

Tool provides a better way to ocr pdfs that contain structured data. Will use GPT4 Vision to structured data as Json. Each page will be extracted to a Json file...

License

Notifications You must be signed in to change notification settings

maximoguerrero/PDF-GPT4-JSON

Repository files navigation

PDF-GPT4-JSON

This project is designed to convert PDF files into JSON format using GPT-4. For each page in the PDF, a JSON file will be generated. The hierarchy of the JSON structure will be inferred from the layout of the data in the PDF. Can be used as python module or in the cli.

Theory of Generating Structured JSON using GPT-4 Vision

GPT-4 Vision is a state-of-the-art language model that has been fine-tuned for image understanding and analysis. It leverages the power of deep learning to extract meaningful information from PDF files and convert them into structured JSON format.

The process of generating structured JSON using GPT-4 Vision involves the following steps:

  1. PDF Parsing: The PDF file is parsed to extract the textual content and layout information of each page.

  2. Text Extraction: The extracted text is processed to remove any noise or irrelevant information, such as headers, footers, and page numbers.

  3. Layout Analysis: GPT-4 Vision analyzes the layout of the text on each page to identify the hierarchical structure of the data. It looks for patterns, indentation, and formatting cues to infer the relationships between different elements.

  4. JSON Generation: Based on the layout analysis, GPT-4 Vision generates a structured JSON representation of the PDF content. Each page is represented as a separate JSON file, with nested objects and arrays to capture the hierarchical relationships.

By leveraging the power of GPT-4 Vision, the PDF-GPT4-JSON project simplifies the process of converting PDF files into structured JSON format. This enables developers to easily extract and analyze data from PDFs, opening up a wide range of possibilities for data processing and automation.

Installation

  1. Install via pip:

    pip install pdf_gpt4_json
  2. Set your OpenAI api key:

    export OPENAI_API_KEY=sk-xxxxxxxxxxx

    You can also pass in as a command line arugment to the tool --openai-key

Usage

  1. Run the conversion script:

    pdf-gpt4-json ./sample.pdf

    Will generate tmp working folder and an output folder with json for each page. If you already have a folder 'output' it will get renamed, if the working folders exist they will be deleted and recreated to insure that they are empyt

  2. Final output folder will be samplepdf_final_folders in this case. It will use the pdfs filename as the prefix to the output folder. If there are errors it will be in {filename}_errors folder. In both cases the cli will return a message.

Parameter

--prompt-file (str, optional): Path to a file containing a prompt for the model.
--openai-key (str, optional): OpenAI API key. If not provided, it will be read from the environment.
--model (str, optional): Model to use. Default is "gpt-4-vision-preview".
--verbose (bool, optional): If True, print additional debug information. Default is False.
--cleanup (bool, optional): If True, cleanup temporary files after processing. Default is False.

By adjusting these parameters, users can tailor the PDF-to-JSON conversion to their specific needs and preferences.

Contributing

Contributions are welcome! Please follow the guidelines in CONTRIBUTING.md.

License

This project is licensed under the GLP-3 LICENSE.

About

Tool provides a better way to ocr pdfs that contain structured data. Will use GPT4 Vision to structured data as Json. Each page will be extracted to a Json file...

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published