Upload 4 files

Dockerfile

@@ -0,0 +1,16 @@
+# Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,49 @@
+from fastapi import FastAPI, Request
+from fastapi.middleware.cors import CORSMiddleware
+import openai
+import csv
+import os
+import json
+
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+# Carregar dados das pizzas do CSV
+pizzas = []
+with open('pizzas.csv', newline='', encoding='utf-8') as csvfile:
+    reader = csv.DictReader(csvfile)
+    for row in reader:
+        pizzas.append(row)
+
+@app.post("/chat")
+async def chat(request: Request):
+    data = await request.json()
+    user_message = data.get("message")
+
+    # Consultar os dados das pizzas caso uma pizza específica seja mencionada
+    pizza_info = ""
+    for pizza in pizzas:
+        if pizza['nome'].lower() in user_message.lower():
+            pizza_info = f"{pizza['nome']} - Ingredientes: {pizza['ingredientes']}, Valor: R${pizza['valor']}\n"
+            break
+
+    # Criação da mensagem para o GPT-4
+    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:"
+
+    response = openai.Completion.create(
+        engine="gpt-4",
+        prompt=prompt,
+        max_tokens=150
+    )
+
+    reply = response.choices[0].text.strip()
+    return {"reply": reply}

pizzas.csv

@@ -0,0 +1,5 @@
+nome,ingredientes,valor
+Margherita,Molho de tomate, mussarela, manjericão,20.00
+Pepperoni,Molho de tomate, mussarela, pepperoni,25.00
+Calabresa,Molho de tomate, mussarela, calabresa, cebola,23.00
+Quatro Queijos,Molho de tomate, mussarela, gorgonzola, parmesão, provolone,28.00

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+openai
+python-dotenv