app.py

import gradio as gr
import random
import time

css = '''
.show img {height: 320px}
.result img {height: 320px}
.qrcode img {height: 225px}
.btn {
    justify-content: normal
}
'''

def get_new_question_and_options():
    # time.sleep(1.5)
    number = random.randint(1,65536)
    return {
        "question_text": f"问题{number}是这样的。",
        "option_A": f"问题{number}给出的选项A，吃饭",
        "option_B": f"问题{number}给出的选项B，睡觉",
        "option_C": f"问题{number}给出的选项C，打豆豆",
    }

def get_option_result(question_text, option_text):
    # time.sleep(1.5)
    numbers = [
        random.randint(-10,10),
        random.randint(-10,10),
        random.randint(-10,10),
        random.randint(-10,10)
    ]
        
    return {
        "result_text": f"问题“{question_text}”执行“{option_text}”选项，结果完成了",
        "score_text": f"导致四个维度增减情况为 {numbers[0]} {numbers[1]} {numbers[2]} {numbers[3]}",
        "deltas": numbers    
    } 

def typePUA(a):

    x = '0' if a[0] <= 50 else '1'
    y = '0' if a[1] <= 50 else '1'
    z = '0' if a[2] <= 50 else '1'
    w = '0' if a[3] <= 50 else '1'
    
    puatype = ''.join([x,y,z,w])

    pua = {
        "0000": "1 反PUA斗士",
        "0001": "2",
        "0010": "3",
        "0011": "4",
        "0100": "5",
        "0101": "6",
        "0110": "7",
        "0111": "8",
        "1000": "9",
        "1001": "10",
        "1010": "11",
        "1011": "12",
        "1100": "13",
        "1101": "14",
        "1110": "15",
        "1111": "16 天生牛马打工人",
    }

    return pua[puatype]

