|
import openai,os |
|
import gradio as gr |
|
from gradio.components import Textbox |
|
|
|
openai.api_key = 'OPENAI_API_KEY' |
|
|
|
messages = [ |
|
{"role": "system", "content": "You are a helpful and kind AI Assistant."}, |
|
] |
|
|
|
def ChatGPT_Bot(input): |
|
if input: |
|
messages.append({"role": "user", "content": input}) |
|
chat = openai.ChatCompletion.create( |
|
model="gpt-3.5-turbo", messages=messages |
|
) |
|
reply = chat.choices[0].message.content |
|
messages.append({"role": "assistant", "content": reply}) |
|
return reply |
|
|
|
inputs = Textbox(lines=7, label="请输入你的问题") |
|
outputs = Textbox(lines=7, label="来自ChatGPT的回答") |
|
|
|
gr.Interface(fn=ChatGPT_Bot, inputs=inputs, outputs=outputs, title="ChatGPT AI助理", |
|
description="我是您的AI助理,您可以问任何你想知道的问题", |
|
theme=gr.themes.Default()).launch() |
|
|