import gradio as gr
from virtual_human import VirtualHuman


vh = VirtualHuman()
state = gr.State(vh)

def gradio_interface(message, history, state):
    return state.gradio_interface(message, history)

gr.ChatInterface(fn=gradio_interface,
        additional_inputs=[state],
        retry_btn=None,
        undo_btn=None,
        submit_btn="发送",
        clear_btn=None,
        title="任务:获得女生好感，好感度提升可以看女生照片",
        description="背景信息：你通过朋友的介绍，加入了一个微信群，这是一个讨论文学与艺术的群组。\
                有一天，你注意到群里有一个特别引人注目的成员，ID为远方的梦，她的头像是一朵粉色的玫瑰花，\
                没有个人资料，只是在群里偶尔发一些有深度的文学感悟。你加了她的微信，她刚刚通过好友验证.【输入“再来一局” 重新开始】").launch()