:sparkles: Create a semi-working buggy first version

app.py

@@ -6,7 +6,7 @@ logging.basicConfig(level="DEBUG")
 
 
 def main():
-    demo = interface.get_demo(wip=True)
+    demo = interface.get_demo(wip=False)
     demo.launch(debug=True)
 
 

src/constants.py

@@ -1,3 +1,4 @@
 MAX_ATTEMPTS = 3
 
-STARTING_INDEX = 20
+STARTING_INDEX = 50
+END_WORD_NUMBER = 10

src/handlers.py

@@ -2,6 +2,7 @@ import logging
 
 from transformers import PreTrainedTokenizer
 
+from src.constants import MAX_ATTEMPTS
 from src.params import ReducerParams
 from src.shared import all_tokens
 from src.utils import get_current_prompt_text
@@ -42,8 +43,7 @@ def handle_tie(params: ReducerParams) -> ReducerParams:
     return params
 
 
-def handle_next_attempt(params: ReducerParams) -> ReducerParams:
-    params.remaining_attempts -= 1
+def handle_both_wrong(params: ReducerParams) -> ReducerParams:
     params.bottom_html = f"That was not it... {params.remaining_attempts} attempts left"
     return params
 
@@ -58,4 +58,7 @@ def handle_next_word(params: ReducerParams) -> ReducerParams:
     params.button_label = "Guess!"
     params.bottom_html = ""
     params.prompt_text = get_current_prompt_text(params.word_number)
+    params.current_guesses = ""
+    params.remaining_attempts = MAX_ATTEMPTS
+    params.guess_field = ""
     return params

src/interface.py

@@ -4,6 +4,7 @@ from src import reducer
 from src import shared
 from src.constants import MAX_ATTEMPTS
 from src.constants import STARTING_INDEX
+from src.utils import get_current_lm_guess_str
 
 
 def build_demo():
@@ -42,7 +43,9 @@ def build_demo():
                 )
                 current_guesses = gr.Textbox(label="Your guesses")
             with gr.Column():
-                lm_guesses = gr.Textbox(label="LM guesses")
+                lm_guesses = gr.Textbox(
+                    label="LM guesses", value=get_current_lm_guess_str(0, remaining_attempts=MAX_ATTEMPTS)
+                )
 
         with gr.Row():
             with gr.Column():

src/reducer.py

@@ -1,14 +1,19 @@
 import logging
 
+from src.constants import END_WORD_NUMBER
+from src.constants import STARTING_INDEX
+from src.handlers import handle_both_wrong
 from src.handlers import handle_lm_win
-from src.handlers import handle_next_attempt
 from src.handlers import handle_next_word
 from src.handlers import handle_no_input
 from src.handlers import handle_out_of_attempts
 from src.handlers import handle_player_win
 from src.handlers import handle_tie
 from src.params import ReducerParams
+from src.shared import token_predictions
 from src.shared import tokenizer
+from src.utils import get_current_lm_guess_str
+from src.utils import get_current_prompt_text
 from src.utils import guess_is_correct
 from src.utils import lm_is_correct
 
@@ -24,25 +29,55 @@ def handle_guess(*args) -> ReducerParams:
 
 
 def _handle_guess(params: ReducerParams) -> ReducerParams:
-    logger.debug(params)
+    if params.word_number >= END_WORD_NUMBER:
+        # FIXME: BAD PRACTICE
+        if params.player_points > params.lm_points:
+            s = "YOU WIN"
+        elif params.player_points < params.lm_points:
+            s = "YOU LOSE"
+        else:
+            s = "CAN YOU BELIEVE IT!? A TIE"
+
+        return ReducerParams(
+            "",
+            params.player_points,
+            params.lm_points,
+            "",
+            "",
+            None,
+            "",
+            "Restart",
+            s,
+            0,
+        )
 
+    logger.debug(params)
+    logger.debug(token_predictions[STARTING_INDEX + params.word_number])
     if params.button_label == "Next word":
-        return handle_next_word(params)
-
-    if not params.guess_field:
-        return handle_no_input(params)
-
-    params.current_guesses += "\n" + params.guess_field
-    player_correct = guess_is_correct(params, tokenizer)
-    lm_correct = lm_is_correct(params)
-
-    if player_correct and lm_correct:
-        return handle_tie(params)
-    elif player_correct and not lm_correct:
-        return handle_player_win(params)
-    elif lm_correct and not player_correct:
-        return handle_lm_win(params)
-    elif params.remaining_attempts == 0:
-        return handle_out_of_attempts(params)
+        params = handle_next_word(params)
+    elif not params.guess_field:
+        params = handle_no_input(params)
     else:
-        return handle_next_attempt(params)
+
+        params.current_guesses += params.guess_field + "\n"
+        player_correct = guess_is_correct(params, tokenizer)
+        lm_correct = lm_is_correct(params)
+
+        if params.remaining_attempts > 0:
+            params.remaining_attempts -= 1
+            if player_correct and lm_correct:
+                params = handle_tie(params)
+            elif player_correct and not lm_correct:
+                params = handle_player_win(params)
+            elif lm_correct and not player_correct:
+                params = handle_lm_win(params)
+            else:
+                params = handle_both_wrong(params)
+
+        else:
+            params = handle_out_of_attempts(params)
+
+    params.lm_guesses = get_current_lm_guess_str(params.word_number, params.remaining_attempts)
+    params.prompt_text = get_current_prompt_text(params.word_number)
+
+    return params

src/text.py

