miniChatBot / app.py
joey530's picture
Upload folder using huggingface_hub
ee2cb0d
import os
import openai
import gradio as gr
from dotenv import load_dotenv
load_dotenv()
openai.api_key = os.environ['OPENAI_API_KEY']
messages = [
{"role": "system",
"content": "你是算塔罗牌的专家,帮助人们算塔罗牌."},
]
def chat(message, history):
if message:
messages.append({"role": "user", "content": message})
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo", messages=messages
)
reply = response.choices[0].message.content
# messages.append({"role": "assistant", "content": reply})
return reply
# def chat(user_input):
# if user_input:
# messages.append({"role": "user", "content": user_input})
# response = openai.ChatCompletion.create(
# model="gpt-3.5-turbo", messages=messages
# )
# reply = response.choices[0].message.content
# messages.append({"role": "assistant", "content": reply})
# return reply
inputs = gr.inputs.Textbox(label="User input")
outputs = gr.outputs.Textbox(label="Response")
chatInterface = gr.ChatInterface(
fn=chat,
# inputs=inputs,
# outputs=outputs,
title="AI 塔罗",
)
chatInterface.launch(share=True)