Update app.py

app.py

@@ -2,6 +2,7 @@ import re
 import fitz
 import gradio as gr
 import pandas as pd
+tmpfile = None
 import tempfile
 import pytesseract
 from PIL import Image, ImageEnhance, ImageFilter
@@ -28,6 +29,7 @@ faixas = {
     "TROPONINA": (0, 0.5)
 }
 
+
 def classificar(nome, valor):
     try:
         v = float(valor.replace(">", "").replace("<", "").strip())
@@ -41,14 +43,16 @@ def classificar(nome, valor):
     except:
         return valor
 
+
 # Ajustes para melhorar OCR
-def melhorar_imagem(img: Image.Image) -> Image.Image:
+ def melhorar_imagem(img: Image.Image) -> Image.Image:
     img = img.convert("L")
     img = ImageEnhance.Contrast(img).enhance(2)
     return img.filter(ImageFilter.SHARPEN)
 
+
 # Extrai texto nativo + OCR
-def extrair_texto_pdf(pdf_input):
+ def extrair_texto_pdf(pdf_input):
     if isinstance(pdf_input, dict):
         pdf_path = pdf_input.get("name") or pdf_input.get("file_path")
     elif hasattr(pdf_input, "name") and isinstance(pdf_input.name, str):
@@ -67,66 +71,12 @@ def extrair_texto_pdf(pdf_input):
     tocr = re.sub(r"\s+", " ", " ".join(pytesseract.image_to_string(im) for im in ocr_imgs))
     return tn, tocr
 
+
 # Padrões de extração incluindo EAS completo
-exames = {
-    # Hemograma
-    "LEUCO": r"leuc[óo]citos.*?([\d.,]+)\s*/u?l",
-    "B":    r"bas[óo]filos.*?([\d.,]+)\s?%",
-    "SS":   r"segmentados.*?([\d.,]+)\s?%",
-    "EOS":  r"eosin[óo]filos.*?([\d.,]+)\s?%",
-    "LINF": r"linf[oó]citos.*?([\d.,]+)\s?%",
-    "MONO": r"mon[óo]citos.*?([\d.,]+)\s?%",
-    "HB":   r"hemoglobina.*?([\d.,]+)\s?g/dl",
-    "HT":   r"hemat[óo]crito.*?([\d.,]+)\s?%",
-    "PLT":  r"plaquetas.*?([\d.,]+).*?/u?l",
-    # Bioquímica
-    "AMIL": r"amilase.*?resultado[:\s]*([\d.,]+)\s?u/l",
-    "BT":   r"bilirrubina total.*?([\d.,]+)\s?mg/dl",
-    "BD":   r"bilirrubina direta.*?([\d.,]+)\s?mg/dl",
-    "BI":   r"bilirrubina indireta.*?([\d.,]+)\s?mg/dl",
-    "CR":   r"creatinina.*?resultado[:\s]*([\d.,]+)\s?mg/dl",
-    "UREIA":r"ureia.*?resultado[:\s]*([\d.,]+)\s?mg/dl",
-    "FAL":  r"fosfatase alcalina.*?resultado[:\s]*([\d.,]+)\s?u/l",
-    "GGT":  r"ggt.*?resultado[:\s]*([\d.,]+)\s?u/l",
-    "TGO":  r"tgo.*?resultado[:\s]*([\d.,]+)\s?u/l",
-    "TGP":  r"tgp.*?resultado[:\s]*([\d.,]+)\s?u/l",
-    "GLI":  r"glicose(?! qualitativa).*?resultado[:\s]*([\d.,]+)\s?mg/dl",
-    "LIP":  r"lipase.*?resultado[:\s]*([\d.,]+)\s?u/l",
-    "MG++": r"magn[eé]sio.*?resultado[:\s]*([\d.,]+)\s?mg/dl",
-    # Coagulação
-    "TAP":      r"tempo de protrombina.*?resultado[:\s]*([\d.,]+)",
-    "INR":      r"inr.*?([\d.,]+)",
-    "TTP":      r"ttpa.*?resultado[:\s]*([\d.,]+)",
-    "DIMERO D": r"d[ií]mero d.*?resultado[:\s]*([\d.,]+)",
-    # Inflamatório e Cardíacos
-    "PCR":        r"pcr.*?resultado[:\s]*([\d.,]+)",
-    "CKMB":       r"ck[- ]?mb.*?resultado[:\s]*([\d.,]+)",
-    "CPK":        r"cpk.*?resultado[:\s]*([\d.,]+)",
-    "TROPONINA":  r"troponina(?! qualitativa).*?resultado[:\s]*([>\d.,]+)",
-    "TROPONINA QUAL": r"troponina qualitativa.*?resultado[:\s]*(positivo|negativo)",
-    # EAS completo (Urina)
-    "PROTEINA UR":    r"prote[ií]na\s*(ausente|positivo|negativo)",
-    "GLI UR":         r"glicose\s*(ausente|positivo|negativo)",
-    "CETONAS UR":     r"corpos cet[oô]nicos.*?(ausente|positivo|negativo)",
-    "SANGUE UR":      r"sangue\s*(ausente|positivo|negativo)",
-    "LEUC ESTERASE":  r"leuc[óo]citos? esterase\s*[:\-]?\s*(ausente|positivo|negativo)",
-    "NITRITO UR":     r"nitrito\s*(ausente|positivo|negativo)",
-    "LEUCO EAS":      r"leuc[óo]citos?\s*([\d]+[-\/–][\d]+)",
-    "HEMA EAS":       r"hem[áa]cias?\s*([\d]+[-\/–][\d]+)",
-    "BACTERIAS UR":   r"bact[ée]rias?\s*(raras|ausentes|positivas|negativas)"
-}
+exames = { ... }  # mantêm definição existente
+ordem = [ ... ]   # mantêm definição existente
+
 
