Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,22 +2,17 @@ import re
 from datetime import datetime
 from typing import List
 
-import gradio as gr
+import streamlit as st
 from pypdf import PdfReader
-
 from reportlab.lib.pagesizes import A4
 from reportlab.pdfgen import canvas
 from reportlab.lib.units import mm
 
-
 KEY_RE = re.compile(r"\b\d{44}\b")
 
-
 def extract_keys_from_pdf(pdf_path: str) -> List[str]:
     reader = PdfReader(pdf_path)
-    found: List[str] = []
-    seen = set()
-
+    found, seen = [], set()
     for page in reader.pages:
         text = page.extract_text() or ""
         for k in KEY_RE.findall(text):
@@ -26,42 +21,33 @@ def extract_keys_from_pdf(pdf_path: str) -> List[str]:
                 found.append(k)
     return found
 
-
 def safe_date_default() -> str:
     return datetime.now().strftime("%d/%m/%Y")
 
-
 def render_print_pdf(out_path: str, data_coleta: str, hora_coleta: str, keys: List[str]):
     c = canvas.Canvas(out_path, pagesize=A4)
     w, h = A4
-
     left = 20 * mm
     top = h - 20 * mm
     line_h = 6.2 * mm
 
-    def draw(text: str, y: float, bold=False, size=11) -> float:
+    def draw(text: str, y: float, size=11, bold=False) -> float:
         c.setFont("Helvetica-Bold" if bold else "Helvetica", size)
         c.drawString(left, y, text)
         return y - line_h
 
     y = top
-
-    # Cabeçalho
     y = draw("CLIENTE:", y)
     y -= line_h * 1.3
-
     y = draw("DATA DA COLETA:", y)
     y = draw(data_coleta, y)
     y -= line_h * 0.6
-
     y = draw("HORA DA COLETA:", y)
     y = draw("_____ : _____", y)
     y -= line_h * 0.9
 
-    # ===== ALTERAÇÃO AQUI =====
     y = draw("CHAVES DE ACESSO:", y)
     y -= line_h * 0.2
-    # =========================
 
     c.setFont("Helvetica", 10.8)
     for k in keys:
@@ -75,7 +61,6 @@ def render_print_pdf(out_path: str, data_coleta: str, hora_coleta: str, keys: Li
     y -= line_h * 0.8
     y = draw(f"TOTAL DA REMESSA:  {len(keys)} VOLUMES", y)
 
-    # Assinaturas
     if y < 55 * mm:
         c.showPage()
 
@@ -85,62 +70,24 @@ def render_print_pdf(out_path: str, data_coleta: str, hora_coleta: str, keys: Li
     c.drawString(left + 140 * mm, y_sig + 12, "ASSINATURA DO MOTORISTA")
     c.line(left + 55 * mm, y_sig + 10, left + 118 * mm, y_sig + 10)
     c.line(left + 130 * mm, y_sig + 10, left + 193 * mm, y_sig + 10)
-
     c.save()
 
+st.title("Extrator de Chaves NF-e (PDF → PDF para imprimir)")
 
-def build_txt(out_path: str, keys: List[str]):
-    with open(out_path, "w", encoding="utf-8") as f:
-        f.write("\n".join(keys))
-        f.write("\n")
+pdf = st.file_uploader("Envie o PDF", type=["pdf"])
+data_coleta = st.text_input("Data da coleta", value=safe_date_default())
+hora_coleta = st.text_input("Hora da coleta", value="_____ : _____")
 
+if pdf and st.button("Gerar PDF"):
+    with open("entrada.pdf", "wb") as f:
+        f.write(pdf.read())
 
-def run(pdf_file, data_coleta, hora_coleta):
-    if pdf_file is None:
-        raise gr.Error("Envie um PDF.")
-
-    keys = extract_keys_from_pdf(pdf_file)
+    keys = extract_keys_from_pdf("entrada.pdf")
     if not keys:
-        raise gr.Error("Não encontrei chaves de 44 dígitos no PDF.")
-
-    ts = datetime.now().strftime("%Y%m%d-%H%M%S")
-    out_pdf = f"chaves_prontas_{ts}.pdf"
-    out_txt = f"chaves_{ts}.txt"
-
-    render_print_pdf(
-        out_pdf,
-        data_coleta=data_coleta or safe_date_default(),
-        hora_coleta=hora_coleta or "_____ : _____",
-        keys=keys,
-    )
-    build_txt(out_txt, keys)
-
-    preview = (
-        f"Total extraído: {len(keys)}\n\n"
-        "Primeiras chaves:\n" + "\n".join(keys[:10])
-    )
-    return preview, out_pdf, out_txt
-
-
-with gr.Blocks(title="Extrator de Chaves NF-e") as demo:
-    gr.Markdown("### Envie um PDF e gere um arquivo pronto para imprimir.")
-
-    with gr.Row():
-        pdf = gr.File(label="PDF", file_types=[".pdf"])
-        preview = gr.Textbox(label="Prévia", lines=12)
-
-    with gr.Row():
-        data_coleta = gr.Textbox(label="Data da coleta", value=safe_date_default())
-        hora_coleta = gr.Textbox(label="Hora da coleta", value="_____ : _____")
-
-    btn = gr.Button("Gerar arquivos")
-    out_pdf = gr.File(label="PDF pronto para imprimir")
-    out_txt = gr.File(label="TXT (opcional)")
-
-    btn.click(
-        fn=run,
-        inputs=[pdf, data_coleta, hora_coleta],
-        outputs=[preview, out_pdf, out_txt],
-    )
-
-demo.launch()
+        st.error("Não encontrei chaves de 44 dígitos no PDF.")
+    else:
+        out_pdf = f"chaves_prontas_{datetime.now().strftime('%Y%m%d-%H%M%S')}.pdf"
+        render_print_pdf(out_pdf, data_coleta, hora_coleta, keys)
+        st.success(f"{len(keys)} chaves encontradas.")
+        with open(out_pdf, "rb") as f:
+            st.download_button("Baixar PDF pronto para imprimir", f, file_name=out_pdf, mime="application/pdf")