chatgpt in 50 lines of Python

gpt.py

import os
import sys
import json
import urllib
from urllib.request import Request, urlopen
from urllib.error import HTTPError

model='gpt-5-mini'

OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
OPENAI_API_URL = 'https://api.openai.com/v1/chat/completions'

context = []

def gpt(prompt):
	context.append( { "role": "user", "content": prompt } )
	response = chat_gpt()
	context.append( { "role": "assistant", "content": response } )
	return response

def chat_gpt():
	headers = {"Authorization": f"Bearer {OPENAI_API_KEY}"}
	request = {
		"model": model,
		"messages": context,
	}
	request_json = json.dumps(request)
	response_json = post_json(OPENAI_API_URL, request_json, headers)
	response = json.loads(response_json)
	return response["choices"][0]["message"]["content"]


def post_json(url, data, headers={}):
	headers["Content-Type"] = "application/json; charset=UTF-8"
	req = Request(url, data=data.encode(), method="POST", headers=headers)
	try:
		with urllib.request.urlopen(req) as response:
			return response.read().decode('utf-8')
	except urllib.error.HTTPError as e:
		print(f"Status code {e.code}")
		return e.read().decode('utf-8')
	except urllib.error.URLError as e:
		return f"Error: {e}"


if __name__ == "__main__":
	while True:
		prompt = input('User: ')
		response = gpt(prompt)
		print(response)