app.py

import streamlit as st
from LogCustomizado import LogCustomizado as logc
from Inferencia import Inferencia
from PIL import Image

st.set_page_config(page_title="CADIN - Cadastro Inteligente")

st.title("CADIN - Cadastro Inteligente")

arquivos_upload = st.file_uploader(
    label="Adicione todos os documentos que estão em formato de imagens.",
    type=["jpg", "jpeg", "png", "pdf"],
    accept_multiple_files=True,
    help="O arquitvo PDF precisa ser uma imagem"
)

inferencia = Inferencia()

if arquivos_upload is not None:
    logc.gerar_log(mensagem_log="INICIANDO PROCESSO DE LEITURA DOS DADOS DOS DOCUMENTOS")
    for arquivo in arquivos_upload:
        st.write(arquivo.name)
        st.image(arquivo)
        imagem_convertida = Image.open(arquivo)
        prompt_analisa_tipo_documento = """
            Faça a análise desse documento e entenda se é uma CNH ou RG ou Comprovante de Residência ou Holerite.
            A resposta precisa retornar apenas um JSON nesse formato:
            {
                "tipo_documento": "VALOR_TIPO_DOCUMENTO"
            }
            Os valores do tipo documento são: CNH|RG|COMPROVANTE_RESIDENCIA|HOLERITE
        """
        dados_imagem = inferencia.extrair_dados_imagem(
            prompt_analisa_tipo_documento,
            imagem_convertida
        )
        st.write(dados_imagem)
        logc.gerar_log(mensagem_log=f"DADOS IMAGEM: {dados_imagem}")

        # "Extraia o nome, cpf, data de nascimento e número de registro. Retorne apenas um JSON com esses dados."

        logc.gerar_log(mensagem_log=f"ARQUIVO PROCESSADO: {arquivo.name}")

    logc.gerar_log(mensagem_log=f"FIM DO PROCESSAMENTO")