Всем привет!
В этой статье я расскажу о том, как можно сохранить диалоги в ChatGPT вместе с изображениями из чата!
Несмотря на уже довольно долгое существование ChatGPT в нем до сих пор отсутствует функция сохранения диалогов с изображениями. Более того, при наличии изображений в диалоге, ссылка для сохранения вообще не создается. Именно поэтому я решил создать скрипт, который позволит сохранять диалоги напрямую через веб-интерфейс ChatGPT, используя Selenium.
Подготовка
Для начала необходимо установить необходимые зависимости. Создаем файл requirements.txt с содержимым:
undetected-chromedriver
selenium
requests
Затем устанавливаем зависимости с помощью команды (в консоли):pip install -r requirements.txt
Далее создаем файл credentials.txt, в который записываем логин и пароль для автоматического входа в ChatGPT:
myemail@example.com
mypassword
Создание скрипта
Начнем с импорта необходимых библиотек и чтения учетных данных:
import os import time import requests from undetected_chromedriver import Chrome, ChromeOptions from selenium.webdriver.common.by import By from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC # Чтение учетных данных из файла with open('credentials.txt', 'r', encoding='utf-8') as f: credentials = f.read().splitlines() email = credentials[0] password = credentials[1] Для автоматизации браузера (и обхода cloudflare) используем библу undetected_chromedriver.
Настраиваем его и открываем страницу для входа в ChatGPT:
# Настройка undetected_chromedriver options = ChromeOptions() options.add_argument('--no-sandbox') options.add_argument('--disable-dev-shm-usage') with Chrome(options=options) as driver: driver.get('https://chat.openai.com/') wait = WebDriverWait(driver, 10) login_button = wait.until(EC.element_to_be_clickable((By.XPATH, '/html/body/div[1]/div[1]/div[2]/div[1]/div/div/button[1]/div/span'))) login_button.click() # Ввод логина email_field = wait.until(EC.presence_of_element_located((By.XPATH, '/html/body/div/div/main/section/div[2]/div[1]/input'))) email_field.send_keys(email) next_button = driver.find_element(By.XPATH, '/html/body/div/div/main/section/div[2]/button') next_button.click() # Ввод пароля password_field = wait.until(EC.presence_of_element_located((By.XPATH, '/html/body/div[1]/main/section/div/div/div/form/div[1]/div/div[2]/div/input'))) password_field.send_keys(password) login_button = driver.find_element(By.XPATH, '/html/body/div[1]/main/section/div/div/div/form/div[2]/button') login_button.click() input("Выберите нужный диалог и нажмите Enter для продолжения...")Создаем папки для сохранения данных и функции для сохранения текста и скачивания файлов:
# Создание папок для сохранения данных if not os.path.exists('Диалог'): os.makedirs('Диалог/Вложения') # Функция для сохранения текста сообщения def save_text(message, filename): with open(filename, 'w', encoding='utf-8') as f: f.write(message) # Функция для скачивания файла по URL def download_file(url, dest_path): response = requests.get(url) with open(dest_path, 'wb') as f: f.write(response.content)Далее создаем основной HTML-файл для сохранения диалогов. В HTML включены стили, сделанные в цветах веб-интерфейса ChatGPT (для крОсоты):
# Создание основного HTML файла html_content = """ <!DOCTYPE html> <html lang="ru"> <head> <meta charset="UTF-8"> <title>Диалог</title> <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.3.1/styles/default.min.css"> <script src="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.3.1/highlight.min.js"></script> <script> hljs.highlightAll(); function copyCode(button) { const code = button.parentElement.querySelector('code').innerText; navigator.clipboard.writeText(code); button.innerText = 'Скопировано'; setTimeout(() => { button.innerText = 'Копировать'; }, 2000); } function showModal(imageSrc) { const modal = document.getElementById('imageModal'); const modalImg = document.getElementById('modalImg'); modal.style.display = 'block'; modalImg.src = imageSrc; } function closeModal() { const modal = document.getElementById('imageModal'); modal.style.display = 'none'; } </script> <style> body { font-family: Arial, sans-serif; line-height: 1.6; margin: 20px; background-color: #212121; color: #d1d5da; } .user-message { color: #ececec !important; background-color: #2f2f2f !important; padding: 10px; border-radius: 8px; margin-bottom: 10px; } .ai-message { color: #ececec !important; background-color: #212121; padding: 10px; border-radius: 8px; margin-bottom: 10px; } .message-container { margin-bottom: 20px; } .code-block { background-color: #000000 !important; border-left: 4px solid #00aaff; padding: 10px; position: relative; border-radius: 5px; } .code-block code { display: block; background-color: #000000 !important; color: #f8f8f2; } .copy-button { position: absolute; top: 10px; right: 10px; padding: 5px; background-color: #444654; border-radius: 8px; border: 1px solid #ccc; cursor: pointer; } .image-container { width: 200px; display: inline-block; margin: 10px; } .image-container img { width: 100%; cursor: pointer; border-radius: 5px; } .image-modal { display: none; position: fixed; z-index: 1000; padding-top: 100px; left: 0; top: 0; width: 100%; height: 100%; overflow: auto; background-color: rgba(0,0,0,0.9); } .image-modal-content { margin: auto; display: block; width: 80%; max-width: 700px; border-radius: 5px; } .image-modal-content, .image-modal { animation-name: zoom; animation-duration: 0.6s; } @keyframes zoom { from { transform: scale(0) } to { transform: scale(1) } } .close { position: absolute; top: 50px; right: 35px; color: #f1f1f1; font-size: 40px; font-weight: bold; transition: 0.3s; } .close:hover, .close:focus { color: #bbb; text-decoration: none; cursor: pointer; } </style> </head> <body> <div id="imageModal" class="image-modal"> <span class="close" onclick="closeModal()">×</span> <img class="image-modal-content" id="modalImg"> </div> """ Теперь приступаем к сбору всех сообщений и их обработке. Будем определять «авторство» сообщения и обрабатывать текст, код и изображения отдельно:
# Инициализация счетчиков total_messages = 0 user_messages = 0 ai_messages = 0 skipped_msg = 0 # Сбор всех сообщений messages = driver.find_elements(By.XPATH, '//div[contains(@class, "group") and contains(@class, "flex") and not(contains(@class, "h-48"))]') for idx, message in enumerate(messages): if not message.text.strip(): skipped_msg += 1 continue # Пропускаем пустые сообщения total_messages += 1 role = "" # Проверка, является ли автор сообщения пользователем или ИИ try: author_role_element = message.find_element(By.XPATH, './/div[@data-message-author-role]') author_role = author_role_element.get_attribute('data-message-author-role') if author_role == 'user': user_messages += 1 role = "Пользователь" role_class = "user-message" elif author_role == 'assistant': ai_messages += 1 role = "ИИ" role_class = "ai-message" else: continue # Пропускаем неизвестные роли except: continue # Пропускаем сообщения без атрибута data-message-author-role # Заменяем роль на значок для ИИ if role == "ИИ": role_icon = "<img src='../GPT_icon.png' alt='ChatGPT Icon' width='20' height='20'>" role_html = f"<p>{role_icon}</p>" else: role_html = "" html_content += f"<div class='message-container'><div class='{role_class}'>{role_html}" # Сохранение текста и кода text_elements = message.find_elements(By.XPATH, './/div[contains(@class, "whitespace-pre-wrap")]') if text_elements: text_content = text_elements[0].text save_text(text_content, f'Диалог/message_{idx}.txt') code_elements = message.find_elements(By.XPATH, './/code[contains(@class, "!whitespace-pre hljs")]') if code_elements: for code_element in code_elements: code_content = code_element.text save_text(code_content, f'Диалог/code_{idx}.txt') text_content = text_content.replace(code_content, f"<div class='code-block'><button class='copy-button' onclick='copyCode(this)'>Копировать</button><pre><code>{code_content}</code></pre></div>") html_content += f"<p>{text_content}</p>" print(f"Сообщение {idx - skipped_msg} сохранено (Текст: 1, Код: {len(code_elements)})") # Сохранение изображений img_elements = message.find_elements(By.TAG_NAME, 'img') if img_elements: for img_idx, img_element in enumerate(img_elements): img_url = img_element.get_attribute('src') img_path = f'Вложения/image_{idx}_{img_idx}.png' download_file(img_url, f'Диалог/{img_path}') html_content += f"<div class='image-container'><img src='{img_path}' onclick='showModal(\"{img_path}\")'></div>" print(f"Изображения: {len(img_elements)}", end=", ") else: print(f"Изображения: 0", end=", ") # Сохранение прикрепленных файлов file_elements = message.find_elements(By.XPATH, './/a[contains(@href, "/attachment")]') if file_elements: for file_element in file_elements: file_url = file_element.get_attribute('href') file_name = file_element.text file_path = f'Вложения/{file_name}' download_file(file_url, f'Диалог/{file_path}') html_content += f"<a href='{file_path}'>{file_name}</a>" print(f"Файлы: {len(file_elements)})") else: print(f"Файлы: 0)") html_content += "</div></div>"Не забудем корректно закрыть HTML-структуру, сохранить основной HTML-файл и вывести статистику:
# Закрытие HTML структуры html_content += """ </body> </html> """ # Сохранение основного HTML файла with open('Диалог/диалог.html', 'w', encoding='utf-8') as f: f.write(html_content) # Вывод статистики print(f"Всего сообщений: {total_messages}") print(f"Сообщений пользователя: {user_messages}") print(f"Сообщений ИИ: {ai_messages}") print("Диалог сохранен.")Ссылка на репо с исходным кодом:
https://github.com/Devastor/DevastorSaveDialogues_ChatGPT
там же можно скачать файл иконки (для обозначения ИИ в финальном диалоге)
Что можно (и нужно) улучшить
-
Добавить возможность сохранения не только изображений, но и других форматов вложений, например, таблицы xls, csv (другие вложения живут в чате недолго, в отличие от таблиц).
-
Пофиксить ошибку «OSError: [WinError 6] Неверный дескриптор», которая вылетает после сохранения диалога (у меня не получилось, хотя я не особо старался, ибо лень, и работе скрипта не мешает, это должно быть довольно просто).
-
Добавить сохранение всех диалогов аккаунта в цикле (легко).
-
Добавить сохранение названия диалога (не так тривиально, как кажется — нужно как-то придумать, каким образом ловить имя контейнера того диалога, который выбран).
-
Убрать лишнюю кнопку «Копировать» из правого верхнего угла 🙂
-
Сделать логи отключаемыми (лучше всего прописать в args)
ссылка на оригинал статьи https://habr.com/ru/articles/823428/
Добавить комментарий