// ==UserScript==
// @name         Chat Snapshot
// @namespace    ChatGPT_Plugin
// @version      1
// @author       Gavin
// @description  Adds a button to ChatGPT to take a screenshot of the conversation
// @match        https://chat.openai.com/*
// @grant        GM_download
// @require      https://cdnjs.cloudflare.com/ajax/libs/html2canvas/1.4.0/html2canvas.min.js
// ==/UserScript==

(function() {
    'use strict';

    // Add button to ChatGPT UI
    var button = document.createElement("button");
    button.innerHTML = "📸";
    button.className = "btn flex cursor-pointer absolute bottom-[124px] md:bottom-[120px] z-10 ounded-full border border-gray-200 bg-gray-50 text-gray-600 dark:border-white/10 dark:bg-white/10 dark:text-gray-200";
    button.style.right = "60px";
    button.onclick = function() {
        takeScreenshot();
    };
    document.body.appendChild(button);

    // Define function to take screenshot
    function takeScreenshot() {
        html2canvas(document.querySelector(".flex.flex-col.items-center.text-sm")).then(canvas => {
            var imageData = canvas.toDataURL();
            GM_download({
                url: imageData,
                name: "chatGPT_snapshot_"+Date.now() + ".png"
            });
        });
    }
})();