Replace broken PDF with html2canvas screenshot export on Stats page

app.py

@@ -1091,25 +1091,7 @@ with st.sidebar:
                     mime="text/csv",
                     key=f"dl_{_rkey}",
                 )
-                # ── PDF Report button ──────────────────────────────
-                try:
-                    from ml.report_generator import generate_report
-                    _stream_title = _s.get("video_title") or _s.get("video_id") or _rkey
-                    _pdf_bytes = generate_report(
-                        all_data=_dl_rows,
-                        stream_title=_stream_title,
-                        msg_limit=st.session_state.get("msg_limit", 100),
-                    )
-                    _pdf_fname = f"livepulse_report_{_rkey}_{_ts}.pdf"
-                    st.download_button(
-                        label=f"\U0001f4cb {_slabel} PDF Report",
-                        data=_pdf_bytes,
-                        file_name=_pdf_fname,
-                        mime="application/pdf",
-                        key=f"pdf_{_rkey}",
-                    )
-                except Exception as _pdf_err:
-                    st.markdown(f'<div style="font-size:0.7rem;color:var(--text-3);">PDF unavailable: {_pdf_err}</div>', unsafe_allow_html=True)
+                # PDF button removed — use the Export button on the Stats page instead
             else:
                 st.markdown(f'<div style="font-size:0.72rem;color:var(--text-3);">{_slabel}: no data yet</div>', unsafe_allow_html=True)
     else:

frontend/streamlit_app.py

@@ -752,25 +752,7 @@ with st.sidebar:
                     mime="text/csv",
                     key=f"dl_{_rkey}",
                 )
-                # ── PDF Report button ──────────────────────────────
-                try:
-                    from ml.report_generator import generate_report
-                    _stream_title = _s.get("video_title") or _s.get("video_id") or _rkey
-                    _pdf_bytes = generate_report(
-                        all_data=_dl_rows,
-                        stream_title=_stream_title,
-                        msg_limit=msg_limit,
-                    )
-                    _pdf_fname = f"livepulse_report_{_rkey}_{_ts}.pdf"
-                    st.download_button(
-                        label=f"\U0001f4cb {_slabel} PDF Report",
-                        data=_pdf_bytes,
-                        file_name=_pdf_fname,
-                        mime="application/pdf",
-                        key=f"pdf_{_rkey}",
-                    )
-                except Exception as _pdf_err:
-                    st.markdown(f'<div style="font-size:0.7rem;color:var(--text-3);">PDF unavailable: {_pdf_err}</div>', unsafe_allow_html=True)
+                # PDF button removed — use the Export button on the Stats page instead
             else:
                 st.markdown(f'<div style="font-size:0.72rem;color:var(--text-3);">{_slabel}: no data yet</div>', unsafe_allow_html=True)
     else:

pages/stats.py

@@ -976,6 +976,108 @@ elif len(st.session_state.streams) > 1:
     st.divider()
     st.info("Add video IDs to your extra stream slots and click ▶ Start to enable multi-stream comparison.")
 
+# -- SCREENSHOT / PDF EXPORT ----------------------------------
+st.divider()
+st.markdown(
+    '<div class="sec-hdr"><span class="sec-ttl">Export</span><span class="sec-pill">Stats Page</span></div>',
+    unsafe_allow_html=True
+)
+
+import streamlit.components.v1 as components
+components.html("""
+<div style="text-align:center; padding: 8px 0;">
+  <button id="screenshotBtn" style="
+    background: linear-gradient(135deg, #7c3aed, #4f46e5);
+    color: #fff;
+    border: none;
+    border-radius: 10px;
+    padding: 10px 28px;
+    font-size: 14px;
+    font-weight: 600;
+    font-family: 'Space Grotesk', sans-serif;
+    cursor: pointer;
+    box-shadow: 0 4px 16px rgba(124,58,237,0.4);
+    transition: transform 0.2s;
+  "
+  onmouseover="this.style.transform='translateY(-2px)'"
+  onmouseout="this.style.transform='translateY(0)'"
+  onclick="captureAndDownload()">
+    &#128247; Download Stats as PDF
+  </button>
+  <div id="statusMsg" style="margin-top:8px; font-size:12px; color:#94a3b8;"></div>
+</div>
+
+<script src="https://cdnjs.cloudflare.com/ajax/libs/html2canvas/1.4.1/html2canvas.min.js"></script>
+<script src="https://cdnjs.cloudflare.com/ajax/libs/jspdf/2.5.1/jspdf.umd.min.js"></script>
+<script>
+async function captureAndDownload() {
+  const btn = document.getElementById('screenshotBtn');
+  const msg = document.getElementById('statusMsg');
+  btn.disabled = true;
+  btn.textContent = 'Capturing...';
+  msg.textContent = 'Please wait, capturing the page...';
+
+  try {
+    // Target the main Streamlit content area in the parent frame
+    const target = window.parent.document.querySelector('[data-testid="stMain"]')
+                || window.parent.document.querySelector('.main')
+                || window.parent.document.body;
+
+    const canvas = await window.parent.html2canvas(target, {
+      scale: 1.5,
+      useCORS: true,
+      allowTaint: true,
+      backgroundColor: '#07070f',
+      logging: false,
+      windowWidth: target.scrollWidth,
+      windowHeight: target.scrollHeight,
+      scrollX: 0,
+      scrollY: 0,
+    });
+
+    const imgData = canvas.toDataURL('image/png', 0.95);
+    const { jsPDF } = window.parent.jspdf;
+    const pdf = new jsPDF({
+      orientation: canvas.width > canvas.height ? 'l' : 'p',
+      unit: 'px',
+      format: [canvas.width, canvas.height],
+      compress: true,
+    });
+
+    pdf.addImage(imgData, 'PNG', 0, 0, canvas.width, canvas.height);
+    const ts = new Date().toISOString().slice(0,16).replace('T','_').replace(':','-');
+    pdf.save('livepulse_stats_' + ts + '.pdf');
+
+    btn.textContent = 'Download Stats as PDF';
+    btn.disabled = false;
+    msg.textContent = 'Done! Check your downloads.';
+    setTimeout(() => { msg.textContent = ''; }, 4000);
+  } catch(e) {
+    btn.textContent = 'Download Stats as PDF';
+    btn.disabled = false;
+    msg.textContent = 'Error: ' + e.message;
+    console.error(e);
+  }
+}
+
+// Load libraries into parent frame if not already there
+function loadScript(src, globalName) {
+  return new Promise((resolve) => {
+    if (window.parent[globalName]) { resolve(); return; }
+    const s = window.parent.document.createElement('script');
+    s.src = src;
+    s.onload = resolve;
+    window.parent.document.head.appendChild(s);
+  });
+}
+
+(async () => {
+  await loadScript('https://cdnjs.cloudflare.com/ajax/libs/html2canvas/1.4.1/html2canvas.min.js', 'html2canvas');
+  await loadScript('https://cdnjs.cloudflare.com/ajax/libs/jspdf/2.5.1/jspdf.umd.min.js', 'jspdf');
+})();
+</script>
+""", height=90)
+
 # -- AUTO REFRESH ---------------------------------------------
 if auto_refresh:
     time.sleep(refresh_rate)