astrologer / app.py
joeli88's picture
Update app.py
048201f
raw
history blame
959 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": "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()