Update app.py

app.py

@@ -4,11 +4,32 @@ from invoice import (
     do_parse, do_generate, load_pricing, compute_totals,
     today_str, new_invoice_code, logo_data_uri
 )
+import os, re, shutil, uuid
 
 PREVIEW_LOGO = logo_data_uri()  # empty string if no asset file is present
 
 
 # ---------- helpers ----------
+def _stage_for_download(path):
+    """
+    Copy `path` to /tmp with a safe filename so Hugging Face Spaces can serve it.
+    Returns the new path or None.
+    """
+    if not path or not os.path.isfile(path):
+        return None
+    base = os.path.basename(path)
+    root, ext = os.path.splitext(base)
+    # sanitize: keep only safe chars
+    safe_root = re.sub(r'[^A-Za-z0-9_.-]+', '_', root).strip('_') or f"file_{uuid.uuid4().hex[:8]}"
+    safe_ext = ext if ext else ""
+    safe = f"/tmp/{safe_root}{safe_ext}"
+    try:
+        if os.path.abspath(path) != os.path.abspath(safe):
+            shutil.copyfile(path, safe)
+        return safe
+    except Exception:
+        return None
+
 def _unit(modules, unit_override):
     tiers = load_pricing()
     try:
@@ -221,9 +242,22 @@ def on_change(inv_type, inv_date, inv_number, modules, audit_type,
 
 def on_generate(uploaded_file, inv_date, inv_num, inv_type, modules, audit_type, s_audit_date, s_name, s_addr, s_email, s_phone, unit_override):
     # backend handles date, number, unit automatically
-    meta, xlsx, pdf = do_generate(uploaded_file, inv_type, modules, audit_type, s_audit_date, s_name, s_addr, s_email, s_phone)
+    meta, xlsx, pdf = do_generate(
+        uploaded_file, inv_type, modules, audit_type, s_audit_date, s_name, s_addr, s_email, s_phone
+    )
+    # refresh preview
     preview_html = _preview(inv_type, inv_date, inv_num, modules, audit_type, s_name, s_addr, s_email, s_phone, s_audit_date, unit_override)
-    return preview_html, xlsx, (pdf or None)
+
+    # IMPORTANT: stage files into /tmp with safe names so Spaces can serve them
+    staged_xlsx = _stage_for_download(xlsx)
+    staged_pdf  = _stage_for_download(pdf)
+
+    return (
+        preview_html,
+        gr.DownloadButton.update(value=staged_xlsx, visible=bool(staged_xlsx)),
+        gr.DownloadButton.update(value=staged_pdf,  visible=bool(staged_pdf))
+    )
+
 
 def update_audit_dropdown(inv_type_value, current_audit):
     choices = _choices_for(inv_type_value)
@@ -439,7 +473,7 @@ with gr.Blocks(title="Professional Invoice Generator", css=custom_css) as demo:
             gr.HTML('<div class="controls-title">Invoice Settings</div>')
             with gr.Group(elem_classes=["upload-section"]):
                 gr.HTML('<div style="text-align: center; margin-bottom: 8px; font-weight: 600; color: #374151; font-size: 11px;">Upload Report</div>')
-                up = gr.UploadButton("Choose File (.pdf or .docx)", file_types=[".pdf", ".docx"], file_count="single", size="sm")
+                up = gr.File(label="Upload Report (.pdf or .docx)", file_types=[".pdf", ".docx"], file_count="single")
                 gr.HTML('<div style="font-size: 10px; color: #6b7280; text-align: center; margin-top: 4px;">Auto-fill invoice details</div>')
 
             with gr.Group(elem_classes=["form-controls"]):
@@ -464,8 +498,9 @@ with gr.Blocks(title="Professional Invoice Generator", css=custom_css) as demo:
                     1, "NHVR Maintenance Audit", "", "", "", "", "", "auto"
                 ))
             with gr.Group(elem_classes=["download-section"]):
-                dl_xlsx = gr.DownloadButton("Download Excel", elem_classes=["download-btn"], size="sm")
-                dl_pdf = gr.DownloadButton("Download PDF", elem_classes=["download-btn"], size="sm")
+                dl_xlsx = gr.DownloadButton("Download Excel", elem_classes=["download-btn"], size="sm", visible=False)
+                dl_pdf  = gr.DownloadButton("Download PDF",   elem_classes=["download-btn"], size="sm", visible=False, mime_type="application/pdf")
+
 
     # Hidden states
     s_name = gr.State("")