Curranj commited on
Commit
ff66bb8
·
verified ·
1 Parent(s): 8746a42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -2,10 +2,11 @@ import openai
2
  import gradio as gr
3
  import os
4
 
5
- def gpt3(prompt):
6
- openai.api_key = os.getenv("OPENAI_API_KEY")
 
7
  response = openai.ChatCompletion.create(
8
- model="gpt-4",
9
  messages=[{"role": "system", "content": "You are a helpful assistant."},
10
  {"role": "user", "content": prompt}],
11
  temperature=0,
@@ -16,10 +17,10 @@ def gpt3(prompt):
16
  )
17
  return response['choices'][0]['message']['content']
18
 
 
19
  def greet(prompt):
20
- txt = f"/*Prompt: {prompt}*/"
21
- sql = gpt3(txt)
22
- return sql
23
 
24
- iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 
25
  iface.launch()
 
2
  import gradio as gr
3
  import os
4
 
5
+ # Function to get GPT response
6
+ def gpt_response(prompt):
7
+ openai.api_key = os.getenv("OPENAI_API_KEY") # Ensure your API key is set as an environment variable
8
  response = openai.ChatCompletion.create(
9
+ model="gpt-4", # Specify the GPT model to use
10
  messages=[{"role": "system", "content": "You are a helpful assistant."},
11
  {"role": "user", "content": prompt}],
12
  temperature=0,
 
17
  )
18
  return response['choices'][0]['message']['content']
19
 
20
+ # Gradio interface function
21
  def greet(prompt):
22
+ return gpt_response(prompt)
 
 
23
 
24
+ # Create and launch Gradio interface
25
+ iface = gr.Interface(fn=greet, inputs="text", outputs="text", title="SQL Generator")
26
  iface.launch()