File size: 641 Bytes
4315cb9
424191b
 
 
a9d0589
1f79e6d
424191b
 
7253976
424191b
f7af7f0
 
 
 
424191b
 
102e098
424191b
102e098
c6590cd
ebf8e7d
424191b
 
102e098
424191b
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 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

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