Python: добавление PDF во вход API ассистента с использованием ChatGPT

В современной цифровой эпохе интеграция различных форматов файлов с искусственными интеллектуальными технологиями открывает множество возможностей для автоматизации и улучшенной обработки данных. Среди этих интеграций особое внимание заслуживает возможность обработки PDF-документов с использованием ChatGPT от OpenAI за его потенциал изменить способы взаимодействия и анализа текстовых данных. В данной статье рассматривается методология интеграции PDF-документов во вход для API ассистента с помощью Python, включая функциональный мини-проект для иллюстрации процесса.

Предварительные требования

Перед началом убедитесь, что у вас есть следующее:

• Установленный Python на вашей системе.
• Активный ключ API OpenAI.
• Базовое знакомство с Python и обработкой PDF-файлов.

ОНЛАЙН-ПРАКТИКУМ

ЗАПУСК нейросети DEEPSEEK R1 ЛОКАЛЬНО НА СВОЕМ КОМПЬЮТЕРЕ

ЧТО БУДЕТ НА ОБУЧЕНИИ?

• ПОКАЖЕМ, КАК РАЗВЕРНУТЬ МОДЕЛЬ нейросети DEEPSEEK R1 ПРЯМО НА СВОЁМ КОМПЬЮТЕРЕ
• Где и как применять? Потестируем модель после установки на разных задачах
• Как дообучить модель под себя?

Участвовать бесплатно

Настройка среды

• Установка необходимых библиотек: начните с установки библиотеки openai для взаимодействия с API OpenAI и PyMuPDF (также известной как fitz) для обработки PDF-файлов в Python. Выполните следующие команды в вашем терминале:

pip install openai

pip install PyMuPDF

• Настройка ключа API: сохраните ваш ключ API OpenAI. Рекомендуется использовать переменные среды для этой цели. Добавьте следующее в ваш .bashrc или .bash_profile файл:

  export OPENAI_API_KEY='ваш_ключ_api_здесь'

Замените ваш_ключ_api_здесь на ваш реальный ключ API OpenAI. После добавления строки, перезагрузите настройки вашей оболочки.

Обработка PDF

Чтобы передать PDF-документ в ChatGPT через API ассистента, сначала необходимо извлечь текст из него. Вот как это сделать:

• Чтение PDF-файла: используйте библиотеку fitz для открытия и чтения файла. Вот базовый фрагмент для извлечения текста:

import fitz # Импорт библиотеки

def extract_text_from_pdf(pdf_path):

doc = fitz.open(pdf_path)

text = ""

for page in doc: # Итерируем по каждой странице

text += page.get_text()

doc.close()

return text

• Предварительная обработка текста (по желанию): в зависимости от ваших требований, возможно, потребуется предварительная обработка извлеченного текста для удаления ненужных элементов, таких как заголовки, подвалы или специальные символы.

Взаимодействие с API ассистента

После извлечения и, при необходимости, предварительной обработки текста из вашего PDF-документа, следующим шагом будет отправка этого текста в API ассистента. Вот простой пример:

import openai

def query_chatgpt(text):

response = openai.Completion.create(

engine="text-davinci-003",

prompt=text,

temperature=0.5,

max_tokens=100,

top_p=1.0,

frequency_penalty=0.0,

presence_penalty=0.0

)

return response.choices[0].text.strip()

Мини-проект: суммаризатор PDF для ChatGPT

Давайте объединим всё в мини-проект, который извлекает текст из файла и получает краткое изложение от ChatGPT.

import os

import fitz

import openai

# Извлечение текста из PDF

def extract_text_from_pdf(pdf_path):

doc = fitz.open(pdf_path)

text = ""

for page in doc:

text += page.get_text()

doc.close()

return text

# Запрос к ChatGPT для получения краткого изложения

def get_summary_from_chatgpt(text):

openai.api_key = os.getenv("OPENAI_API_KEY")

response = openai.Completion.create(

engine="text-davinci-003",

prompt=f"Суммируйте этот документ:\n{text}",

temperature=0.5,

max_tokens=150,

top_p=1.0,

frequency_penalty=0.0,

presence_penalty=0.0

)

return response.choices[0].text.strip()

# Основная функция

def main(pdf_path):

text = extract_text_from_pdf(pdf_path)

summary = get_summary_from_chatgpt(text)

print("Краткое изложение:\n", summary)

if __name__ == "__main__":

pdf_path = "путь_к_вашему_файлу.pdf" # Замените это на путь к файлу

main(pdf_path)

Заключение

Интеграция PDF-документов во вход API ассистента с использованием Python открывает новые перспективы для использования искусственного интеллекта в обработке и анализе документов. Следуя описанным выше шагам, разработчики могут улучшить свои приложения, добавив возможность интерпретации и анализа содержимого PDF-файлов через призму ChatGPT. Это не только расширяет область потенциального применения, но и значительно повышает эффективность в задачах, связанных с управлением и пониманием документов.