Update app.py

app.py

@@ -240,25 +240,21 @@ def on_change(inv_type, inv_date, inv_number, modules, audit_type,
     return _preview(inv_type, inv_date, inv_number, modules, audit_type,
                     s_name, s_addr, s_email, s_phone, s_audit_date, unit_override)
 
-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
+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):
     meta, xlsx, pdf = do_generate(
-        uploaded_file, inv_type, modules, audit_type, s_audit_date, s_name, s_addr, s_email, s_phone
+        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)
-
-    # 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)
-
+    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 updates for the two download buttons
     return (
         preview_html,
-        gr.DownloadButton.update(value=staged_xlsx, visible=bool(staged_xlsx)),
-        gr.DownloadButton.update(value=staged_pdf,  visible=bool(staged_pdf))
+        gr.update(value=xlsx, visible=bool(xlsx)),
+        gr.update(value=pdf,  visible=bool(pdf)),
     )
 
-
 def update_audit_dropdown(inv_type_value, current_audit):
     choices = _choices_for(inv_type_value)
     value = current_audit if current_audit in choices else choices[0]
@@ -474,6 +470,8 @@ with gr.Blocks(title="Professional Invoice Generator", css=custom_css) as demo:
             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.File(label="Upload Report (.pdf or .docx)", file_types=[".pdf", ".docx"], file_count="single")
+                # ADD THIS LINE HERE
+                file_display = gr.File(label="Selected file", interactive=False, show_download_button=False)
                 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"]):
@@ -522,15 +520,18 @@ with gr.Blocks(title="Professional Invoice Generator", css=custom_css) as demo:
             uploaded,
             inv_t, mods, normalized,
             out[3], out[4], out[5], out[6], out[7],
-            today_str(), "Auto"
+            today_str(), "Auto",
+            uploaded
         )
 
     up.upload(
     _remember_and_parse,
     inputs=[up],
-    outputs=[uploaded_file, inv_type, modules, audit_type, s_name, s_addr, s_email, s_phone, s_audit_date, inv_date, inv_num],
+    outputs=[uploaded_file, inv_type, modules, audit_type, s_name, s_addr, s_email,
+             s_phone, s_audit_date, inv_date, inv_num, file_display],   # + file_display
     api_name=False
     )
+
     
     inv_type.change(
         update_audit_dropdown,