update prompt

app.py

@@ -7,7 +7,6 @@ import openai
 
 from src.semantle import get_guess, get_secret, get_puzzle_num
 from src.functions import get_functions
-from src.utils import add_guess
 
 GPT_MODEL = "gpt-3.5-turbo"
 TITLE = "やりとりイミトル"
@@ -15,13 +14,17 @@ TITLE = "やりとりイミトル"
 with open("data/rule.md", "r", encoding="utf-8") as f:
     RULEBOOK = "\n".join(f.readlines())
 
-def _execute_function(function_call, chat_messages):
+def get_rulebook():
+    return RULEBOOK
+
+def _execute_function(function_call, chat_messages, guess_result):
         # Step 3: call the function
         # Note: the JSON response may not always be valid; be sure to handle errors
         available_functions = {
             "guess_word": get_guess,
             "lookup_answer": get_secret,
-            "retrive_puzzle": get_puzzle_num
+            "retrieve_puzzle_num": get_puzzle_num,
+            "read_rule": get_rulebook,
         }
         function_name = function_call["name"]
         function_to_call = available_functions[function_name]
@@ -30,29 +33,33 @@ def _execute_function(function_call, chat_messages):
             **function_args
         )
         if function_call["name"] == "guess_word":
-            print("update guess")
+            guess_result = function_response
+        print(function_response)
         # Step 4: send the info on the function call and function response to GPT
         chat_messages.append(
             {"role": "function",
              "name": function_name,
-             "content": function_response["choices"][0]}
+             "content": f"{function_response}"}
         )   # extend conversation with function response
         next_response = openai.ChatCompletion.create(
             model=GPT_MODEL,
             messages=chat_messages,
+            functions=get_functions(),
+            temperature=0
         )   # get a new response from GPT where it can se the function response
         chat_messages.append(next_response.choices[0].message.to_dict())
-        return next_response, chat_messages
+        return next_response.choices[0].message.to_dict(), chat_messages, guess_result
 
-def create_chat(key, user_input):
-    openai.api_key = key
-    chat_messages = [
-        {"role": "system", "content": "あなたはiritoruというゲームの進行役です。ユーザーが答えを言ったり、ゲームに関するリクエストをしたら、短くても丁寧に答えてください。"},
-        {"role": "user", "content": user_input}]
+def create_chat(system_input, user_input, guess_result=dict()):
+    chat_messages = []
+    for s in system_input:
+        chat_messages.append({"role": "system", "content": s})
+    chat_messages.append({"role": "user", "content": user_input})
     response = openai.ChatCompletion.create(
         model=GPT_MODEL,
         messages=chat_messages,
-        functions=get_functions()
+        functions=get_functions(),
+        temperature=0
     )
     response_message = response.choices[0].message.to_dict()
     chat_messages.append(response_message) # extend conversation with assistant's reply
@@ -61,22 +68,24 @@ def create_chat(key, user_input):
     while response_message.get("function_call"):
         # Step 3: call the function
         # Note: the JSON response may not always be valid; be sure to handle errors
-        response_message, chat_messages = _execute_function(response_message["function_call"], chat_messages)
-    
-    return chat_messages[-1]
+        response_message, chat_messages, guess_result = _execute_function(response_message["function_call"], chat_messages, guess_result)
+    print(chat_messages)
+    return response_message, chat_messages, guess_result
 
 with gr.Blocks() as demo:
     with gr.Row():
         gr.Markdown(
             """
-            # やりとりSemantle
-            [semantle日本語版](https://semantoru.com/)をchatbotと楽しめるspaceです。
+            # やりとりxイミトル
+            「イミトル」は[semantle日本語版](https://semantoru.com/)の名前で、こちらはイミトルをassistantと楽しめるspaceです。
             ## ゲームのやり方
             - 正解は一つの単語で、これを答えるとゲームの勝利になります。
             - 推測した単語が正解じゃない場合、類似度スコアと順位が表示されます。それは正解を推測する大事なヒントになります。
-            ## chatbotの仕事
+            ## assistantの仕事
             - 単語のスコアとランク以外に他のヒントがもらえます。
             - ゲームに関して困っている時、何か質問してみてください。
+            ## ご了承のお願い
+            - ゲームをするため、openaiのapiが必要です。答えによって少々tokenを使うようになります。
             """
         )
 
@@ -86,10 +95,11 @@ with gr.Blocks() as demo:
             idx = gr.State(value=0)
             guessed = gr.State(value=set())
             guesses = gr.State(value=list())
