Salvando alterações

app.py

@@ -22,11 +22,23 @@ if arquivos_upload is not None:
         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(
-            "Extraia o nome, cpf, data de nascimento e número de registro. Retorne apenas um JSON com esses dados.",
+            prompt_analisa_tipo_documento,
             imagem_convertida
         )
-        print(dados_imagem)
+        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")
+    logc.gerar_log(mensagem_log=f"FIM DO PROCESSAMENTO")