Baixar aipdf - baixar o código-fonte aipdf

aipdf

Código-Fonte de IA

v0.0.4

Baixar

AIPDF: OCR de PDF simples com modelos multimodais semelhantes a GPT

Dane-se os OCRs tradicionais ou bibliotecas pesadas para obter dados de PDFs, o GenAI faz um trabalho melhor!

AIPDF é uma biblioteca Python pura autônoma, minimalista, mas poderosa, que aproveita modelos de IA de geração multimodal (OpenAI, llama3 ou alternativas compatíveis) para extrair dados de PDFs e convertê-los em vários formatos, como Markdown ou JSON.

Instalação

pip install aipdf

no macOS você precisará instalar o poppler

brew install poppler

Início rápido

 from aipdf import ocr

# Your OpenAI API key   
api_key = 'your_openai_api_key'

file = open ( 'somepdf.pdf' , 'rb' )
markdown_pages = ocr ( file , api_key )

Ollama

Você pode usar com qualquer modelo multimodal ollama

 ocr ( pdf_file , api_key = 'ollama' , model = "llama3.2" , base_url = 'http://localhost:11434/v1' , prompt = ...)

Qualquer sistema de arquivos

Optamos que você passe um objeto de arquivo, pois assim fica flexível você usar isso com qualquer tipo de sistema de arquivos, s3, localfiles, urls etc.

Do URL

 pdf_file = io . BytesIO ( requests . get ( 'https://arxiv.org/pdf/2410.02467' ). content )

# extract
pages = ocr ( pdf_file , api_key , prompt = "extract tables, return each table in json" )

Do S3

 s3 = boto3 . client ( 's3' , config = Config ( signature_version = 's3v4' ),
                  aws_access_key_id = access_token ,
                  aws_secret_access_key = '' , # Not needed for token-based auth
                  aws_session_token = access_token )


pdf_file = io . BytesIO ( s3 . get_object ( Bucket = bucket_name , Key = object_key )[ 'Body' ]. read ())
# extract 
pages = ocr ( pdf_file , api_key , prompt = "extract charts data, turn it into tables that represent the variables in the chart" )

Por que AIPDF?

Simplicidade : AIPDF fornece uma função simples, requer configuração, dependências e configurações mínimas.
Flexibilidade : Extraia dados em Markdown, JSON, HTML, YAML, qualquer que seja... formato de arquivo e esquema.
Poder da IA : aproveita modelos multimodais de última geração (gpt, llama, ..).
Personalizável : adapte o processo de extração às suas necessidades específicas com prompts personalizados.
Eficiente : utiliza processamento paralelo para extração mais rápida de PDFs de várias páginas.