OpenAI Codex for refactoring in Emacs

codex-refactor.el

(defun codex-refactor (&optional b e) 
  (interactive "r")
  (shell-command-on-region b e "codex-refactor.py" nil nil))

codex-refactor.py

#!/usr/bin/env python

# Make this file executable (chmod +x codex.py)
# and put it somewhere in your PATH

import sys
import openai

openai.api_key = "sk-[YOUR API KEY HERE]"

code = "".join(sys.stdin.readlines())

prompt = f"""
## Naive implementation:

{code}


## A more elegant way to implement this:"""

completion = openai.Completion.create(
  engine="davinci-codex",
  prompt=prompt,
  max_tokens=int(len(code)/2),
  stop=["\n\n\n"],
  temperature=0,
  frequency_penalty=0.2
)

print(completion.choices[0].text)