Update app.py

app.py

@@ -1,194 +1,193 @@
-# app.py — versão com tema/estilo padronizado
-import gradio as gr
-import tempfile
-from pathlib import Path
-from processador import processar_misto  # pipeline que aceita PDF(s) e ZIP(s)
-
-# -------------------- Tema + CSS do portfólio --------------------
-CUSTOM_CSS = """
-:root{
-  --bg:#000;           /* fundo geral */
-  --panel:#0b0b0b;     /* blocos/painéis */
-  --panel-2:#0e0e0e;   /* inputs/dropdowns */
-  --border:#2a2a2a;    /* borda padrão */
-  --text:#e5e5e5;      /* texto branco suave */
-  --muted:#a3a3a3;     /* texto secundário */
-  --accent:#6ee7b7;    /* cor do foco/seleção (verde menta) */
-}
-
-/* fonte geral (system UI) */
-html, body, .gradio-container {
-  background: var(--bg)!important;
-  color: var(--text)!important;
-  font-family: ui-sans-serif, system-ui, -apple-system, "Segoe UI", Roboto, Inter, "Helvetica Neue", Arial, "Noto Sans", "Liberation Sans", sans-serif !important;
-}
-
-/* blocos/painéis */
-.gradio-container .block,
-.gradio-container .gr-box,
-.gradio-container .gr-panel {
-  background: var(--panel) !important;
-  border: 1px solid var(--border) !important;
-  border-radius: 12px !important;
-}
-
-/* remover o bloco atrás do TÍTULO (qualquer .block que contenha h1) */
-.gradio-container .block:has(h1){
-  background: transparent !important;
-  border: 0 !important;
-  box-shadow: none !important;
-}
-
-/* botões estilo "pílula" */
-button, .gr-button{
-  border-radius: 9999px !important;
-  border: 1px solid var(--border) !important;
-  background: var(--panel-2) !important;
-}
-button:hover{ border-color:#4a4a4a !important; }
-
-/* inputs/textarea/file/dropdown: mesmo tom escuro */
-input, textarea, select,
-.gradio-container .gr-textbox,
-.gradio-container .gr-input,
-.gradio-container .gradio-dropdown,
-.gradio-container .gr-file,
-.gradio-container .gr-file-download,
-.gradio-container .gr-select-container,
-.gradio-container .wrap .items-center select {
-  background: var(--panel-2) !important;
-  border: 1px solid var(--border) !important;
-  color: var(--text) !important;
-  border-radius: 12px !important;
-}
-
-/* opções do select */
-select > option { background: var(--panel-2); color: var(--text); }
-
-/* foco/seleção visível (inputs, selects, file, etc.) */
-input:focus, textarea:focus, select:focus,
-.gradio-container .gr-textbox:focus-within,
-.gradio-container .gr-input:focus-within,
-.gradio-container .gradio-dropdown:focus-within,
-.gradio-container .gr-file:focus-within,
-.gradio-container .gr-select-container:focus-within {
-  outline: none !important;
-  border-color: var(--accent) !important;
-  box-shadow: 0 0 0 2px rgba(110,231,183,0.18) !important; /* halo */
-}
-
-/* uploader QUADRADO (remove círculo/arredondamento exagerado) */
-.gradio-container [data-testid="file"] .rounded-full,
-.gradio-container [data-testid="files"] .rounded-full { border-radius:12px !important; }
-.gradio-container [data-testid="file"] [class*="aspect-"],
-.gradio-container [data-testid="files"] [class*="aspect-"] { aspect-ratio:auto !important; }
-.gradio-container [data-testid="file"] .h-full.w-full,
-.gradio-container [data-testid="files"] .h-full.w-full { border-radius:12px !important; }
-
-/* gallery e saídas de arquivo */
-.gradio-container .gr-gallery,
-.gradio-container .gr-file-download{
-  background: var(--panel-2) !important;
-  border: 1px solid var(--border) !important;
-  border-radius: 12px !important;
-}
-
-/* badges/labels */
-.badge, .token { background:#0f0f0f !important; border-radius:9999px !important; }
-
-.gradio-container .fixed.bottom-0,
-.gradio-container div[class*="fixed"][class*="bottom-0"],
-.gradio-container footer,
-body > div.fixed.bottom-0,
-div.fixed.bottom-0 {
-  display: none !important;
-  visibility: hidden !important;
-  height: 0 !important;
-  overflow: hidden !important;
-  pointer-events: none !important;
-}
-"""
-
-THEME = gr.themes.Soft(primary_hue="zinc", neutral_hue="zinc")
-
-# -----------------------------------------------------------
-
-# Converte buffers em arquivos temporários para a Gallery
-def _galeria_temp(imagens, formato):
-    gal = []
-    ext = "ico" if formato == "ico" else formato.lower()
-    for legenda, img_io in imagens:
-        tmp = tempfile.NamedTemporaryFile(delete=False, suffix=f".{ext}")
-        tmp.write(img_io.read())
-        tmp.close()
-        img_io.seek(0)
-        gal.append((tmp.name, legenda))
-    return gal
-
-# Função chamada pelo submit
-def processar(arquivos, modo, paginas_input, formato_opcao):
-    if not arquivos:
-        return "Por favor, envie ao menos um PDF (ou .zip contendo PDFs).", None
-
-    # gr.Files pode vir string ou lista:
-    paths = [Path(arquivos)] if isinstance(arquivos, str) else [Path(p) for p in arquivos]
-
-    # Do dropdown “jpeg [Recomendado …]” pegamos só o formato:
-    formato = formato_opcao.split(" [")[0].lower()
-
-    # Interpreta páginas quando necessário
-    paginas = None
-    if modo == "Extrair páginas específicas":
-        try:
-            paginas = [int(p.strip()) for p in paginas_input.split("-") if p.strip().isdigit()]
-            if not paginas:
-                return "Nenhuma página válida foi informada.", None
-        except Exception as e:
-            return f"Erro ao interpretar as páginas: {e}", None
-
-    try:
-        imagens, zip_path = processar_misto(paths, modo, paginas, formato)
-    except Exception as e:
-        return f"Erro ao processar: {e}", None
-
-    galeria = _galeria_temp(imagens, formato)
-    return galeria, zip_path
-
-FORMATOS = [
-    "  jpeg [Recomendado - compacto e boa qualidade]",
-    "  png [Alta qualidade, suporta transparência]",
-    "  bmp [Sem compressão - ideal para edição bruta]",
-    "  ico [Favicon para sites, atalhos e apps]",
-]
-
-demo = gr.Interface(
-    fn=processar,
-    inputs=[
-        gr.Files(
-            label="Envie PDF(s) ou .zip com PDFs",
-            file_count="multiple",
-            file_types=[".pdf", ".zip"],
-        ),
-        gr.Radio(
-            ["Extrair todas as páginas", "Extrair páginas específicas"],
-            label="Modo",
-            value="Extrair todas as páginas",
-        ),
-        gr.Textbox(
-            label="Páginas (se usar 'Específicas')",
-            placeholder="Ex.: 3 - 5 - 10",
-        ),
-        gr.Dropdown(choices=FORMATOS, value=FORMATOS[0], label="Formato das imagens"),
-    ],
-    outputs=[
-        gr.Gallery(label="Imagens"),
-        gr.File(label="ZIP com as imagens"),
-    ],
-    title="Extrator de Imagens de PDF",
-    allow_flagging="never",
-    theme=THEME,
-    css=CUSTOM_CSS,
-)
-
-if __name__ == "__main__":
-    demo.launch()
+# app.py — versão final p/ Hugging Face (tema + UI padronizados)
+import gradio as gr
+import re
+from pathlib import Path
+from typing import List
+from processador import processar_misto  # devolve [(caminho_img, legenda)], caminho_zip
+
+# -------------------- Tema + CSS do portfólio --------------------
+CUSTOM_CSS = """
+:root{
+  --bg:#000;           /* fundo geral */
+  --panel:#0b0b0b;     /* blocos/painéis */
+  --panel-2:#0e0e0e;   /* inputs/dropdowns */
+  --border:#2a2a2a;    /* borda padrão */
+  --text:#e5e5e5;      /* texto branco suave */
+  --muted:#a3a3a3;     /* texto secundário */
+  --accent:#6ee7b7;    /* cor do foco/seleção (verde menta) */
+}
+
+/* fonte geral (system UI) */
+html, body, .gradio-container {
+  background: var(--bg)!important;
+  color: var(--text)!important;
+  font-family: ui-sans-serif, system-ui, -apple-system, "Segoe UI", Roboto, Inter, "Helvetica Neue", Arial, "Noto Sans", "Liberation Sans", sans-serif !important;
+}
+
+/* blocos/painéis */
+.gradio-container .block,
+.gradio-container .gr-box,
+.gradio-container .gr-panel {
+  background: var(--panel) !important;
+  border: 1px solid var(--border) !important;
+  border-radius: 12px !important;
+}
+
+/* remover o bloco atrás do TÍTULO */
+.gradio-container .block:has(h1){
+  background: transparent !important;
+  border: 0 !important;
+  box-shadow: none !important;
+}
+
+/* botões estilo "pílula" */
+button, .gr-button{
+  border-radius: 9999px !important;
+  border: 1px solid var(--border) !important;
+  background: var(--panel-2) !important;
+}
+button:hover{ border-color:#4a4a4a !important; }
+
+/* inputs/textarea/file/dropdown */
+input, textarea, select,
+.gradio-container .gr-textbox,
+.gradio-container .gr-input,
+.gradio-container .gradio-dropdown,
+.gradio-container .gr-file,
+.gradio-container .gr-file-download,
+.gradio-container .gr-select-container,
+.gradio-container .wrap .items-center select {
+  background: var(--panel-2) !important;
+  border: 1px solid var(--border) !important;
+  color: var(--text) !important;
+  border-radius: 12px !important;
+}
+select > option { background: var(--panel-2); color: var(--text); }
+
+/* foco/seleção visível */
+input:focus, textarea:focus, select:focus,
+.gradio-container .gr-textbox:focus-within,
+.gradio-container .gr-input:focus-within,
+.gradio-container .gradio-dropdown:focus-within,
+.gradio-container .gr-file:focus-within,
+.gradio-container .gr-select-container:focus-within {
+  outline: none !important;
+  border-color: var(--accent) !important;
+  box-shadow: 0 0 0 2px rgba(110,231,183,0.18) !important;
+}
+
+/* uploader QUADRADO */
+.gradio-container [data-testid="file"] .rounded-full,
+.gradio-container [data-testid="files"] .rounded-full { border-radius:12px !important; }
+.gradio-container [data-testid="file"] [class*="aspect-"],
+.gradio-container [data-testid="files"] [class*="aspect-"] { aspect-ratio:auto !important; }
+.gradio-container [data-testid="file"] .h-full.w-full,
+.gradio-container [data-testid="files"] .h-full.w-full { border-radius:12px !important; }
+
+/* gallery e saídas */
+.gradio-container .gr-gallery,
+.gradio-container .gr-file-download{
+  background: var(--panel-2) !important;
+  border: 1px solid var(--border) !important;
+  border-radius: 12px !important;
+}
+
+/* badges/labels */
+.badge, .token { background:#0f0f0f !important; border-radius:9999px !important; }
+
+/* esconder rodapé do Gradio */
+.gradio-container .fixed.bottom-0,
+.gradio-container div[class*="fixed"][class*="bottom-0"],
+.gradio-container footer,
+body > div.fixed.bottom-0,
+div.fixed.bottom-0 {
+  display: none !important;
+  visibility: hidden !important;
+  height: 0 !important;
+  overflow: hidden !important;
+  pointer-events: none !important;
+}
+"""
+
+THEME = gr.themes.Soft(primary_hue="zinc", neutral_hue="zinc")
+
+# -------------------- Util --------------------
+def _normalize_paths(arquivos) -> List[Path]:
+    """gr.Files pode entregar str, Path, objeto com .name ou até dict-like."""
+    items = arquivos if isinstance(arquivos, list) else [arquivos]
+    paths: List[Path] = []
+    for a in items:
+        if isinstance(a, (str, Path)):
+            paths.append(Path(a))
+        else:
+            name = getattr(a, "name", None)
+            if name is None:
+                try:
+                    name = a["name"]  # alguns retornam dict-like
+                except Exception:
+                    name = str(a)
+            paths.append(Path(name))
+    return paths
+
+# -------------------- Função do app --------------------
+def processar(arquivos, modo, paginas_input, formato_opcao):
+    if not arquivos:
+        return "Por favor, envie ao menos um PDF (ou .zip com PDFs).", None
+
+    paths = _normalize_paths(arquivos)
+
+    # pega "jpeg" de "jpeg [Recomendado - ...]"
+    formato = formato_opcao.split(" [")[0].strip().lower()
+
+    paginas = None
+    if "específica" in modo.lower():  # aceita "Específicas"/"Específica"
+        nums = re.findall(r"\d+", paginas_input or "")
+        if not nums:
+            return "Nenhuma página válida foi informada.", None
+        paginas = [int(n) for n in nums]
+
+    try:
+        galeria, zip_path = processar_misto(paths, modo, paginas, formato)
+        return galeria, zip_path  # Gallery aceita lista de (caminho, legenda)
+    except Exception as e:
+        return f"Erro ao processar: {e}", None
+
+# opções do dropdown (sem espaços no início)
+FORMATOS = [
+    "jpeg [Recomendado - compacto e boa qualidade]",
+    "png [Alta qualidade, suporta transparência]",
+    "bmp [Sem compressão - ideal para edição bruta]",
+    "ico [Favicon para sites, atalhos e apps]",
+]
+
+demo = gr.Interface(
+    fn=processar,
+    inputs=[
+        gr.Files(
+            label="Envie PDF(s) ou .zip com PDFs",
+            file_count="multiple",
+            file_types=[".pdf", ".zip"],
+        ),
+        gr.Radio(
+            ["Extrair todas as páginas", "Extrair páginas específicas"],
+            label="Modo",
+            value="Extrair todas as páginas",
+        ),
+        gr.Textbox(
+            label="Páginas (se usar 'Específicas')",
+            placeholder="Ex.: 3 - 5 - 10",
+        ),
+        gr.Dropdown(choices=FORMATOS, value=FORMATOS[0], label="Formato das imagens"),
+    ],
+    outputs=[
+        gr.Gallery(label="Imagens"),
+        gr.File(label="ZIP com as imagens"),
+    ],
+    title="Extrator de Imagens de PDF",
+    allow_flagging="never",
+    theme=THEME,
+    css=CUSTOM_CSS,
+)
+
+if __name__ == "__main__":
+    demo.launch()