ranjangoel/GPT-PDF
5
1a
2 �Qd2 � @ sN d dl mZ d dlZddlmZmZ ddlmZ ddlm Z G dd� d�Z3dS ) � )� PdfReaderN� )� BASE_POINTS�READING_PROMT_V2)�Paper)� OpenAIModelc @ s` e Zd ZdZddedfdd�Zdd� Zd d4� Zd d � Ze d
�dd�Z5dd� Z dd� Z dd� Z
dS )� PaperReadera� 6 A class for summarizing research papers using the OpenAI API.7 8 Attributes:9 openai_key (str): The API key to use the OpenAI API.10 token_length (int): The length of text to send to the API at a time.11 model (str): The GPT model to use for summarization.12 points_to_focus (str): The key points to focus on while summarizing.13 verbose (bool): A flag to enable/disable verbose logging.14 15 i� z
gpt-3.5-turboFc C sN |t _t�|�| _d| _g | _g | _|| _d| _ || _16|| _ t ||d�| _17d S )NzXYou are a researcher helper bot. Now you need to read the summaries of a research paper.� )�api_key�model)
�openair18 r �format� init_prompt�summary_prompt�messages� summary_msgZ token_len�19keep_roundr �verboser )�selfZ20openai_keyZ token_lengthr �points_to_focusr � r �;/Users/alpaca/ChatGPT-Paper-Reader/gpt_reader/pdf_reader.py�__init__ s zPaperReader.__init__c C sR t |�| jd d d krJ|d g}tdt |��D ]}|�|| � q2|S |S d S )Nr r r � )�lenr �range�append)r �msg�new_msg�ir r r �drop_conversation) s 21z PaperReader.drop_conversationc C s | j �|�S �N)r �send_msg)r r r r r r" 3 s zPaperReader.send_msgc C sB | j �d|d�� | �| j �}| j �d|d�� | �| j �| _ |S �N�user��role�content�system)r r r" r )r �message�responser r r �_chat6 s22 zPaperReader._chat)�paperc C s� d| j d�g| _d| jd�g| _|�� D ]T\}}}td�||�� | �d�|||��}| jrdt|� | j� dd�|�d�� q(| j� ddd�� | �23| j�}|S )Nr( r% zpage: {}, part: {}u$ now I send you page {}, part {}：{}r$ z{}z,Now please make a summary of the whole paper) r r r r Z24iter_pages�printr
r+ r r r" )r r, Zpage_idxZpart_idx�text�summary�resultr r r � summarizeD s 25� zPaperReader.summarizec C s& t |�}t|�}td� | �|�}|S )Nzreading pdf finished)r r r- r1 )r Zpdf_path�26pdf_readerr, r/ r r r �read_pdf_and_summarize` s27 28z"PaperReader.read_pdf_and_summarizec C s | j S r! )r )r r r r �get_summary_of_each_partk s z$PaperReader.get_summary_of_each_partc C s4 | j �d|d�� | �| j �}| j �d|d�� |S r# )r r r" )r �questionr* r r r r5 o s zPaperReader.questionN)�__name__�29__module__� __qualname__�__doc__r r r r" r+ r r1 r3 r4 r5 r r r r r s �3031 r ) ZPyPDF2r r �promptr r r, r Zmodel_interfacer r r r r r �<module> s32