No more "Untitled12-final-final.ipynb"! A python script that ingests Jupyter Notebook cells and comes up with 10 relevant names to give it.

jupyter_namer.py

# No more "Untitled12-final-final.ipynb"! 
# A python script that ingests Jupyter Notebook cells and comes up with 10 relevant names to give it.

import nbformat
import openai
import argparse

def extract_cell_contents(notebook_path):
    with open(notebook_path, "r", encoding="utf-8") as f:
        notebook = nbformat.read(f, nbformat.NO_CONVERT)

    cell_contents = []
    for cell in notebook.cells:
        if cell.cell_type == "code":
            cell_contents.append(cell.source)
        elif cell.cell_type == "markdown":
            cell_contents.append(cell.source)

    return cell_contents

def generate_text(prompt):
    response = openai.ChatCompletion.create(
        model='gpt-4',
        messages=[{"role": "system", "content": "You are a helpful assistant."},
                  {"role": "user", "content": prompt}],
        temperature=0.8,
    )
    
    return response


if __name__ == "__main__":
    # take notebook path as argument
    parser = argparse.ArgumentParser()
    parser.add_argument("nb_path", help="path to the notebook file")
    args = parser.parse_args()
    
    # Set up your OpenAI API credentials
    openai.api_key = 'OPENAI-API_KEY'
    
    # extract cell contents
    contents = extract_cell_contents(args.nb_path)
    
    cells = "\n".join(contents)
    prompt = f"I need you to list 10 ideas of relevant names for the Jupyter Notebook that contains the cells I enclosed by three backticks. The name must be specific and descriptive without being verbose. ```{cells}```"
    
    # generate text
    generated_text = generate_text(prompt)
    
    # print the generated text
    print(generated_text['choices'][0]['message']['content'])
    

Demo