Salvando alterações

PromptDocumento.py

@@ -57,8 +57,7 @@ class PromptDocumento:
             
             A resposta precisa retornar apenas um JSON nesse formato:
             {
-                "tipo_documento": "VALOR_TIPO_DOCUMENTO",
-                "data": {...}
+                "tipo_documento": "VALOR_TIPO_DOCUMENTO"
             }
             
             Os valores possíveis para tipo_documento são: CNH | RG | COMPROVANTE_RESIDENCIA | HOLERITE.

app.py

@@ -31,17 +31,19 @@ if arquivos_upload is not None:
 
         st.write(dados_imagem)
 
-        logc.gerar_log(mensagem_log=f"DADOS IMAGEM: {dados_imagem}")
+        print(dados_imagem)
 
-        dados_imagem_dict = Inferencia.string_para_dicionario(dados_imagem)
+        logc.gerar_log(mensagem_log=f"DADOS IMAGEM: {dados_imagem}")
 
-        if dados_imagem_dict["tipo_documento"] == "CNH":
-            dados_cnh = inferencia.extrair_dados_imagem(
-                PromptDocumento.cnh(),
-                imagem_convertida
-            )
-            logc.gerar_log(mensagem_log=f"DADOS CNH: {dados_cnh}")
-            logc.gerar_log(mensagem_log=f"ARQUIVO PROCESSADO: {arquivo.name}")
+        # dados_imagem_dict = Inferencia.string_para_dicionario(dados_imagem)
+        #
+        # if dados_imagem_dict["tipo_documento"] == "CNH":
+        #     dados_cnh = inferencia.extrair_dados_imagem(
+        #         PromptDocumento.cnh(),
+        #         imagem_convertida
+        #     )
+        #     logc.gerar_log(mensagem_log=f"DADOS CNH: {dados_cnh}")
+        #     logc.gerar_log(mensagem_log=f"ARQUIVO PROCESSADO: {arquivo.name}")
 
     logc.gerar_log(mensagem_log=f"FIM DO PROCESSAMENTO")