-            cur_guess = gr.State()
+            cur_guess = gr.JSON(visible=False)
+            history = gr.State(list())
             guesses_table = gr.DataFrame(
-                value=pd.DataFrame(columns=["#", "答え", "スコア", "ランク"]),
-                headers=["#", "答え", "score", "score"],
+                value=pd.DataFrame(columns=["i", "guess", "sim", "rank"]),
+                headers=["i", "guess", "score", "rank"],
                 datatype=["number", "str", "number", "str"],
                 elem_id="guesses-table",
                 interactive=False
@@ -105,27 +115,42 @@ with gr.Blocks() as demo:
 
         def unfreeze():
             return msg.update(interactive=True, placeholder="正解と思う言葉を答えてください。")
-        def greet():
-            return "", [("[START]", "ゲームを始まります！好きな言葉をひとつだけいってみてください。")]
+        def reset_history():
+            return list()
+        def greet(key, gradio_messages):
+            openai.api_key = key
+            system_input = [get_rulebook(), "あなたは「イミトル」というゲームの進行役です。ユーザーが答えをするところです。よろしくお願いします。"]
+            user_input = ""
+            reply, _, _ = create_chat(system_input, user_input, guess_result=dict())
+            gradio_messages.append(("", reply["content"]))
+            # gradio_messages.append(("", f"今日のゲームのパズル番号は{get_puzzle_num()}です。それでは、始めましょう！言葉を当ててみてください。"))
+            time.sleep(2)
+            return gradio_messages
         
-        def respond(key, user_input, chat_messages):
-            reply = create_chat(key, user_input)
-            chat_messages.append((user_input, reply["content"]))
+        def respond(user_input, gradio_messages,  guess_result=dict()):
+            system_input = [get_rulebook(), """あなたは「イミトル」というゲームの進行役です。
+                            普段、ユーザーは答えをしますが、たまにゲームに関するヒントをリクエストをします。短くても丁寧に答えてください。
+                            ヒントを教える場合、正解を必ず隠してください。絶対、正解を言わないでください。"""]
+
+            _user_input = "答え:" + user_input
+            reply, messages, guess_result = create_chat(system_input, _user_input, guess_result=dict())
+            gradio_messages.append((user_input, reply["content"]))
             time.sleep(2)
-            return "", chat_messages
+            return gradio_messages, guess_result
         
         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
+            if cur.get('guess') and cur['guess'] not in guessed_words:
+                guessed_words.add(cur['guess'])
+                cur['sim'] = round(cur['sim'], 2)
+                cur['i'] = i
+                guesses_df.loc[i] = cur
                 i += 1
-                guesses_df = guesses_df.sort_values(by=["score"], ascending=False)
+                guesses_df = guesses_df.sort_values(by=["sim"], 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]).then(
-            update_guesses, [cur_guess, idx, guessed, guesses_table], [idx, guessed, guesses_table]
-        )
+        api_key.change(unfreeze, [], [msg]).then(reset_history, [], [history]).then(greet, [api_key, chatbot], [chatbot])
+        msg.submit(respond, [msg, chatbot, cur_guess], [chatbot, cur_guess])
+        cur_guess.change(update_guesses, [cur_guess, idx, guessed, guesses_table], [idx, guessed, guesses_table])
             
     gr.Examples(
         [
@@ -140,4 +165,5 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue(concurrency_count=20).launch()
+    demo.queue(concurrency_count=20).launch()
+# demo.queue(concurrency_count=20).launch()

data/rule.md

@@ -1,8 +1,10 @@
-# イミトル(Imitoru)
+# イミトル
 
 ## ゲームの遊び方
 今日の言葉を当てるゲームです。 正解と推測される単語を答えると、正解かどうかがわかります。正解でない場合、正解に近い度合いと順位がわかります。正解に近い程度とは単語の意味や文脈、単語の言語的特性などによって決まります。 推測した単語の近い度合いと順位を参考にして、答えを当て続けてみてください。
-もし正解が難しかったり、ゲームのルールがわかりにくい場合は、ヒントを求めることができます。正解を直接聞くことはできませんが、正解に関する様々な情報を得ることができます。
 
-## 正解の単語について
-正解の単語には、固有名詞、助詞、感動詞、及び記号以外の品詞の基本形の単語が含まれています。 正解の単語は、よく使われる表記に従っています。ただし推理する場合は、すべての表記を試すことができます。 同じ単語でも、表記によって類似度が異なる場合は、単語が使用される文脈が反映されていると考えることもできます。 ひらがなやカタカナの表記をする場合は、同音異字の場合にも注意してください。
+## ヒントのについて
+もし正解が難しかったり、ゲームのルールがわかりにくい場合は、ヒントを求めることができます。正解単語を使う例文、正解単語とある単語の意味や文法的な共通点など正解に関する様々な情報を聞くことができます。正し、ヒントに正解単語は`<mask>`などで隠すようになっています。
+
+## 「近い度合いと順位」について
+「近い度合い」は正解単語との類似度で、-100から+100までの数字で表します。「順位」はデータベースの中、近い度合いの相対的な位置で、１位から1000位まで表します。「近い度合い('sim')」が100の単語が正解単語です。

src/functions.py

@@ -7,25 +7,25 @@ guess_word = {"name": "guess_word",
                               "type": "string",
                               "description": "A single Japanese word to guess, which is can be a noun, verb, adverb or adjective. e.g. 空, 近い, 行く, etc."
                               },
-                          "puzzle_num": {
-                              "type": "integer",
-                              "description": "An index indicating today's puzzle."
-                          }
+                        #   "puzzle_num": {
+                        #       "type": "integer",
+                        #       "description": "An index indicating today's puzzle."
+                        #   }
                       },
-                      "required": ["word", "puzzle_num"]
+                      "required": ["word"]
                   }}
 
 lookup_answer = {"name": "lookup_answer",
-                "description": "You can check what the correct answer of today's puzzle is by this function.",
+                "description": "Use this function to check the correct answer of today's puzzle.",
                 "parameters": {
                     "type": "object",
                     "properties": {
-                        "puzzle_num": {
-                            "type": "integer",
-                            "description": "An index indicating today's puzzle."
+                        # "puzzle_num": {
+                        #     "type": "integer",
+                        #     "description": "An index indicating today's puzzle."
                         }
-                    },
-                    "required": ["puzzle_num"]
+                    # },
+                    # "required": ["puzzle_num"]
                 }}
 
 retrieve_puzzle_num = {"name": "retrieve_puzzle_num",
@@ -52,14 +52,18 @@ update_history = {"name": "update_history",
                       },
                       "required": ["current_guess", "guess_history"]
                   }}