-ordem = [
-    "LEUCO","B","SS","EOS","LINF","MONO",
-    "HB","HT","PLT","AMIL","BT","BD","BI",
-    "CR","UREIA","FAL","GGT","TGO","TGP","GLI","LIP","MG++",
-    "PCR","CKMB","CPK","TROPONINA","TROPONINA QUAL",
-    "TAP","INR","TTP","DIMERO D",
-    # EAS
-    "PROTEINA UR","GLI UR","CETONAS UR","SANGUE UR","LEUC ESTERASE","NITRITO UR","LEUCO EAS","HEMA EAS","BACTERIAS UR"
-]
-
-# Montagem do texto formatado
 def extrair_exames_formatado(pdf_file):
     if not pdf_file:
         return "Nenhum arquivo enviado.", None
@@ -138,13 +88,11 @@ def extrair_exames_formatado(pdf_file):
         if not m:
             continue
         raw = m.group(1).strip().upper()
-        # se for valor qualitativo ou EAS, mantém texto
         if "QUAL" in nome or nome.endswith("UR") or nome.endswith("EAS"):
             resultados[nome] = raw
         else:
             resultados[nome] = classificar(nome, raw.replace(",", "."))
 
-    # Linhas EAS e principais
     eas_fields = [f"{k}: {resultados[k]}" for k in ordem if k in resultados and (k.endswith("UR") or k.endswith("EAS"))]
     main_fields = [f"{r}: {resultados[r]}" for r in ordem if r in resultados and not (r.endswith("UR") or r.endswith("EAS"))]
     line_eas = f"🟤 EAS → {' / '.join(eas_fields)}" if eas_fields else ""
@@ -152,11 +100,12 @@ def extrair_exames_formatado(pdf_file):
     final = '\n'.join([l for l in (line_eas, line_main) if l])
 
     # CSV
-df = pd.DataFrame([[k, resultados[k]] for k in resultados], columns=["Exame","Valor"])
-tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".csv")
-df.to_csv(tmp.name, index=False)
+    df = pd.DataFrame([[k, resultados[k]] for k in resultados], columns=["Exame", "Valor"])
+    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".csv")
+    df.to_csv(tmp.name, index=False)
     return final, tmp.name
 
+
 # Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("## 🧪 Extrator Avançado com OCR + EAS + Troponina (Quant. e Qual.)")
@@ -167,4 +116,4 @@ with gr.Blocks() as demo:
     btn.click(extrair_exames_formatado, inputs=pdf_input, outputs=[out_txt, dl])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()