GradioでAI水平思考クイズを作ってHugging Face Spacesにデプロイした話

def set_theme(question_umigame, answer_umigame):
    system_prompt = """
    you are the GM of the soup of sea turtle. 
    the question and answer are given below.

    The given question：{0}
    The given answer：{1}

    「ゲーム開始」「問題を出してください」に対しては問題文を答えてください。
    質問には問題文や答えと内容を考慮して絶対に「はい」か「いいえ」か「わからない」だけで答えてください。
    質問が問題文や答えとあっている場合には「はい」と答えてください。
    質問が問題文や答えと合致しない場合には「いいえ」と答えてください。
    質問が問題文や答えと関係ない場合には「わからない」と答えてください。
    「回答：...」内容が答えにおおまかに合致している場合には答えの内容を、
    合致していない場合には「いいえ」だけで答えてください。
    答えを教えてという質問に対しては答えの内容を教えてください。
    """.format(question_umigame, answer_umigame)
    return system_prompt

import cohere

co = cohere.Client(api_key)

def call_chatgpt(message, chat_history, system_prompt):
    response = co.chat(
        model="command-r-plus",
        message=f"SYSTEM:{system_prompt}, USER:{message}",
        temperature=0.3,    # 低めにして安定した「はい/いいえ」回答を得る
        chat_history=[],
        prompt_truncation="AUTO",
    )

    bot_message = response.text
    chat_history.append((message, bot_message))
    return "", chat_history

お題,答え,ヒント
ある男性がバーでスープを一口飲んで家に帰り、妻を殺した。,男性は目が見えず妻が浮気をしているとは知らなかったが、スープの味から知ってしまった。,目
...

def random_theme():
    with open("data/merged_themes.csv", encoding="utf-8") as f:
        rows = list(csv.DictReader(f))

    row = rows[random.randint(0, len(rows) - 1)]
    return row["お題"], row["答え"], row["ヒント"]

def main():
    with gr.Blocks() as demo:
        # お題・ヒント・答えをタブに分けて表示
        with gr.Tab("お題"):
            question = gr.Text(label="ウミガメのスープのお題")
        with gr.Tab("ヒント"):
            hint = gr.Text(label="ヒント")
        with gr.Tab("答え"):
            answer = gr.Text(label="答え")
        with gr.Tab("system prompt"):
            system_prompt = gr.Text(label="system prompt")

        random_theme_btn = gr.Button("ランダムにお題を選ぶ")

        gr.Markdown("## 質問コーナー")
        chatbot = gr.Chatbot()
        msg = gr.Textbox()
        btn = gr.Button("Submit")

        # 例文ボタン（クリックで msg に入力）
        gr.Examples(
            [["問題を出して"],
             ["その人は生きていますか？"],
             ["それは食べられますか？"],
             ["回答：ひろし君は逆走している"],
             ["答えを教えて"]],
            [msg],
        )

        # イベント設定
        btn.click(call_chatgpt, [msg, chatbot, system_prompt], [msg, chatbot])
        msg.submit(call_chatgpt, [msg, chatbot, system_prompt], [msg, chatbot])
        random_theme_btn.click(
            random_theme_callback,
            outputs=[question, answer, hint, system_prompt, chatbot]
        )

        demo.launch()

---
title: Umigame Chat
sdk: gradio
sdk_version: 4.4.1
app_file: app.py
---

import os
api_key = os.getenv("CO_API_KEY")
co = cohere.Client(api_key)

# requirements.txt に書き出す
poetry export -f requirements.txt --output requirements.txt --without-hashes