import openai
import gradio as gr
import os

def gpt3(prompt):
    openai.api_key = os.getenv("OPENAI_API_KEY")
    response = openai.ChatCompletion.create(
        model="gpt-4",
        messages=[{"role": "system", "content": "You are a helpful assistant."},
                  {"role": "user", "content": prompt}],
        temperature=0,
        max_tokens=750,
        top_p=1,
        frequency_penalty=0.0,
        presence_penalty=0.0
    )
    return response['choices'][0]['message']['content']

def greet(prompt):
    txt = f"/*Prompt: {prompt}*/"
    sql = gpt3(txt)
    return sql

iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch()