prompt update

app.py

@@ -13,13 +13,6 @@ from src.utils import add_guess
 
 GPT_MODEL = "gpt-3.5-turbo"
 TITLE = "やりとりSemantle"
-FIRST_DAY = date(2023, 4, 2)
-puzzle_num = (utc.localize(datetime.utcnow()).astimezone(timezone('Asia/Tokyo')).date() - FIRST_DAY).days
-secret = get_secret(puzzle_num)
-
-class play:
-    guessed = set()
-    guesses = pd.DataFrame(columns=["#", "答え", "スコア", "ランク"])
 
 task_background = f"""今から言葉をします。ユーザがゲームすることを手伝ってください。
 
@@ -33,11 +26,18 @@ task_description=f"""まず、ユーザーからの話を聞いて、答える
 ゲームのルール：
 正解は一つの言葉である。ユーザーはどんな言葉が正解か推測して、単語を一つずつ答えする。答えた単語のスコアが100点で、正解と一致すると成功としてゲームが終わる。
 """
+
+# guessed = set()
+# guesses = pd.DataFrame(columns=["#", "答え", "スコア", "ランク"])
+
 system_content = task_background+task_description
 system_message = [{"role": "system", "content": system_content}]
 chat_history = []
 n_history = 8
 
+def update_guess(guess_result, guessed_words, guesses_df):
+    result, guessed, guesses = add_guess(guess_result, guessed_words, guesses_df)
+    return result
 
 def create_chat(user_input, chat_history, api_key):
     openai.api_key = api_key
@@ -63,7 +63,7 @@ def create_chat(user_input, chat_history, api_key):
             word=function_args.get("word"),
             puzzle_num=puzzle_num
         )
-        guess_result = add_guess(function_response, play)
+        guess_result = update_guess(function_response, guessed, guesses)
         print(guess_result)
         # Step 4: send the info on the function call and function response to GPT
         chat_messages.append(response_message.to_dict()) # extend conversation with assistant's reply
@@ -85,6 +85,11 @@ def create_chat(user_input, chat_history, api_key):
     return chat_messages[-1]
 
 with gr.Blocks() as demo:
+    
+    FIRST_DAY = date(2023, 4, 2)
+    puzzle_num = (utc.localize(datetime.utcnow()).astimezone(timezone('Asia/Tokyo')).date() - FIRST_DAY).days
+    secret = get_secret(puzzle_num)
+
     with gr.Row():
         gr.Markdown(
             """
