Words_To_SQL / app.py
Curranj's picture
Update app.py
72a6792 verified
raw
history blame
771 Bytes
import os
from openai import OpenAI
import gradio as gr
client = OpenAI(api_key=os.environ["Secret"])
def gpt3(texts):
response = client.chat.completions.create(
model="gpt-4",
messages=[{"role": "user", "content": texts}],
temperature=0,
max_tokens=750,
top_p=1,
frequency_penalty=0.0,
presence_penalty=0.0,
stop=[";", "*/", "</code>"]
)
return response.choices[0].message.content
def greet(prompt):
txt = f'''/*Prompt: {prompt}*/ \nSQL code:'''
sql = gpt3(txt)
return sql
e = gr.Interface(
fn=greet,
inputs=["text"],
outputs="text",
title="Natural Language to SQL Converter",
description="Convert your natural language queries into SQL code."
)
e.launch()