Words_To_SQL / app.py
Curranj's picture
Update app.py
ff66bb8 verified
raw
history blame
848 Bytes
import openai
import gradio as gr
import os
# Function to get GPT response
def gpt_response(prompt):
openai.api_key = os.getenv("OPENAI_API_KEY") # Ensure your API key is set as an environment variable
response = openai.ChatCompletion.create(
model="gpt-4", # Specify the GPT model to use
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']
# Gradio interface function
def greet(prompt):
return gpt_response(prompt)
# Create and launch Gradio interface
iface = gr.Interface(fn=greet, inputs="text", outputs="text", title="SQL Generator")
iface.launch()