Update templates/index.html

templates/index.html

@@ -75,6 +75,39 @@
         iframe.srcdoc = data.html;
         outputEl.appendChild(iframe);
 
+        // Add Download Report button
+        const btn = document.createElement("button");
+        btn.textContent = "📥 Download Report";
+        btn.style.marginTop = "12px";
+        btn.onclick = async () => {
+          btn.disabled = true;
+          btn.textContent = "⏳ Generating report...";
+          try {
+            const resp = await fetch("/generate_report", {
+              method: "POST",
+              headers: { "Content-Type": "application/json" },
+              body: JSON.stringify({ url: data.link }),
+              });
+            if (!resp.ok) throw new Error("Failed to generate report");
+            const blob = await resp.blob();
+            const url = window.URL.createObjectURL(blob);
+            const a = document.createElement("a");
+            a.href = url;
+            a.download = "UI_Report.pdf";
+            a.click();
+            window.URL.revokeObjectURL(url);
+            btn.textContent = "📥 Download Report";
+            } catch (err) {
+            alert("Error generating report — check logs");
+            console.error(err);
+            btn.textContent = "❌ Failed";
+            } finally {
+            btn.disabled = false;
+            }
+          };
+        outputEl.appendChild(btn);
+
+
         // 🌐 Add link display for full-screen view
         if (data.link) {
           const viewLink = document.createElement("a");