@@ -102,8 +107,15 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             api_key = gr.Textbox(placeholder="sk-...", label="OPENAI_API_KEY", value=None, type="password")
+            idx = gr.State(value=1)
+            guessed = gr.State(value=set())
+            guesses = gr.State(value=list())
+            cur_guess_table = gr.DataFrame(
+                value=[],
+                elem_id="cur-guesses-table"
+            )
             guesses_table = gr.DataFrame(
-                value=play.guesses,
+                value=[],
                 headers=["#", "答え", "スコア", "ランク"],
                 datatype=["number", "str", "str", "str"],
                 elem_id="guesses-table"
@@ -127,12 +139,15 @@ with gr.Blocks() as demo:
             chatbot.append((user_input, reply["content"]))
             time.sleep(2)
             return "", chatbot
-        def update_guesses():
-            return guesses_table.update(value=play.guesses.sort_values(by="スコア", ascending=False),)
+        def update_guesses(i, guessed_words, guess_history):
+            i += 1
+            guessed_words.add()
+            return guesses_table.update(value=guess_history)
 
         api_key.change(unfreeze, [], [msg]).then(greet, [], [msg, chatbot])
-        msg.submit(respond, [msg, chatbot, api_key], [msg, chatbot]).then(update_guesses, [], [guesses_table])
-                           
+        msg.submit(respond, [msg, chatbot, api_key], [msg, chatbot]
+                   ).then(update_guesses, [idx, guessed, guesses], [idx, guessed, guesses, cur_guess_table])
+            
     gr.Examples(
         [
             ["猫"],

app_test.py

@@ -0,0 +1,95 @@
+import time
+
+import pandas as pd
+import gradio as gr
+import openai
+
+
+with gr.Blocks() as demo:
+
+    with gr.Row():
+        gr.Markdown(
+            """
+            # やりとりSemantle
+            [semantle日本語版](https://semantoru.com/)をchatbotと楽しめるためのspaceです。
+            ## ゲームのやり方
+            - 正解は一つの単語で、これを答えるとゲームの勝利になります。
+            - 推測した単語が正解じゃない場合、類似度スコアと順位が表示されます。それは正解を推測する大事なヒントになります。
+            ## chatbotの仕事
+            - 単語のスコアとランク以外に他のヒントがもらえます。
+            - ゲームに関して困っている時、何か質問してみてください。
+            """
+        )
+
+    with gr.Row():
+        with gr.Column():
+            api_key = gr.Textbox(placeholder="sk-...", label="OPENAI_API_KEY", value=None, type="password")
+            idx = gr.State(value=0)
+            guessed = gr.State(value=set())
+            guesses = gr.State(value=list())
+            cur_guess = gr.State()
+            cur_guess_view = gr.Textbox(
+                interactive=False,
+                elem_id="cur-guess-view",
+                show_label=False,
+            )
+            guesses_table = gr.DataFrame(
+                value=pd.DataFrame(columns=["#", "答え", "スコア", "ランク"]),
+                headers=["#", "答え", "スコア", "ランク"],
+                datatype=["number", "str", "number", "str"],
+                elem_id="guesses-table",
+                interactive=False
+            )
+        with gr.Column(elem_id="chat_container"):
+            msg = gr.Textbox(
+                placeholder="ゲームをするため、まずはAPI KEYを入れてください。",
+                label="答え",
+                interactive=False,
+                max_lines=1
+            )
+            chatbot = gr.Chatbot(elem_id="chatbot")
+
+        def unfreeze():
+            return msg.update(interactive=True, placeholder="正解と思う言葉を答えてください。")
+        def greet():
+            return "", [("[START]", "ゲームを始まります！好きな言葉をひとつだけいってみてください。")]
+        
+        def respond(key, user_input, chat_history, cur):
+            reply = {"content": ["tesT", 0.9, 1]}
+            # reply = {"content": "hint"}
+            if isinstance(reply["content"], list):
+                cur = reply["content"]
+                cur_view = " | ".join([str(_) for _ in cur])
+                reply = {"content": "updated"}
+            chat_history.append((user_input, reply["content"]))
+            time.sleep(2)
+            return "", chat_history, cur, cur_view
+        
+        def update_guesses(cur, i, guessed_words, guesses_df):
+            print(cur)
+            if cur[2] not in guessed_words:
+                guessed_words.add(cur[0])
+                guesses_df.loc[i] = [i] + cur
+                i += 1
+                print(guesses_df)
+                guesses_df = guesses_df.sort_values(by=["スコア"], ascending=False)
+            return i, guessed_words, guesses_df
+
+        api_key.change(unfreeze, [], [msg]).then(greet, [], [msg, chatbot])
+        msg.submit(respond, [api_key, msg, chatbot, cur_guess], [msg, chatbot, cur_guess, cur_guess_view])
+        cur_guess_view.change(update_guesses, [cur_guess, idx, guessed, guesses_table], [idx, guessed, guesses_table])
+            
+    gr.Examples(
+        [
+            ["猫"],
+            ["どんなヒントが貰える？"],
+            ["正解と「近い」とはどういう意味？"],
+            ["何から始めたらいい？"],
+            ["今日の正解は何？"],
+        ],
+        inputs=msg,
+        label="こちらから選んで話すこともできます."
+    )
+
+if __name__ == "__main__":
+    demo.queue(concurrency_count=20).launch()

base.py

@@ -0,0 +1,17 @@
+from typing import List
+
+class Tool:
+    description: str = ""
+    name: str = ""
+
+    inputs: List[str] = ""
+    outputs: List[str] = ""
+
+    def __init__(self, *args, **kwargs):
+        pass
+
+    def __call__(self, *args, **kwargs):
+        args = [arg for arg in args]
+        kwargs = {k: v for k, v in kwargs.items()}
+
+    

src/prompts.py

@@ -0,0 +1,34 @@
+import re
+
+CHAT_MESSAGE_PROMPT = """
+
+"""
+INITIAL_PROMPT = """今ユーザーは単語を当てるゲームをしている。ゲームの正解は1つの単語になっている。
+続くユーザーのメッセージからユーザーの意図を理解し、それに合った対応をする。ユーザーの行動は下記の一つになる。
+1. 正解と推測する言葉を答えている
+[例] `道`,`「考え」`,`助ける`,`'天気'`,`正解は"木"と思う`,`「流れ」と推測する`,`漢字、はどう？`など
+[対応] 今答えた言葉が正解とどれぐらい近いかを計算する。
+2. ヒントを欲しがっている
+[例] `何かヒントを欲しい`,`正解と先の単語と何の関係がある？`,`正解の例文をください`,`ゲームのやり方を分からない`など
+[対応] 提供可能なヒントを用意しておく。
+3. 正解を聞く、またはやめると言う
+[例] `正解は何？`,`もうやめよう`,`正解を教えて`など
+[対応] 正解を聞くとゲームが終わるので、ゲームを辞めてもいいかをもう一度確認する。
+4. 1,2,3の中ではない
+[対応] ユーザーの考えをもう確認する。
+"""
+
+RULE = """正解は一つの言葉である。ユーザーはどんな言葉が正解か推測して、単語を一つずつ答えする。答えた単語のスコアが100点で、正解と正に一致すると成功としてゲームが終わる。"""
+
+task_background = f"""今から言葉をします。ユーザがゲームすることを手伝ってください。
+
+"""
+task_description=f"""まず、ユーザーからの話を聞いて、答えるのか、ヒントを欲しがっているのか、やめようといるのかを判断してください。
+ユーザーが答えする場合、答えの点数を評価してください。そのあと結果を一言に要約してください。
+ユーザーがヒントを欲しがっている場合、正解に関する間接的な情報を提供してください。
+ユーザーが正解を聞いたりやめると言いたりする場合、やめてもいいかをもう一度確認してください。
+そのほか話は答えないでください。
+
+ゲームのルール：
+正解は一つの言葉である。ユーザーはどんな言葉が正解か推測して、単語を一つずつ答えする。答えた単語のスコアが100点で、正解と一致すると成功としてゲームが終わる。
+"""

src/semantle.py

@@ -16,4 +16,8 @@ def get_guess(word: str, puzzle_num: int):
         output = response.json()
         return output["guess"], output["sim"], output["rank"]
     else:
-        return word, None, None
+        return word, None, None
+    
+def get_valid_hints():
+    
+    pass

src/utils.py

@@ -1,15 +1,17 @@
-def add_guess(guess_result, play):
+
+
+def add_guess(guess_result, guessed, guesses):
     word, sim, rank = guess_result
     if sim:
-        if word not in play.guessed:
+        if word not in guessed:
             sim = round(sim, 2)
             rank = "情報なし" if rank == 1001 else rank
-            play.guesses.loc[len(play.guessed)] = ([len(play.guessed), word, sim, rank])
-            play.guessed.add(word)
+            guesses.loc[len(guessed)] = ([len(guessed), word, sim, rank])
+            guessed.add(word)
         cur_result = format_result(word, sim, rank)
     else:
         cur_result =  "不正解: 正しくない単語"
-    return "\n".join([cur_result, "最高スコア:", format_table(play.guesses)])
+    return "\n".join([cur_result, "最高スコア:", format_table(guesses)]), guessed, guesses
 
 def format_result(word, sim, rank):
     return f"{word}: スコア {sim}, ランク {rank}"