import os
import gradio as gr
import openai

openai.api_base = os.environ.get("OPENAI_API_BASE")
openai.api_key = os.environ.get("OPENAI_API_KEY")
bbusr = os.environ.get("BBUSER")
bbpwd = os.environ.get("BBPWD")

messages = [{"role": "system",
             "content": "You are a friendly AI assistant and an expert in health ."}]


def ChatResponse(user_input, history):
    messages.append({"role": "user", "content": user_input})
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        stream=True,
        messages=messages,
        max_tokens=512, 
        temperature=0.5, 
        top_p=0.95
    )
    partial_response = ""
    for stream_response in response:
        token = stream_response["choices"][0]["delta"].get("content", "")
        partial_response += token
        yield partial_response


mychatbot = gr.Chatbot(
    avatar_images=["./user.png", "./aibot.png"], bubble_full_width=False, show_label=False, show_copy_button=True,)


demo = gr.ChatInterface(fn=ChatResponse, 
        chatbot=mychatbot,
        title="🫐BuruBuru Chat🫐",
        retry_btn=None,
        undo_btn=None
    )

demo.queue().launch(auth=(bbusr, bbpwd),show_api=False)