with gr.Blocks(css=css).queue() as demo:
            
    data = gr.State({
        "total": 3,
        "isComfirmed": False,
        "canOptionA": True,
        "canOptionB": True,
        "canOptionC": True,
        "canOptionD": False,
        "canFillD": True,
        "canRestart": True,
        "canRefresh": True,
        "canNext": False,
    })

    gr.Markdown("# 😅别再PUA啦：基于多智能体的被PUA指数心理测评游戏")
    gr.Markdown("### 😋机智选择，远离PUA😈")

    with gr.Tab("测评"):

        with gr.Column():
            with gr.Row():
                with gr.Column():

                    hello = gr.Markdown("👇 请输入昵称")
                    with gr.Group():
                        with gr.Row():
                            user_name = gr.Textbox(placeholder="输入您的昵称", value="", container=False, scale=6)
                            confirm = gr.Button(value="确认", scale=1)

                    progress_slider = gr.Slider(label=f"测评进度（共{data.value['total']}题）", value=0, minimum=0, maximum=data.value["total"], step=1, interactive=False)
                
                with gr.Column():
                    gr.Markdown("当前指数")
                    with gr.Group():
                        with gr.Row():
                            dim1_slider = gr.Slider(label="维度1", value=50, minimum=0, maximum=100, step=1, interactive=False)
                            dim2_slider = gr.Slider(label="维度2", value=50, minimum=0, maximum=100, step=1, interactive=False)
                        with gr.Row():
                            dim3_slider = gr.Slider(label="维度3", value=50, minimum=0, maximum=100, step=1, interactive=False)
                            dim4_slider = gr.Slider(label="维度4", value=50, minimum=0, maximum=100, step=1, interactive=False)
            
            gr.Markdown("---")

            with gr.Row():

                with gr.Column():
                    question_text = gr.TextArea(label="事件背景", value="", placeholder="请等待生成", interactive=False, lines=4)
                    result_text   = gr.TextArea(label="事件结果", value="", placeholder="请根据事件背景，结合个人实际情况做出选择。注意，选择之间并无对错之分。", interactive=False, lines=3)
                    score_text    = gr.TextArea(label="指数变更", value="", placeholder="请根据事件背景，结合个人实际情况做出选择。注意，选择之间并无对错之分。", interactive=False, lines=3)
                
                with gr.Column():
                    option_A_button = gr.Button(value=f"A: ", elem_classes="btn", interactive=False)
                    option_B_button = gr.Button(value=f"B: ", elem_classes="btn", interactive=False)
                    option_C_button = gr.Button(value=f"C: ", elem_classes="btn", interactive=False)
                    
                    with gr.Group():
                        option_D_button = gr.Button(value="D: 我选这个", elem_classes="btn", interactive=False)
                        option_D_text = gr.Textbox(show_label=False, value="", placeholder="我有独特的主见", container=False, interactive=False)


                    with gr.Column():
                        
                        restart  = gr.Button(value="▶️ 开始", interactive=False, variant="primary")
                        refresh  = gr.Button(value="🔄️ 刷新", interactive=False)
                        next_btn = gr.Button(value="➡️ 下一题", interactive=False)
                        
        gr.Markdown("---")
        gr.Markdown("### 测评结果")
        with gr.Row():
            gr.Image(elem_classes="result", value="./src/home.png", label="测评结果图示", interactive=False, scale=0.8)
            with gr.Group():
                with gr.Row():
                    gr.Number(label="您的被PUA指数", value=50, interactive=False)
                    gr.Textbox(label="您的被PUA类型", value="测评结束后生成", interactive=False, placeholder="测评结束后生成")
                gr.TextArea(label="您的测评结果解释", placeholder="测评结束后生成", interactive=False, lines=5)
                gr.TextArea(label="可能的解决方案", placeholder="测评结束后生成", interactive=False, lines=5)

        gr.Button(value="如果对您有用，欢迎分享网站与测评结果 🥰", variant="primary")

    # 历史测评结果
    with gr.Tab("历史"):
        pass

    # 测评原理解释
    with gr.Tab("说明"):
        pass

    # 作者声明
    with gr.Tab("关于"):
        gr.Markdown("")
        gr.Markdown("**Datawhale AI夏令营（第四期）x 浪潮信息 “源”大模型应用开发活动**")
        gr.Markdown("---")
        gr.Markdown("**项目名称**：别再PUA啦：基于多智能体的被PUA指数心理测评游戏")
        gr.Markdown("**项目Slogan**：机智选择，远离PUA")
        gr.Markdown("**项目简介**：“别再PUA啦”是一款创新的心理测评游戏，它基于多智能体技术，旨在通过互动游戏的形式，测量个体对PUA策略的抵抗力和识别能力，从而帮助用户提高自我保护意识和社交技巧。")
        gr.Markdown("**项目技术栈**：Yuan 2.0，RAG，Langchain，Muiti-Agent, Gradio")
        gr.Markdown("**项目演示视频**：【TODO】")
        gr.Markdown("**项目构建分析与复盘博客**：【TODO】")
        gr.Markdown("---")
        gr.Markdown("**作者**：胡钧耀（南开大学计算机视觉在读一年级直博生，个人研究兴趣方向为视觉AIGC）")
        gr.Markdown("**联系方式**：微信（LittleDream_hjy），[Bilibili](https://space.bilibili.com/2042113)，[GitHub](https://github.com/JunyaoHu)，[个人主页](https://junyaohu.github.io)")
        with gr.Row():
            gr.Image(elem_classes="qrcode", value="./src/weixin.png", label="微信", height=250)
            gr.Image(elem_classes="qrcode", value="./src/zanshang.png", label="感谢支持", height=250)

    ############################## 后端 ##################################

    def onComfirmClick(user_name):
        if user_name.strip():
            return f"您好，{user_name}。请根据如下提示完成测评！小心不要被 PUA 到啦 ~", gr.update(value="已确认", interactive=False), gr.update(interactive=True)
        else:
            return "👇 请输入昵称", gr.update(), gr.update()
        
    confirm.click(onComfirmClick, user_name, [hello, confirm, restart])

    def onRestart():
        res = get_new_question_and_options()
        return res["question_text"], "", "", \
                gr.update(value="A: " + res["option_A"], interactive=True), \
                gr.update(value="B: " + res["option_B"], interactive=True), \
                gr.update(value="C: " + res["option_C"], interactive=True), \
                gr.update(value="", interactive=True), \
                1, 50, 50, 50, 50, gr.update(value="➡️ 下一题", interactive=False), gr.update(visible=False), gr.update(interactive=True)
    
    restart.click(onRestart, None, [ \
                question_text, result_text, score_text, \
                option_A_button, option_B_button, option_C_button, option_D_text, \
                progress_slider, dim1_slider, dim2_slider, dim3_slider, dim4_slider, next_btn, restart, refresh])
    
    def onRefresh():
        res = get_new_question_and_options()
        return res["question_text"], "", "", \
                gr.update(value="A: " + res["option_A"], interactive=True), \
                gr.update(value="B: " + res["option_B"], interactive=True), \
                gr.update(value="C: " + res["option_C"], interactive=True), \
                gr.update(value="", interactive=True), \
                gr.update(), gr.update(), gr.update(), gr.update(), gr.update(interactive=False), "↩️ 重新开始", gr.update(interactive=True)
    
    refresh.click(onRefresh, None, [ \
                question_text, result_text, score_text, \
                option_A_button, option_B_button, option_C_button, option_D_text, \
                dim1_slider, dim2_slider, dim3_slider, dim4_slider, next_btn, restart, refresh])

    def onNext(progress_slider):
        if progress_slider == data.value['total']:
            return "", "", "", \
                    gr.update(value="A: 查看测评结果", interactive=False), \
                    gr.update(value="B: 查看测评结果", interactive=False), \
                    gr.update(value="C: 查看测评结果", interactive=False), \
                    gr.update(value="", interactive=False), \
                    progress_slider, gr.update(interactive=False), "↩️ 重新开始", gr.update(interactive=False)
        else:
            res = get_new_question_and_options()
            return res["question_text"], "", "", \
                    gr.update(value="A: " + res["option_A"], interactive=True), \
                    gr.update(value="B: " + res["option_B"], interactive=True), \
                    gr.update(value="C: " + res["option_C"], interactive=True), \
                    gr.update(value="", interactive=True), \
                    progress_slider+1, gr.update(interactive=False), "↩️ 重新开始", gr.update(interactive=True)
    
    next_btn.click(onNext, progress_slider, [ \
        question_text, result_text, score_text, \
        option_A_button, option_B_button, option_C_button, option_D_text, \
        progress_slider, next_btn, restart, refresh])
    
    def onClickABC(btn, question_text, progress_slider, dim1_slider, dim2_slider, dim3_slider, dim4_slider):
        option_text = btn[3:]
        res = get_option_result(question_text, option_text)

        dim1_slider = max(min(dim1_slider + res["deltas"][0], 100), -100)
        dim2_slider = max(min(dim2_slider + res["deltas"][1], 100), -100)
        dim3_slider = max(min(dim3_slider + res["deltas"][2], 100), -100)
        dim4_slider = max(min(dim4_slider + res["deltas"][3], 100), -100)

        if progress_slider == data.value['total']:
            text = "✨查看结果"
        else:
            text = "➡️ 下一题"

        print(res["result_text"], res["score_text"], dim1_slider, dim2_slider, dim3_slider, dim4_slider, text)
        return res["result_text"], res["score_text"], dim1_slider, dim2_slider, dim3_slider, dim4_slider, gr.update(value=text, interactive=True), gr.update(value="↩️ 重新开始", visible=True), gr.update(interactive=False), \
            gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False)

    option_A_button.click(onClickABC, \
                        [option_A_button, question_text, progress_slider, dim1_slider, dim2_slider, dim3_slider, dim4_slider], \
                        [result_text, score_text, dim1_slider, dim2_slider, dim3_slider, dim4_slider, next_btn, restart, refresh, \
                         option_A_button, option_B_button, option_C_button, option_D_button, option_D_text])
    option_B_button.click(onClickABC, \
                        [option_B_button, question_text, progress_slider, dim1_slider, dim2_slider, dim3_slider, dim4_slider], \
                        [result_text, score_text, dim1_slider, dim2_slider, dim3_slider, dim4_slider, next_btn, restart, refresh, \
                         option_A_button, option_B_button, option_C_button, option_D_button, option_D_text])
    option_C_button.click(onClickABC, \
                        [option_C_button, question_text, progress_slider, dim1_slider, dim2_slider, dim3_slider, dim4_slider], \
                        [result_text, score_text, dim1_slider, dim2_slider, dim3_slider, dim4_slider, next_btn, restart, refresh, \
                         option_A_button, option_B_button, option_C_button, option_D_button, option_D_text])
    
    def onClickD(option_D_text, question_text, progress_slider, dim1_slider, dim2_slider, dim3_slider, dim4_slider):
        res = get_option_result(question_text, option_D_text)

        dim1_slider = max(min(dim1_slider + res["deltas"][0], 100), -100)
        dim2_slider = max(min(dim2_slider + res["deltas"][1], 100), -100)
        dim3_slider = max(min(dim3_slider + res["deltas"][2], 100), -100)
        dim4_slider = max(min(dim4_slider + res["deltas"][3], 100), -100)

        if progress_slider == data.value["total"]:
            text = "✨查看结果"
        else:
            text = "➡️ 下一题"

        print(res["result_text"], res["score_text"], dim1_slider, dim2_slider, dim3_slider, dim4_slider, text)
        return res["result_text"], res["score_text"], dim1_slider, dim2_slider, dim3_slider, dim4_slider, gr.update(value=text, interactive=True), gr.update(value="↩️ 重新开始", visible=True), gr.update(interactive=False), \
            gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False)
    
    option_D_button.click(onClickD, \
                        [option_D_text  , question_text, progress_slider, dim1_slider, dim2_slider, dim3_slider, dim4_slider], \
                        [result_text, score_text, dim1_slider, dim2_slider, dim3_slider, dim4_slider, next_btn, restart, refresh, \
                         option_A_button, option_B_button, option_C_button, option_D_button, option_D_text])

    def onChangeD(option_D_text):
        if option_D_text.strip():
            return gr.update(interactive=True)
        else:
            return gr.update(interactive=False)

    option_D_text.change(onChangeD, option_D_text, option_D_button)

if __name__ == "__main__":
    # demo.launch()
    demo.launch(share=True)