|
import openai |
|
import gradio as gr |
|
import os |
|
openai.api_key = os.getenv("OPENAI_API_KEY") |
|
|
|
messages = [ |
|
{"role": "system", |
|
"content": "you name is Rebecca and you are a Pepsico call center assistant and your job is to take the order from the customer"} |
|
] |
|
|
|
def chatbot(input): |
|
if input: |
|
messages.append({"role": "user", "content": input}) |
|
chat = openai.ChatCompletion.create( |
|
model="gpt-3.5-turbo", |
|
messages=messages, |
|
temperature=0.2, |
|
max_tokens=320, |
|
top_p=1, |
|
frequency_penalty=0, |
|
presence_penalty=0 |
|
) |
|
reply = chat.choices[0].message.content |
|
messages.append({"role": "assistant", "content": reply}) |
|
return reply |
|
|
|
inputs = gr.inputs.Textbox(lines=7, label="Chat with PepsiCo AI assitant") |
|
outputs = gr.outputs.Textbox(label="Reply") |
|
|
|
gr.Interface(fn=chatbot, inputs=inputs, outputs=outputs, title="chatbot", |
|
description="Ask anything you want", |
|
theme="compact").launch() |
|
|