diff --git a/ttf/talk.py b/ttf/talk.py index 7131534..a783d5a 100644 --- a/ttf/talk.py +++ b/ttf/talk.py @@ -5,6 +5,15 @@ def chat_with_pdf(path: str, prompt:str) -> str: + """_summary_ + + Args: + path (str): _description_ + prompt (str): _description_ + + Returns: + str: _description_ + """ reader = PdfReader(path) text = "\n".join([page.extract_text() for page in reader.pages]) @@ -18,6 +27,12 @@ def chat_with_pdf(path: str, prompt:str) -> str: @click.option("--pdf", "-p", type=click.Path(exists=True), required=True) @click.option("--prompt", "-pr", type=str, default="Summarize the following text") def main(pdf, prompt): + """_summary_ + + Args: + pdf (_type_): _description_ + prompt (_type_): _description_ + """ chat_with_pdf(pdf, prompt) if __name__ == "__main__": diff --git a/ttf/utils.py b/ttf/utils.py index 2c015df..9d162c0 100644 --- a/ttf/utils.py +++ b/ttf/utils.py @@ -8,8 +8,18 @@ # Function to send a message to the OpenAI chatbot model and return its response -def send_message(message_log): +def send_message(message_log: list) -> str: + """_summary_ + Args: + message_log (list): _description_ + + Raises: + ValueError: _description_ + + Returns: + str: _description_ + """ try: api_key = os.environ["ANTHROPIC_API_KEY"] except KeyError: @@ -27,7 +37,12 @@ def send_message(message_log): -def chat_with(user_input:str): +def chat_with(user_input:str)-> None: + """_summary_ + + Args: + user_input (str): _description_ + """ # Initialize the conversation history with a message from the chatbot # message_log = [{"role": "system", "content": "You are a helpful assistant."}] message_log = [] @@ -82,6 +97,16 @@ def chat_with(user_input:str): def summarize_pdf_old(client: anthropic.Client, path: str, prompt:str) -> str: + """_summary_ + + Args: + client (anthropic.Client): _description_ + path (str): _description_ + prompt (str): _description_ + + Returns: + str: _description_ + """ reader = PdfReader(path) text = "\n".join([page.extract_text() for page in reader.pages])