Update app_chatgpt.py

app_chatgpt.py

@@ -156,13 +156,13 @@ def reconcile_statements_openai(erp_file, external_file):
     if not erp_file or not external_file:
         error_msg = "Both ERP and External files are required"
         logger.error(f"❌ {error_msg}")
-        yield "❌ Error: Missing files", f"<h3>Error</h3><pre>{error_msg}</pre>"
+        yield "❌ Error: Missing files", f"<h3>Error</h3><pre>{error_msg}</pre>", None
         return
 
     logger.info(f"📁 ERP file: {erp_file.name if erp_file else 'None'}")
     logger.info(f"🏦 External file: {external_file.name if external_file else 'None'}")
 
-    yield "⏳ Processing your request...", ""
+    yield "⏳ Processing your request...", "", None
 
     try:
         # Extract data from files
@@ -231,8 +231,11 @@ Here is the External (Bank or Vendor) data:
         # Calculate processing time
         processing_time = (datetime.now() - session_start).total_seconds()
         logger.info(f"⏱️ Processing completed in {processing_time:.2f} seconds")
+        download_filename = f"reconciliation_{datetime.now().strftime('%Y%m%d_%H%M%S')}.txt"
+        with open(download_filename, "w", encoding="utf-8") as f:
+            f.write(content)
 
-        yield "✅ Done!", html
+        yield "✅ Done!", html, download_filename
 
         # ✅ Upload the files + result to S3
         try:
@@ -255,56 +258,27 @@ Here is the External (Bank or Vendor) data:
         error_msg = f"Error during reconciliation: {e}"
         logger.error(f"❌ {error_msg}")
         logger.error(f"🐛 Full error details: {str(e)}")
-        yield "❌ Error occurred", f"<h3>Error</h3><pre>{error_msg}</pre>"
+        yield "❌ Error occurred", f"<h3>Error</h3><pre>{error_msg}</pre>", None
 
 
-# ✅ Gradio UI for CSV-only
-logger.info("🎨 Building Gradio interface...")
-
-with gr.Blocks(css="""
-#company-logo {
-    width: 25%;
-    margin: auto;
-    display: block;
-}
-""") as iface:
+# ✅ Gradio UI
+with gr.Blocks(css="""#company-logo { width: 25%; margin: auto; display: block; }""") as iface:
     gr.Image("logo_Icon.png", elem_id="company-logo", label="Beiing Human")
     gr.Markdown("## 📊 ERP vs Bank/Vendor Reconciliation Tool")
 
-    gr.Markdown(
-        "⚠️ **Important:** Please upload your statements as CSV files only. "
-        "Include columns: **Date**, **Amount**, **Description**."
-    )
-
     with gr.Row():
-        erp_file = gr.File(
-            label="📁 Upload ERP Statement (CSV only)",
-            type="filepath",
-            file_types=[".csv"]
-        )
-        external_file = gr.File(
-            label="🏦 Upload Bank or Vendor Statement (CSV only)",
-            type="filepath",
-            file_types=[".csv"]
-        )
+        erp_file = gr.File(label="📁 Upload ERP Statement (CSV only)", type="filepath", file_types=[".csv"])
+        external_file = gr.File(label="🏦 Upload Bank or Vendor Statement (CSV only)", type="filepath", file_types=[".csv"])
 
     btn = gr.Button("🔍 Reconcile")
 
-    with gr.Row():
-        support_btn = gr.Button(
-            "📧 Contact Support",
-            link="mailto:rishi@beiinghuman.com?subject=Support%20Request&body=Hi%20Beiing%20Human%20Team%2C%0A%0AI%20have%20a%20question%20about%20the%20reconciliation%20tool.%20Here%20are%20the%20details..."
-        )
-        gr.Markdown("📬 Or email us at: `rishi@beiinghuman.com` (copy & paste)")
-
     status = gr.Markdown()
     result = gr.HTML()
+    download = gr.File(label="⬇️ Download Reconciliation Report", type="filepath")
 
     btn.click(
         fn=reconcile_statements_openai,
         inputs=[erp_file, external_file],
-        outputs=[status, result]
+        outputs=[status, result, download]
     )
-iface.launch()
-# Log when interface is ready
-logger.info("✅ Gradio interface built successfully")
+iface.launch()