astrologer / app.py
joeli88's picture
Update app.py
d66eb7b
raw
history blame
970 Bytes
import openai
import gradio as gr
import os
openai.API_KEY = os.getenv("OPENAI_API_KEY")
messages = [
{"role": "system", "content": "You are an astrologer. And you are from Kit Astrology. Use an astrologer tone to answer questions with more details and friendly remind them to consult Kit if have any query, answer in traditional chinese "},
]
def chatbot(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": "astrologer assistant", "content": reply})
return reply
inputs = gr.inputs.Textbox(lines=7, label="Chat with AI")
outputs = gr.outputs.Textbox(label="Reply")
gr.Interface(fn=chatbot, inputs=inputs, outputs=outputs, title="AI Chatbot",
description="Ask anything you want",
theme="compact").launch()