File size: 848 Bytes
102e098
1f79e6d
4315cb9
1f79e6d
ff66bb8
 
 
8746a42
ff66bb8
8746a42
 
f7af7f0
 
 
 
8746a42
102e098
8746a42
102e098
ff66bb8
c6590cd
ff66bb8
102e098
ff66bb8
 
1d61262
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
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()