rmacdonn/Text2SQL

1import openai2import gradio as gr3import os4 5# OpenAi call6def gpt3(prompt):7    response = openai.Completion.create(8        engine="text-davinci-003",9        prompt=prompt,10        temperature=0,11        max_tokens=750,12        top_p=1,13        frequency_penalty=0.0,14        presence_penalty=0.0,15        stop=(";", "/*", "</code>")16    )17    x = response.choices[0].text18    return x19 20# Function to elicit sql response from model21def greet(prompt, variable_names, table_name):22    txt = f'''/* Prompt: {prompt}\nVariable Names: {variable_names}\nSchema Name: {table_name} */\n---SQL Code:\n'''23    sql = gpt3(txt)24    return sql25 26# Code to set up Gradio UI27iface = gr.Interface(28    greet,29    inputs=[30        gr.inputs.Textbox(label="Prompt"),31        gr.inputs.Textbox(label="Variable Names"),32        gr.inputs.Textbox(label="Table Name")33    ],34    outputs="text",35    title="Text2SQL",36    description="Enter a prompt, variable names, and a table name to return an SQL script"37)38iface.launch()39