Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,7 +3,7 @@ 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
@@ -34,13 +34,11 @@ def safe_date_default() -> str:
 
 
 def normalize_hora(hora_input: str) -> str:
-    # Mantém o padrão EXATO por default
     if not hora_input:
         return "_____ : _____"
     s = hora_input.strip()
     if s in {"_____:_____", "_____ : _____", "____ : ____"}:
         return "_____ : _____"
-    # se usuário colocar algo tipo 20:15, mantém
     return s
 
 
@@ -59,7 +57,7 @@ def render_print_pdf(out_path: str, data_coleta: str, hora_coleta: str, keys: Li
 
     y = top
 
-    # Cabeçalho (EXATO)
+    # Cabeçalho (modelo)
     y = draw("CLIENTE:", y)
     y -= line_h * 1.3
 
@@ -71,7 +69,7 @@ def render_print_pdf(out_path: str, data_coleta: str, hora_coleta: str, keys: Li
     y = draw(hora_coleta, y)
     y -= line_h * 0.9
 
-    # Lista única (EXATO)
+    # Título da lista
     y = draw("CHAVES DE ACESSO:", y)
     y -= line_h * 0.2
 
@@ -101,65 +99,54 @@ def render_print_pdf(out_path: str, data_coleta: str, hora_coleta: str, keys: Li
     c.save()
 
 
-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")
+st.set_page_config(page_title="Extrator de Chaves NF-e", layout="centered")
+st.title("Extrator de Chaves NF-e (PDF → PDF para imprimir)")
 
+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="_____ : _____")
 
-def run(pdf_path: str, data_coleta: str, hora_coleta: str):
-    # pdf_path vem como string (caminho) quando type="filepath"
-    if not pdf_path or not os.path.exists(pdf_path):
-        raise gr.Error("Envie um PDF válido.")
+if pdf is not None:
+    st.caption(f"Arquivo: {pdf.name}")
 
-    keys = extract_keys_from_pdf(pdf_path)
+if st.button("Gerar arquivos", disabled=(pdf is None)):
+    tmp_in = os.path.join("/tmp", f"entrada_{datetime.now().strftime('%Y%m%d-%H%M%S')}.pdf")
+    with open(tmp_in, "wb") as f:
+        f.write(pdf.read())
+
+    keys = extract_keys_from_pdf(tmp_in)
     if not keys:
-        raise gr.Error("Não encontrei chaves de acesso (44 dígitos) no PDF.")
+        st.error("Não encontrei chaves de acesso (44 dígitos) no PDF.")
+        st.stop()
 
     data_coleta = (data_coleta or safe_date_default()).strip()
-    hora_coleta = normalize_hora(hora_coleta)
+    hora_coleta_norm = normalize_hora(hora_coleta)
 
     ts = datetime.now().strftime("%Y%m%d-%H%M%S")
-
-    # No HF, /tmp é o caminho mais seguro para saída
     out_pdf = os.path.join("/tmp", f"chaves_prontas_{ts}.pdf")
     out_txt = os.path.join("/tmp", f"chaves_{ts}.txt")
 
-    render_print_pdf(out_pdf, data_coleta, hora_coleta, keys)
-    build_txt(out_txt, keys)
-
-    preview = (
-        f"PDF recebido: {os.path.basename(pdf_path)}\n"
-        f"Chaves encontradas: {len(keys)}\n\n"
-        "Primeiras chaves:\n" + "\n".join(keys[:10])
-    )
-
-    return preview, out_pdf, out_txt
+    render_print_pdf(out_pdf, data_coleta, hora_coleta_norm, keys)
 
+    with open(out_txt, "w", encoding="utf-8") as f:
+        f.write("\n".join(keys) + "\n")
 
-with gr.Blocks(title="Extrator de Chaves NF-e (PDF → PDF para imprimir)") as demo:
-    gr.Markdown("## Extrator de Chaves NF-e (PDF → PDF para imprimir)")
+    st.success(f"{len(keys)} chaves encontradas.")
 
-    with gr.Row():
-        pdf = gr.File(
-            label="Envie o PDF",
-            file_types=[".pdf"],
-            type="filepath",  # <-- ESSENCIAL no Hugging Face para evitar 403
+    with open(out_pdf, "rb") as f:
+        st.download_button(
+            "Baixar PDF pronto para imprimir",
+            data=f,
+            file_name=os.path.basename(out_pdf),
+            mime="application/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],
-    )
+    with open(out_txt, "rb") as f:
+        st.download_button(
+            "Baixar TXT (opcional)",
+            data=f,
+            file_name=os.path.basename(out_txt),
+            mime="text/plain",
+        )
 
-demo.launch()
+    st.text_area("Prévia (primeiras 10 chaves)", value="\n".join(keys[:10]), height=200)