Salvando alterações

LogCustomizado.py

@@ -1,13 +1,19 @@
-from datetime import datetime
+from datetime import datetime, timedelta
 
 class LogCustomizado:
 
     @staticmethod
     def gerar_log(mensagem_log, status='INFO', retornarMensagem=False):
-        data_hora = datetime.now().strftime('%d/%m/%Y %H:%M:%S')
+        data_hora = LogCustomizado.data_hora_atual()
         mensagem_log_completa = f"[{data_hora}] - [{status}] - {mensagem_log}"
 
         if (retornarMensagem):
             return mensagem_log_completa
 
-        print(mensagem_log_completa)
+        print(mensagem_log_completa)
+
+    @staticmethod
+    def data_hora_atual():
+        data_hora_atual = datetime.now()
+        data_hora_reduzida = data_hora_atual - timedelta(hours=5)
+        return data_hora_reduzida.strftime('%d/%m/%Y %H:%M:%S')

app.py

@@ -31,19 +31,17 @@ if arquivos_upload is not None:
 
         st.write(dados_imagem)
 
-        print(dados_imagem)
-
-        logc.gerar_log(mensagem_log=f"DADOS IMAGEM: {dados_imagem}")
-
-        # 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"DEFININDO TIPO DE DOCUMENTO: {dados_imagem}")
+
+        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")