astrologer / app.py
joeli88's picture
Update app.py
40a4d12
raw
history blame
940 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 question 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": "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()