@@ -1,14 +1,30 @@
-target_text = """\
-Obama was born in Honolulu, Hawaii.
-After graduating from Columbia University in 1983, he worked as a community organizer in Chicago.
-In 1988, he enrolled in Harvard Law School, where he was the first black president of the Harvard Law Review.
-After graduating, he became a civil rights attorney and an academic, teaching constitutional law at the University of Chicago Law School from 1992 to 2004.
-Turning to elective politics, he represented the 13th district in the Illinois Senate from 1997 until 2004, when he ran for the U.S.
-Senate.
-Obama received national attention in 2004 with his March Senate primary win, his well-received July Democratic National Convention keynote address, and his landslide November election to the Senate.
-In 2008, after a close primary campaign against Hillary Clinton, he was nominated by the Democratic Party for president and chose Joe Biden as his running mate.
-Obama was elected over Republican nominee John McCain in the presidential election and was inaugurated on January 20, 2009.
-Nine months later, he was named the 2009 Nobel Peace Prize laureate, a decision that drew a mixture of praise and criticism.
+# target_text = """\
+# Obama was born in Honolulu, Hawaii.
+# After graduating from Columbia University in 1983, he worked as a community organizer in Chicago.
+# In 1988, he enrolled in Harvard Law School, where he was the first black president of the Harvard Law Review.
+# After graduating, he became a civil rights attorney and an academic, teaching constitutional law at the University of Chicago Law School from 1992 to 2004.
+# Turning to elective politics, he represented the 13th district in the Illinois Senate from 1997 until 2004, when he ran for the U.S.
+# Senate.
+# Obama received national attention in 2004 with his March Senate primary win, his well-received July Democratic National Convention keynote address, and his landslide November election to the Senate.
+# In 2008, after a close primary campaign against Hillary Clinton, he was nominated by the Democratic Party for president and chose Joe Biden as his running mate.
+# Obama was elected over Republican nominee John McCain in the presidential election and was inaugurated on January 20, 2009.
+# Nine months later, he was named the 2009 Nobel Peace Prize laureate, a decision that drew a mixture of praise and criticism.
+# """
+# target_text = """\
+# Biology
+# Metapod is an insect Pokémon that resembles a green chrysalis.
+# Its body is crescent-shaped with several segments making up the lower point.
+# The front of Metapod's shell resembles a face with heavy-lidded eyes and a sharply pointed nose.
+# The back of its shell consists of several geometrically shaped portions and projections.
+
+# Metapod's soft body is protected by a hard outer shell as it undergoes metamorphosis.
+# While this shell is said to be as hard as steel, a sudden, powerful impact could cause its liquid innards to pop out, leaving it completely exposed. Metapod generally remains motionless, rebuilding its cells for evolution.
+# If an enemy discovers Metapod, it is unable to do anything other than harden its outer shell. Metapod lives in temperate forests and jungles, often in groups. Pikipek is a natural predator of Metapod.
+# """
+target_text = """
+The Icelandic horse (Icelandic: íslenski hesturinn [ˈistlɛnscɪ ˈhɛstʏrɪn]) is a breed of horse developed in Iceland. Although the horses are small, at times pony-sized, most registries for the Icelandic refer to it as a horse. Icelandic horses are long-lived and hardy. In their native country they have few diseases; Icelandic law prevents horses from being imported into the country and exported animals are not allowed to return. The Icelandic displays two gaits in addition to the typical walk, trot, and canter/gallop commonly displayed by other breeds. The only breed of horse in Iceland, they are also popular internationally, and sizable populations exist in Europe and North America. The breed is still used for traditional sheepherding work in its native country, as well as for leisure, showing, and racing.
+
+Developed from ponies taken to Iceland by Norse settlers in the 9th and 10th centuries, the breed is mentioned in literature and historical records throughout Icelandic history; the first reference to a named horse appears in the 12th century. Horses were venerated in Norse mythology, a custom brought to Iceland by the country's earliest settlers. Selective breeding over the centuries has developed the breed into its current form. Natural selection has also played a role, as the harsh Icelandic climate eliminated many horses through exposure and malnourishment. In the 1780s, much of the breed was wiped out in the aftermath of a volcanic eruption at Laki. The first breed society for the Icelandic horse was created in Iceland in 1904, and today the breed is represented by organizations in 19 different nations, organized under a parent association, the International Federation of Icelandic Horse Associations.
 """
 
 

src/utils.py

@@ -11,7 +11,19 @@ from src.shared import token_id_predictions
 logger = logging.getLogger(__name__)
 
 
+def get_current_lm_guess_str(word_number, remaining_attempts):
+    # FIXME: indexerror
+    guess_list = token_id_predictions[(STARTING_INDEX + word_number) - 1][1]
+    guess_list = [shared.tokenizer.decode(i) for i in guess_list]
+    censored_list = ["*****"] * MAX_ATTEMPTS
+    for i in range(MAX_ATTEMPTS - remaining_attempts):
+        censored_list[i] = guess_list[i]
+
+    return "\n".join(censored_list)
+
+
 def get_current_prompt_text(word_number):
+    # FIXME: indexerror
     return shared.tokenizer.decode(shared.all_tokens[: STARTING_INDEX + word_number])
 
 
@@ -34,8 +46,9 @@ def lm_is_correct(params: ReducerParams) -> bool:
 
     idx = MAX_ATTEMPTS - params.remaining_attempts
 
-    current_guess = token_id_predictions[params.word_number][1][idx]
-    current_target = token_id_predictions[params.word_number][0]
+    # FIXME: indexerror
+    current_guess = token_id_predictions[STARTING_INDEX + params.word_number - 1][1][idx]
+    current_target = token_id_predictions[STARTING_INDEX + params.word_number - 1][0]
 
     return current_guess == current_target