FabioSantos commited on
Commit
bcb075a
·
verified ·
1 Parent(s): 793fb4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -7
app.py CHANGED
@@ -1,15 +1,17 @@
1
  from fastapi import FastAPI, Request
2
 
3
- import openai
 
4
  import csv
5
  import os
6
  import json
7
 
8
  app = FastAPI()
9
 
10
-
11
  openai.api_key = os.getenv("OPENAI_API_KEY")
12
 
 
 
13
  # Carregar dados das pizzas do CSV
14
  pizzas = []
15
  with open('pizzas.csv', newline='', encoding='utf-8') as csvfile:
@@ -31,12 +33,15 @@ async def chat(request: Request):
31
 
32
  # Criação da mensagem para o GPT-4
33
  prompt = f"Você é um chatbot de atendimento de uma pizzaria. Se o cliente perguntar sobre uma pizza específica, forneça os detalhes. Caso contrário, responda normalmente.\nCliente: {user_message}\n{pizza_info}Chatbot:"
34
-
35
- response = openai.Completion.create(
36
- engine="gpt-4",
37
- prompt=prompt,
38
- max_tokens=150
 
 
39
  )
 
40
 
41
  reply = response.choices[0].text.strip()
42
  return {"reply": reply}
 
1
  from fastapi import FastAPI, Request
2
 
3
+
4
+ from openai import OpenAI
5
  import csv
6
  import os
7
  import json
8
 
9
  app = FastAPI()
10
 
 
11
  openai.api_key = os.getenv("OPENAI_API_KEY")
12
 
13
+ client = OpenAI()
14
+
15
  # Carregar dados das pizzas do CSV
16
  pizzas = []
17
  with open('pizzas.csv', newline='', encoding='utf-8') as csvfile:
 
33
 
34
  # Criação da mensagem para o GPT-4
35
  prompt = f"Você é um chatbot de atendimento de uma pizzaria. Se o cliente perguntar sobre uma pizza específica, forneça os detalhes. Caso contrário, responda normalmente.\nCliente: {user_message}\n{pizza_info}Chatbot:"
36
+
37
+
38
+ response = client.chat.completions.create(
39
+ model="gpt-4o",
40
+ messages=[
41
+ {"role": "user", "content": prompt}
42
+ ]
43
  )
44
+
45
 
46
  reply = response.choices[0].text.strip()
47
  return {"reply": reply}