Carlosjr5
Add application file
78c0796
raw
history blame
1.4 kB
import os
#pip3 install openai
import openai as oai
#pip3 install gradio
import gradio as gr
from data import OPENAI_API_KEY
#if you have OpenAI API key as an environment variable, enable the below
#openai.api_key = os.getenv("OPENAI_API_KEY")
#if you have OpenAI API key as a string, enable the below
OPEN_AI_KEY = OPENAI_API_KEY
oai.api_key = OPEN_AI_KEY
start_sequence = "\nAI:"
restart_sequence = "\nHuman: "
prompt = "Feel free to ask any question to this bot, it can find you many solutions.\n\n Tell me a story about human beings... "
def openai_create(prompt):
response = oai.Completion.create(
model="text-davinci-003",
prompt=prompt,
temperature=0.9,
max_tokens=150,
top_p=0,
frequency_penalty=0,
presence_penalty=0.6,
stop=[" Human:", " AI:"]
)
return response.choices[0].text
def cjrSmartBot(input, history):
history = history or []
s = list(sum(history, ()))
s.append(input)
inp = ' '.join(s)
output = openai_create(inp)
history.append((input, output))
return history, history
block = gr.Blocks()
with block:
gr.Markdown("""<h1><center>CarlosJR Smart Bot</center></h1>""")
chatbot = gr.Chatbot()
message = gr.Textbox(placeholder=prompt)
state = gr.State()
submit = gr.Button("SEND")
submit.click(cjrSmartBot, inputs=[message, state], outputs=[chatbot, state])
block.launch()