import openai
import numpy as np
import os
import json
import gradio as gr
 
openai.api_key = os.environ["api_key"]
engine = os.environ["engine"]


def happytt(temperature,max_tokens,text,stop):
  try:
    s = json.loads(stop)
  except json.JSONDecodeError:
    s = []
  
  response = openai.Completion.create(
    engine=engine,
    prompt=text,
    temperature=temperature,
    max_tokens=max_tokens,
    top_p=1,
    frequency_penalty=0,
    presence_penalty=0,
    stop=s
)

  return response.choices[0].text


title = "OpenAI Codex"
description = '''OpenAI Codex is an artificial intelligence model developed by OpenAI. 
It parses natural language and generates code in response. 
It is used to power GitHub Copilot, a programming autocompletion 
tool developed for  Code generation.

Try following prompts and tweak temperatures  in following links -

https://www.pragnakalp.com/experimenting-with-openai-codex/

https://betterprogramming.pub/i-beta-tested-openais-codex-and-the-results-are-spooky-good-e282a1874c79

https://beta.openai.com/examples?category=code'''
 
 
iface = gr.Interface( happytt,[ gr.inputs.Slider(0, 1, step=0.1),gr.inputs.Slider(150, 4000, step=1),
                               gr.inputs.Textbox(type='str',   
                                                 label="input prompt"),
                               gr.inputs.Textbox(type='str',   
                                                 label="list of tokens, when to finish generating")],"text", title = title, description = description )
iface.launch(debug=True)