Rathapoom commited on
Commit
e2aceb6
·
verified ·
1 Parent(s): 049e1fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -3,14 +3,21 @@ import openai
3
  import os
4
 
5
  # ใส่ API Key
6
- openai.api_key = os.getenv("OPENAI_API_KEY")
 
 
7
 
8
 
9
  def generate_response(prompt):
10
  try:
11
- response = openai.ChatCompletion.create(
12
- model="gpt-4o-mini", # หรือ gpt-4 หากต้องการ
13
- messages=[{"role": "user", "content": prompt}]
 
 
 
 
 
14
  )
15
  return response.choices[0].message["content"]
16
  except Exception as e:
 
3
  import os
4
 
5
  # ใส่ API Key
6
+ client = OpenAI(
7
+ api_key=os.environ.get("OPENAI_API_KEY"), # This is the default and can be omitted
8
+ )
9
 
10
 
11
  def generate_response(prompt):
12
  try:
13
+ response = client.chat.completions.create(
14
+ messages=[
15
+ {
16
+ "role": "user",
17
+ "content": "Say this is a test",
18
+ }
19
+ ],
20
+ model="gpt-4o-mini",
21
  )
22
  return response.choices[0].message["content"]
23
  except Exception as e: