基于ERNIE Bot SDK开发中国传统酒令游戏--射覆

def ernie_cover_prompt():
    return ("我想要你陪我玩'射覆'游戏："
            "你的任务如下："
            "1.你需要从你的身边随机找一件物品，不要选钟表等过于常见的、特别容易猜中的物品"
            "2.输出一句有趣的打油诗，来生动形象的描述这件物件。（打油诗中不要出现该物品的名称，内容不要过于直白，不要使用户一下子就猜到）"
            "3.将结果以{\"thing\":\"\",\"poem\":\"\"}的JSON格式返回，其中thing表示物品名称，poem表示描述这件物品的打油诗。"
            "注意: 除了JSON格式结果以外，不要返回其他任何内容。")

def shoot_prompt(thing):
    return f"""
    请你用一句打油诗描述{thing},诗中不能出现"{thing}"几个字。除了打油诗以外，不要返回其他任何内容。
    """

def user_shoot_prompt(shoot_poem, cover_thing):
    prompt = f"""
        你现在正在跟我玩射覆游戏，你"覆"的物品是{cover_thing},我给的描述这个物品的打油诗是{shoot_poem}。
        如果你认为我猜对了，请你对我说：我覆的是"{cover_thing}"，你猜对了，你和我还真是心有灵犀呢，让我们共饮一杯吧！
        如果你认为我猜错了，请你对我说：我覆的是"{cover_thing}"，你猜错了，请自罚一杯！
        最后再说一句寓意美好的祝酒辞作为结尾。
    """
    
    return prompt

# 安装依赖!pip install -U erniebot

import erniebotdef set_token(token):
    erniebot.api_type = 'aistudio'
    erniebot.access_token = token

set_token("替换为你自己的token")

 def get_ernie_res(txt):
        model = 'ernie-3.5'
        messages = [{'role': 'user', 'content': txt}]
        response = erniebot.ChatCompletion.create(
            model=model,
            messages=messages,
        )        return response.result#test get_ernie_res("hi")

# 随机给出一个物品名称和描述这件物品的打油诗def ernie_cover_prompt():
    return ("我想要你陪我玩'射覆'游戏："
            "你的任务如下："
            "1.你需要从你的身边随机找一件物品，不要选钟表等过于常见的、特别容易猜中的物品"
            "2.输出一句有趣的打油诗，来生动形象的描述这件物件。（打油诗中不要出现该物品的名称，内容不要过于直白，不要使用户一下子就猜到）"
            "3.将结果以{\"thing\":\"\",\"poem\":\"\"}的JSON格式返回，其中thing表示物品名称，poem表示描述这件物品的打油诗。"
            "注意: 除了JSON格式结果以外，不要返回其他任何内容。")

import json# 将字段值分别输出def process_ernie_cover():
    json_res = get_ernie_res(ernie_cover_prompt())     # 提取JSON部分
    start_index = json_res.find('{')
    end_index = json_res.rfind('}') + 1
    json_str = json_res[start_index:end_index]
    json_obj = json.loads(json_str)    return json_obj["thing"], json_obj["poem"]
process_ernie_cover()

# 根据物品名称，给出一首描述该物品的打油诗def shoot_prompt(thing):
    return f"""
    请你用一句打油诗描述{thing},诗中不能出现"{thing}"几个字。除了打油诗以外，不要返回其他任何内容。
    """# testget_ernie_res(shoot_prompt('雪糕'))

# 根据覆的物品名称 和 射者的打油诗判断游戏结果（猜中、没猜中），并给出对应的输出  def user_shoot_prompt(shoot_poem, cover_thing):
    prompt = f"""
        你现在正在跟我玩射覆游戏，你"覆"的物品是{cover_thing},我给的描述这个物品的打油诗是{shoot_poem}。
        如果你认为我猜对了，请你对我说：我覆的是"{cover_thing}"，你猜对了，你和我还真是心有灵犀呢，让我们共饮一杯吧！
        如果你认为我猜错了，请你对我说：我覆的是"{cover_thing}"，你猜错了，请自罚一杯！
        最后再说一句寓意美好的祝酒辞作为结尾。
    """
    return prompt

get_ernie_res(user_shoot_prompt("红纸轻裁锦绣词，佳期遥盼喜相思。诚心挚意传千里，佳宾云集共此时。", "请柬"))

with gr.Blocks(title="射覆", theme="soft") as game:
    gr.Markdown("<center><h2>中国传统游戏--射覆</h2></center>")
    gr.Markdown("""<center><p1>
    射覆是一种古时的猜物游戏，常用于占卜，也是酒令的一种形式。游戏通常在一个器具（如瓯、盂）下覆盖某一物件，让参与者猜测里面是什么东西。射覆的起源可以追溯到汉代，当时的皇宫中就已经流行这种游戏。游戏所用的物品通常是日常用品，如手巾、扇子、笔墨、盒罐等。
    随着时间的推移，射覆逐渐演变成民间宴饮时的文字游戏。在《红楼梦》中，射覆被描述为一种酒令，参与者通过成语或诗句来猜测物体，这不仅考验参与者的知识储备，还体现了他们的才学和性格。
    射覆的游戏规则是这样的：所谓“射”，是“猜”的意思。所谓“覆”，是“遮盖、隐藏”的意思。<br/>
    ①覆者先说谜面（在本游戏中谜面是一首打油诗）；
    ②射者猜中的话不能直接说出谜底，要另说一个谜面来作答；
    ③彼此都对才算过关;<br/>
    通俗一点来说，就是猜谜语，但是比猜谜语更难，准确来说是以谜猜谜。
    作诗对于我们现代人来说是比较难的，好在有文心大模型的帮忙，我们就可以愉快的玩射覆啦。
</p1>
</center>""")
    token = gr.Textbox(label="Token")    with gr.Tab("你\"射\",文心\"覆\""):        with gr.Row():            with gr.Column():
                cover_button = gr.Button(value="覆")                # 谜底隐藏不显示             
                cover_thing = gr.Textbox(visible=False)
                cover_poem = gr.Textbox(label="谜面")            with gr.Column():
                shoot_thing = gr.Textbox(label="物品名称")
                shoot_poem = gr.Textbox(label="射")
                shoot_button = gr.Button(value="不会写诗，文心帮帮我♥️")        with gr.Row():
            res = gr.Textbox(label="结果")
            judge_button = gr.Button(value="看看你猜对了吗")

        cover_button.click(ernie_cover, [], [cover_thing, cover_poem])
        shoot_button.click(gene_poem, [shoot_thing], [shoot_poem])
        judge_button.click(user_shoot_res, [shoot_thing, cover_thing], [res])