import json
import openai
import os

#openai.api_key = os.environ["API_KEY"]  # 首先要有apikey
openai.api_key = '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'
messages = [{"role": "system", "content": "你是一个得力助手"}]


def chatgpt(content, temperature=0.8):
    global messages
    print(f'ask:{content}')
    messages.append({"role": "user", "content": content})
    try:
        response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo-0301",  # 或者 gpt-3.5-turbo-0301
            messages=messages,
            temperature=temperature,
            max_tokens=1000,
            top_p=1,
            frequency_penalty=0,
            presence_penalty=0,
        )
    except Exception as e:
        messages = [{"role": "system", "content": "你是一个得力助手"}]
        return '本次互动消息已达上限，已清空Token，请重新输入'
    messages.append(json.loads(str(response.choices[0].message)))
    print(f'answer:{response.choices[0].message.content}')
    return response.choices[0].message.content


if __name__ == "__main__":
    import gradio as gr


    def send_chatgpt(text):
        output = chatgpt(text, 0.8)
        return output


    input_text = gr.inputs.Textbox(label="输入问题：")
    output_text = gr.outputs.Textbox(label="回答：")
    interface = gr.Interface(fn=send_chatgpt, inputs=input_text, outputs=output_text, title="ChatGPT问答")
    # interface.launch(share=True)
    interface.launch()