PromptDocumento.py

class PromptDocumento:

    @staticmethod
    def cnh():
        return """
        Extraia o nome, cpf, data de nascimento, número de registro, primeira habilitação,
        validade da CNH, RG (documento de identidade), órgão emissor, estado,
        categoria da habilitação, nacionalidade, filiação pai (se houver), filiação mãe (se houver),
        local, data da emissão e observações (se houver).
        Para os campos "se houver", caso não tenha nenhum valor, precisa manter o campo do JSON e deixar ele com valor vazio.
        Retorne apenas o JSON abaixo preenchido com os valores encontrados:
        
        {
            "nome": "",
            "cpf": "",
            "data_nascimento": "",
            "numero_registro": "",
            "primeiro_habilitacao": "",
            "validade_cnh": "",
            "rg": "",
            "orgao_emissor": "",
            "orgao_emissor_estado": "",
            "categoria_habilitacao": "",
            "nacionalidade": "",
            "filiacao_pai": "",
            "filiacao_mae": "",
            "local": "",
            "data_emissao": "",
            "observacoes": "",
        }         
        """

    @staticmethod
    def selecionar_tipo_documento():
        return """
            Faça a análise desse documento e identifique seu tipo com base nos campos e textos presentes. Siga as instruções abaixo para determinar o tipo de documento:

            - **CNH**: Somente se você encontrar qualquer um dos seguintes indicadores:
              - Texto contendo "CARTEIRA NACIONAL DE HABILITAÇÃO" ou "CATEGORIA HABILITAÇÃO".
              - Campo com a sigla "DENATRAN" ou "DEPARTAMENTO NACIONAL DE TRÂNSITO".
              - Informações sobre validade de habilitação ou número de registro de CNH.
              
            - **RG**: Somente se você encontrar qualquer um dos seguintes indicadores:
              - Texto contendo "REGISTRO GERAL" ou "IDENTIDADE".
              - Campo com a sigla "MINISTÉRIO DA JUSTIÇA" ou "SECRETARIA DE SEGURANÇA PÚBLICA".
              - Informações sobre data de emissão de RG ou número de identificação civil.
              - Essa regra deve ser considerada somente se você não encontrar nada sobre Categoria Habilitaçã. Se esse campo for encontrado, o retorno precisa ser CNH.
            
            - **COMPROVANTE_RESIDENCIA**: Somente se você encontrar qualquer um dos seguintes indicadores:
              - Texto relacionado a serviços públicos, como "CONTA DE ÁGUA", "CONTA DE LUZ", "TELEFONE FIXO".
              - Informações sobre endereço residencial detalhado.
              
            - **HOLERITE**: Somente se você encontrar qualquer um dos seguintes indicadores:
              - Texto contendo "SALÁRIO", "IRRF" (Imposto de Renda Retido na Fonte), "INSS".
              - Informações sobre remuneração ou descontos trabalhistas.
            
            A resposta precisa retornar apenas um JSON nesse formato:
            {
                "tipo_documento": "VALOR_TIPO_DOCUMENTO"
            }
            
            Os valores possíveis para tipo_documento são: CNH | RG | COMPROVANTE_RESIDENCIA | HOLERITE.
            Dentro do campo data você deve adicionar os campos e valores encontrados no na imagem. 
        """