import gradio as gr
import requests
#from transformers import AutoTokenizer
#tokenizer = AutoTokenizer.from_pretrained("liam168/c2-roberta-base-finetuned-dianping-chinese")
conversation_history = ""
def chat(input):
    global conversation_history
    url = 'http://dark.21cnai.com:5000/api/chat'
    data = { "message": input }
    headers = {"Content-Type": "Application/json","Authorization":"Bearer kdfjwoieskdflasdnf"}
    user_message = f'<span class="user">您说：{input}</span><br>'
    response = requests.post(url, json=data, headers=headers)
    bot_message = f'<span class="chatbot">ChatGPT：{response.text}</span><br>'
    conversation_history = user_message + bot_message
    return conversation_history

css = '''
    .input_text, .output_text {
        font-family: Arial, sans-serif;
        font-size: 16px;
    }
    .input_text:focus {
        border: 2px solid #2C7BE5;
        outline: none;
    }
    .output_text {
        background-color: #F8F9FA;
        border: 1px solid #CED4DA;
        color: #495057;
    }
    .input_button {
        background-color: #2C7BE5;
        color: white;
        font-weight: bold;
        border: none;
    }
    .user {
        color: blue;
    }
    .chatbot {
        color: green;
    }
    .container {
        display: flex;
        flex-direction: column;
        height: 100%;
    }
    .output_section {
        flex-grow: 1;
    }
'''

iface = gr.Interface(fn=chat, 
                     outputs=gr.outputs.HTML(),
                     inputs=gr.inputs.Textbox(lines=3, placeholder="在此输入您的问题..."), 
                     title="ChatGPT 对话",
                     description="请输入您的问题，然后按回车键或单击提交。",
                     layout="vertical",
                     css=css)
iface.launch(server_name="0.0.0.0")