add functions

app.py

@@ -14,37 +14,15 @@ from src.utils import add_guess
 GPT_MODEL = "gpt-3.5-turbo"
 TITLE = "やりとりSemantle"
 
-task_background = f"""今から言葉をします。ユーザがゲームすることを手伝ってください。
-
-"""
-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
     chat_messages = [{"role": "user", "content": user_input}]
     response = openai.ChatCompletion.create(
         model=GPT_MODEL,
-        messages=system_message+chat_history+chat_messages,
+        messages=system_message+chat_messages,
         functions=get_functions()
     )
     response_message = response.choices[0].message
@@ -54,7 +32,8 @@ def create_chat(user_input, chat_history, api_key):
         # Step 3: call the function
         # Note: the JSON response may not always be valid; be sure to handle errors
         available_functions = {
-            "evaluate_guess": get_guess,
+            "evaluate_score": get_guess,
+            "get_data_for_hint": get_play_data,
         }
         function_name = response_message["function_call"]["name"]
         function_to_call = available_functions[function_name]
@@ -107,18 +86,16 @@ 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)
+            idx = gr.State(value=0)
             guessed = gr.State(value=set())
             guesses = gr.State(value=list())
-            cur_guess_table = gr.DataFrame(
-                value=[],
-                elem_id="cur-guesses-table"
-            )
+            cur_guess = gr.State()
             guesses_table = gr.DataFrame(
-                value=[],
-                headers=["#", "答え", "スコア", "ランク"],
-                datatype=["number", "str", "str", "str"],
-                elem_id="guesses-table"
+                value=pd.DataFrame(columns=["#", "答え", "スコア", "ランク"]),
+                headers=["#", "答え", "score", "score"],
+                datatype=["number", "str", "number", "str"],
+                elem_id="guesses-table",
+                interactive=False
             )
         with gr.Column(elem_id="chat_container"):
             msg = gr.Textbox(
@@ -134,19 +111,26 @@ with gr.Blocks() as demo:
         def greet():
             return "", [("[START]", "ゲームを始まります！好きな言葉をひとつだけいってみてください。")]
         
-        def respond(user_input, chatbot, api_key):
-            reply = create_chat(user_input, chat_history, api_key)
+        def respond(key, user_input, chat_history, cur):
+            reply = create_chat(key, user_input)
+            if isinstance(reply["content"], list):
+                cur = reply["content"]
             chatbot.append((user_input, reply["content"]))
             time.sleep(2)
-            return "", chatbot
-        def update_guesses(i, guessed_words, guess_history):
-            i += 1
-            guessed_words.add()
-            return guesses_table.update(value=guess_history)
+            return "", chatbot, cur
+        
+        def update_guesses(cur, i, guessed_words, guesses_df):
+            if cur[0] not in guessed_words:
+                guessed_words.add(cur[0])
+                guesses_df.loc[i] = [i+1] + cur
+                i += 1
+                guesses_df = guesses_df.sort_values(by=["score"], ascending=False)
+            return i, guessed_words, guesses_df
 
         api_key.change(unfreeze, [], [msg]).then(greet, [], [msg, chatbot])
-        msg.submit(respond, [msg, chatbot, api_key], [msg, chatbot]
-                   ).then(update_guesses, [idx, guessed, guesses], [idx, guessed, guesses, cur_guess_table])
+        msg.submit(respond, [api_key, msg, chatbot, cur_guess], [msg, chatbot, cur_guess]).then(
+            update_guesses, [cur_guess, idx, guessed, guesses_table], [idx, guessed, guesses_table]
+        )
             
     gr.Examples(
         [

base.py

@@ -1,17 +1,19 @@
-from typing import List
+# from typing import List
 
-class Tool:
-    description: str = ""
-    name: str = ""
+# class Tool:
+#     description: str = "Calculate a score and a rank of the guess word, which indicate a distance from the secret word."
+#     name: str = "evaluate_score"
 
-    inputs: List[str] = ""
-    outputs: List[str] = ""
+#     inputs: List[str] = ["word", "puzzle_num"]
+#     required: List[str] = ["word", "puzzle_num"]
 
-    def __init__(self, *args, **kwargs):
-        pass
+#     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()}
+#     def setup(self):
+        
+#     def __call__(self, *args, **kwargs):
+#         args = [arg for arg in args]
+#         kwargs = {k: v for k, v in kwargs.items()}
 
     

src/functions.py

@@ -1,20 +1,72 @@
-evaluate_guess = {"name": "evaluate_guess",
-                  "description": "Calculate the score of a guess word and get the rank among the 1,000 words.",
+guess_word = {"name": "guess_word",
+                  "description": "Use this function to check if a guessed word is the correct answer or not, and if incorrect, calculate a score and a rank of the guess word.",
                   "parameters": {
                       "type": "object",
                       "properties": {
                           "word": {
                               "type": "string",
-                              "description": "A single Japanese word, which is can be a noun, verb, adverb or adjective. e.g. 空, 近い, 行く, etc."
+                              "description": "A single Japanese word to guess, which is can be a noun, verb, adverb or adjective. e.g. 空, 近い, 行く, etc."
                               },
-                          "puzzle": {
+                          "puzzle_num": {
+                              "type": "integer",
+                              "description": "An index indicating today's puzzle."
+                          }
+                      },
+                      "required": ["word", "puzzle_num"]
+                  }}
+
+prepare_hint = {"name": "prepare_hint",
+                "description": "Use this function to retrieve information for hint.",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "puzzle_num": {
+                            "type": "interger",
+                            "description": "An index for today's puzzle set."
+                        }
+                    },
+                    "required": ["puzzle_num"]
+                }}
+
+get_secret = {"name": "get_secret",
+                "description": "You can check what the correct answer of today's puzzle is by this function.",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "puzzle_num": {
+                            "type": "integer",
+                            "description": "An index indicating today's puzzle."
+                        }
+                    },
+                    "required": ["puzzle_num"]
+                }}
+
+get_puzzle_num = {"name": "get_puzzle_num",
+                  "description": "Use this function to check today's puzzle number.",
+                  "parameters": {
+                      "type": "object",
+                      "properties": {}
+                  },
+                  }
+
+update_history = {"name": "update_history",
+                  "description": "Use this function to add current guess to a table for a user's guess history.",
+                  "parameters": {
+                      "type": "object",
+                      "properties": {
+                          "current_guess": {
+                              "type": "json",
+                              "description": "A currently guessed word and its score and rank."
+                          },
+                          "guess_history": {
                               "type": "object",
-                              "description": "A class containing information about the puzzle; a secret word and scores/ranks for other words."
+                              "description": "A dataframe containing the guessed words and its xore and rank in a row."
                           }
                       },
-                      "required": ["word", "puzzle"]
+                      "required": ["current_guess", "guess_history"]
                   }}
 
+
 def get_functions():
-    functions = [evaluate_guess]
+    functions = [guess_word]
     return functions

src/prompts.py

@@ -1,8 +1,5 @@
 import re
 
-CHAT_MESSAGE_PROMPT = """
-
-"""
 INITIAL_PROMPT = """今ユーザーは単語を当てるゲームをしている。ゲームの正解は1つの単語になっている。
 続くユーザーのメッセージからユーザーの意図を理解し、それに合った対応をする。ユーザーの行動は下記の一つになる。
 1. 正解と推測する言葉を答えている
@@ -10,7 +7,7 @@ INITIAL_PROMPT = """今ユーザーは単語を当てるゲームをしている
 [対応] 今答えた言葉が正解とどれぐらい近いかを計算する。
 2. ヒントを欲しがっている
 [例] `何かヒントを欲しい`,`正解と先の単語と何の関係がある？`,`正解の例文をください`,`ゲームのやり方を分からない`など
-[対応] 提供可能なヒントを用意しておく。
+[対応] ヒントを出すため必要な情報を用意する。
 3. 正解を聞く、またはやめると言う
 [例] `正解は何？`,`もうやめよう`,`正解を教えて`など
 [対応] 正解を聞くとゲームが終わるので、ゲームを辞めてもいいかをもう一度確認する。
@@ -20,15 +17,22 @@ INITIAL_PROMPT = """今ユーザーは単語を当てるゲームをしている
 
 RULE = """正解は一つの言葉である。ユーザーはどんな言葉が正解か推測して、単語を一つずつ答えする。答えた単語のスコアが100点で、正解と正に一致すると成功としてゲームが終わる。"""
 
-task_background = f"""今から言葉をします。ユーザがゲームすることを手伝ってください。
-
+SUBTASK_1 = """下記の答えた結果を見て、れまでの結果や全体の単語の数に比べて今の評価を出す。
+- 答えた結果: <cur_guess>
+- これまで推測した結果: <guess_history>
+- データベースの総単語数: <n_total_words>
+ランクが1,000位以外の場合、またはデータベースにいない単語はランクが表示していない。
 """
-task_description=f"""まず、ユーザーからの話を聞いて、答えるのか、ヒントを欲しがっているのか、やめようといるのかを判断してください。
-ユーザーが答えする場合、答えの点数を評価してください。そのあと結果を一言に要約してください。
-ユーザーがヒントを欲しがっている場合、正解に関する間接的な情報を提供してください。
-ユーザーが正解を聞いたりやめると言いたりする場合、やめてもいいかをもう一度確認してください。
-そのほか話は答えないでください。
 
-ゲームのルール：
-正解は一つの言葉である。ユーザーはどんな言葉が正解か推測して、単語を一つずつ答えする。答えた単語のスコアが100点で、正解と一致すると成功としてゲームが終わる。
+SUBTASK_2 = """ユーザーの要求に合わせてヒントを出す。
+もしヒントの形が明確にならない場合、例の可能なヒントから一つ選んでそれを出す。
+[プレイ情報]
+- これまで推測した単語: <guessed_list>
+- これまで最後点数の単語: <best_score_word>
+- 正解: <secret>
+[例]
+- 正解単語を使って文章をひとつ作る。正し、正解単語は必ず隠すこと。
+- これまで推測した単語の中、あなたが正解と一番近いと思う単語と選び、その理由とともに話す。
+- これまでの最高点数の単語と正解の間の意味、文法、話用論的な共通点をひとつ教える。
+- これまでの最高点数の単語と正解、両方意味的に関係がある単語をひとつ選んでそれを出す。
 """

src/semantle.py

@@ -1,3 +1,5 @@
+from datetime import date, datetime
+from pytz import utc, timezone
 import requests
 
 def get_secret(puzzle_num: int):
@@ -6,18 +8,19 @@ def get_secret(puzzle_num: int):
     if response.status_code == 200:
         return response.content
     else:
-        print("Not found error.")
+        return "Not found error."
 
 def get_guess(word: str, puzzle_num: int):
     request_url = f"https://semantoru.com/guess/{puzzle_num}/{word}"
     response = requests.get(request_url)
     print(response.status_code)
     if response.status_code == 200:
-        output = response.json()
-        return output["guess"], output["sim"], output["rank"]
+        return response.json()
     else:
-        return word, None, None
+        return {"guess": word, 
+                "sim": None,
+                "rank": None}
     
-def get_valid_hints():
-    
-    pass
+def get_puzzle_num():
+    FIRST_DAY = date(2023, 4, 2)
+    return (utc.localize(datetime.utcnow()).astimezone(timezone('Asia/Tokyo')).date() - FIRST_DAY).days