File size: 1,274 Bytes
defb736 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import os
import openai
import gradio as gr
#import requests
#from dotenv import load_dotenv
api_key = os.environ.get("OPENAI_API_KEY")
#def configure():
#load_dotenv()
#openai.api_key = os.getenv
start_sequence = "\nAI:"
restart_sequence = "\nHuman: "
# Define the conversation prompt
conversation_prompt = "Welcome to ChatRobo, kindly type in your enquiries: "
# Define the function to generate a response from the GPT model
def openai_chat_history(input, history):
history = history or []
if input.strip() != "":
s = list(sum(history, ()))
s.append(input)
inp = ' '.join(s)
output = openai_create(inp)
history.append((input, output))
return history[-1][1]
else:
return ""
def openai_create(prompt):
response = openai.Completion.create(
model="text-davinci-003",
prompt=prompt,
temperature=0.9,
max_tokens=150,
top_p=1,
frequency_penalty=0,
presence_penalty=0.6,
stop=["Human:", "AI:"]
)
return response.choices[0].text
block = gr.Interface(
fn=openai_chat_history,
inputs=[gr.inputs.Textbox(placeholder=conversation_prompt)],
outputs=[gr.outputs.Textbox(label="ChatRobo Output")]
)
block.launch()
|