Update app.py

app.py

@@ -1,25 +1,26 @@
-import openai
-import gradio as gr
 import os
+from openai import OpenAI
+import gradio as gr
+
+client = OpenAI(api_key=os.environ["SECRET"])
 
-# Function to get GPT response using text completions
-def gpt_response(prompt):
-    openai.api_key = os.getenv("OPENAI_API_KEY")  # Ensure your API key is set as an environment variable
-    response = openai.Completion.create(
-        model="text-davinci-003",  # Specify a non-chat model
-        prompt=prompt,
+def gpt3(texts):
+    response = client.chat.completions.create(
+        model="gpt-4-turbo-preview",
+        messages=[{"role": "user", "content": texts}],
         temperature=0,
         max_tokens=750,
         top_p=1,
         frequency_penalty=0.0,
-        presence_penalty=0.0
+        presence_penalty=0.0,
+        stop=[";", "*/", "</code>"]
     )
-    return response['choices'][0]['text'].strip()
+    return response.choices[0].message.content
 
-# Gradio interface function
 def greet(prompt):
-    return gpt_response(prompt)
+    txt = f'''/*Prompt: {prompt}*/ \n'''
+    sql = gpt3(txt)
+    return sql
 
-# Create and launch Gradio interface
-iface = gr.Interface(fn=greet, inputs="text", outputs="text", title="SQL Generator")
+iface = gr.Interface(greet, inputs=["text"], outputs="text")
 iface.launch()