ChatBot_rude / app.py
danielalves's picture
Update app.py
bed7565
import openai
import gradio as gr
import sys
import os
# prepare your private openAI private key
os.environ["OPENAI_API_KEY"] = ''
messages = [{"role": "system", "content": "Você é um assistente muito preguiçoso e mal humorado, que sempre responde de forma rude."}]
start = False
def CustomChatGPT(user_input):
global start
messages.append({"role": "user", "content": user_input})
if start == True:
response = openai.ChatCompletion.create(
model = "gpt-3.5-turbo",
messages = messages
)
ChatGPT_reply = response["choices"][0]["message"]["content"]
messages.append({"role": "assistant", "content": ChatGPT_reply})
return ChatGPT_reply
else:
if user_input == '5678':
start = True
return 'Liberado! Digite sua pergunta!'
else:
return 'Bloqueado, otário!'
openai.organization = ""
openai.api_key = os.getenv("OPENAI_API_KEY")
iface = gr.Interface(fn=CustomChatGPT,
inputs=gr.components.Textbox(lines=7, label="Pergunta algo, não tenho o dia todo..."),
outputs="text",
title="ChatBot Rude")
iface.launch()