import gradio as gr
import json
import asyncio
import os
from EdgeGPT import Chatbot, ConversationStyle

#read cookie from local file
# with open('./cookies.json', 'r') as f:
#     cookies = json.load(f)

async def get_model_reply(prompt,style,cookies,context=[]):
    # combines the new question with a previous context
    context += [prompt]
    cookies = json.loads(cookies)
    # given the most recent context (4096 characters)
    # continue the text up to 2048 tokens ~ 8192 charaters
    bot = Chatbot(cookies=cookies)
    raw_data = await bot.ask(prompt, conversation_style=style)
    await bot.close()
    #print(raw_data)
    response = raw_data["item"]["messages"][1]["text"]
    context += [response]

    # list of (user, bot) responses. We will use this format later
    responses = [(u, b) for u, b in zip(context[::2], context[1::2])]

    return responses, context
# query = 'Which is the largest country by area in the world?'
# style="precise"
# responses, context =asyncio.run(get_model_reply(query,style,context=[]))
#
# print(' ' + responses[-1][0])
# print(' ' + responses[-1][1])
with gr.Blocks() as dialog_app:
    with gr.Tab("Cookies"):
        cookies = gr.Textbox(lines=2, label="输入bing.com中的cookies")
    with gr.Tab("New Bing Chat"):
        gr.Markdown("# A Simple Web to use New Bing Without Magic")
        chatbot = gr.Chatbot()
        state = gr.State([])
        markdown = gr.Markdown(label="Output")

        with gr.Row():
            inputs = gr.Textbox(
                label="输入问题",
                placeholder="输入文本，然后按回车键"
            )
            style = gr.Dropdown(label="回答倾向", choices=["创造", "平衡", "精确"], multiselect=False,
                                value="balanced", type="value")

        inputs.submit(get_model_reply, [inputs, style, cookies, state], [chatbot, state])
        send = gr.Button("Send")
        send.click(get_model_reply, [inputs, style, cookies, state], [chatbot, state])

# launches the app in a new local port
dialog_app.launch()
# 为网站设置密码防止滥用
# dialog_app.launch(auth=("admin", "pass1234"))