import os
import time

import gradio as gr
import openai

openai.api_key = os.environ["OPENAI_API_KEY"]
openai.organization = os.environ["Organization_KEY"]
zhanghao = os.environ["zhanghao"]
mima = os.environ["mima"]

system_message = {"role": "system", "content": "You are a helpful assistant."}


def user(user_message, history):
    return "", history + [[user_message, None]]


def bot_3(history, messages_history):
    model = "gpt-3.5-turbo"
    user_message = history[-1][0]
    bot_message, messages_history = ask_gpt(user_message, messages_history, model)
    messages_history += [{"role": "assistant", "content": bot_message}]
    history[-1][1] = bot_message
    time.sleep(1)
    return history, messages_history


def ask_gpt(message, messages_history, model):
    messages_history += [{"role": "user", "content": message}]
    response = openai.ChatCompletion.create(
        model=model,
        messages=messages_history
    )
    return response['choices'][0]['message']['content'], messages_history


def init_history(messages_history):
    messages_history = []
    messages_history += [system_message]
    return messages_history


with gr.Blocks() as demo:
    gr.Markdown("ChatGPT非科学接入")
    with gr.Tab("GPT-3.5"):
        chatbot = gr.Chatbot(label ='对话记录')
        msg = gr.Textbox(label ='输入框')

        sub = gr.Button("发送")
        clear = gr.Button("清除")
        state = gr.State([])

        msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
            bot_3, [chatbot, state], [chatbot, state]
        )
        sub.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
            bot_3, [chatbot, state], [chatbot, state])
        clear.click(lambda: None, None, chatbot, queue=False).success(init_history, [state], [state])

demo.launch(auth_message = "账号密码发送到了微信上了",auth=(zhanghao, mima))