-retrieve_rule = {"name": "retrieve_rule",
-                 "description": "Use this function to retrieve the game rule for clarification of your response.",
-                 "parametrs": {
+read_rule = {"name": "read_rule",
+                 "description": "Use this function to read the game rule for clarification of your response.",
+                 "parameters": {
                      "type": "object",
                      "properties": {},
                  }}
 
 
 def get_functions():
-    functions = [guess_word, lookup_answer, retrieve_puzzle_num, update_history, retrieve_rule]
+    functions = [guess_word, 
+                 lookup_answer, 
+                #  retrieve_puzzle_num, 
+                #  update_history, 
+                 read_rule]
     return functions

src/semantle.py

@@ -2,7 +2,8 @@ from datetime import date, datetime
 from pytz import utc, timezone
 import requests
 
-def get_secret(puzzle_num: int):
+def get_secret():
+    puzzle_num = get_puzzle_num()
     request_url = f"https://semantoru.com/yesterday/{puzzle_num+1}"
     response = requests.get(request_url, timeout=5)
     if response.status_code == 200:
@@ -10,12 +11,20 @@ def get_secret(puzzle_num: int):
     else:
         return "Not found error."
 
-def get_guess(word: str, puzzle_num: int):
+def get_guess(word: str):
+    puzzle_num = get_puzzle_num()
     request_url = f"https://semantoru.com/guess/{puzzle_num}/{word}"
+    print(request_url)
     response = requests.get(request_url, timeout=5)
     print(response.status_code)
     if response.status_code == 200:
-        return response.json()
+        rtn = response.json()
+        print(rtn)
+        if rtn['rank'] == '正解!':
+            return rtn
+        elif rtn['rank'] > 1000:
+            rtn['rank'] = '?'
+            return rtn
     else:
         return {"guess": word, 
                 "sim": None,

src/utils.py

@@ -1,18 +1,3 @@
-
-
-def add_guess(guess_result, guessed, guesses):
-    word, sim, rank = guess_result
-    if sim:
-        if word not in guessed:
-            sim = round(sim, 2)
-            rank = "情報なし" if rank == 1001 else rank
-            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(guesses)]), guessed, guesses
-
 def format_result(word, sim, rank):
     return f"{word}: スコア {sim}, ランク {rank}"