Upload leitor_fluxograma.py

leitor_fluxograma.py

@@ -0,0 +1,17 @@
+from transformers import pipeline
+
+class LeitorFluxograma:
+    def __init__(self):
+        # Carrega o modelo leve LLaVA para interpretação de imagens
+        self.reader_pipe = pipeline("image-to-text", model="llava-hf/llava-1.5-7b-hf")
+
+    def gerar_pseudocodigo(self, imagem, pergunta="O que esse fluxograma faz?"):
+        prompt = f"""
+Você é um assistente útil especializado em interpretar fluxogramas a partir de imagens.
+Sua tarefa é descrever o que o fluxograma faz em forma de pseudocódigo.
+Sempre responda em português.
+
+Pergunta: {pergunta}
+"""
+        resultado = self.reader_pipe(imagem)[0]['generated_text']
+        return resultado