Salvando alterações

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 from LogCustomizado import LogCustomizado as logc
+from Inferencia import Inferencia
+from PIL import Image
 
 logc.gerar_log(mensagem_log="INICIANDO PROCESSO")
 
@@ -14,6 +16,15 @@ arquivos_upload = st.file_uploader(
     help="O arquitvo PDF precisa ser uma imagem"
 )
 
+inferencia = Inferencia()
+
 if arquivos_upload is not None:
-    for item in arquivos_upload:
-        st.write(item)
+    for arquivo in arquivos_upload:
+        st.write(arquivo.name)
+        st.image(arquivo)
+        imagem_convertida = Image.open(arquivo)
+        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.",
+            imagem_convertida
+        )
+        print(dados_imagem)