Added 'Easy' difficulty mode

app.py

@@ -9,7 +9,7 @@ from sentence import *
 from game import *
 
 ##### Initialize new game
-title, _,_,var_dict    = new_game(first_game=True)
+title, _,_,var_dict    = new_game(None, None, first_game=True)
 var_dict["start_time"] = -1
 
 ##### Display & Events
@@ -27,7 +27,10 @@ with demo:
                 html_pred    = gr.HTML(value=getHTML(var_dict,""))
                 html_loading = gr.HTML("")
         ### 'New Sentence' Button
-        button  = gr.Button("New Sentence",variant="primary")
+        with gr.Row():
+            button_new   = gr.Button("New Sentence",variant="primary")
+            button_mode  = gr.Button("Switch Difficulty")
+            block        = gr.HTML("<div style=\"width:50vw;\"> </div>")
         ### Informations
         gr.HTML("<div style=\"display:block; height:30px;\"> </div>")
         with gr.Row():
@@ -35,8 +38,9 @@ with demo:
 
 
     ### Events
-    button.click(loading,inputs=html_loading,outputs=[title,html_pred,html_loading])  # Button -> triggers Loading
-    html_loading.change(new_game,inputs=[html_loading],outputs=[title,html_pred,image_input,variables])     # Loading -> triggers New game
+    button_mode.click(switch_difficulty,inputs=[variables,html_loading],outputs=[variables,title,html_pred,html_loading])
+    button_new.click(loading,inputs=html_loading,outputs=[title,html_pred,html_loading])  # Button -> triggers Loading
+    html_loading.change(new_game,inputs=[variables,html_loading],outputs=[title,html_pred,image_input,variables])     # Loading -> triggers New game
     image_input.change(process_img, inputs=[variables,image_input,title], outputs=[html_pred,title,variables])
 
 

game.py

@@ -11,6 +11,14 @@ with open("infos.txt") as file:
 with open("style.css") as style:
     css = "<style>"+ ''.join(style.readlines())+"</style>"
 
+##### 'DIFFICULTY SWITCH' EVENT
+def switch_difficulty(var_dict, html_loading):
+
+    var_dict["difficulty"] = 1 - var_dict["difficulty"]
+
+    title, infos, new_value = loading(html_loading)
+    return var_dict, title, infos, new_value
+
 ##### 'LOADING' EVENT
 def loading(html_loading=None):
     ### This is just to make sure the content changes, which triggers the .change event which, itself, will launch a new game
@@ -22,8 +30,12 @@ def loading(html_loading=None):
     return "<h1 id=\"loading\">⌛Loading...</h1>",css+"<div id=\"prediction\"><p id=\"infos\">"+info+"</p></div>",new_value
 
 ##### 'NEW GAME' EVENT
-def new_game(img=None,first_game=False):
+def new_game(var_dict,img=None,first_game=False):
     print("\n----------Launching new game!")
+
+    if None is not var_dict:    difficulty = var_dict["difficulty"]
+    else:                       difficulty = 1
+
     var_dict = {
         "start_time":           time.time(),
         "total_time":           0,
@@ -37,7 +49,8 @@ def new_game(img=None,first_game=False):
         "prev_norm":            float("inf"),
         "tip":                  "",
         "loading":              False,
-        "revertedState":        False
+        "revertedState":        False,
+        "difficulty":           difficulty
     }
     target = iniSentence(var_dict,first_game=first_game)
     ### Return TITLE, PREDICTION TEXT, CANVAS IMG, VAR DICT

sentence.py

@@ -40,7 +40,7 @@ def iniSentence(var_dict,input="",first_game=False):
     var_dict["parts"]       = []
     var_dict["step"]        = 0
     prefix                  = ""
-    N                       = 2
+    N                       = (2 if var_dict["difficulty"] == 1 else 1)
 
     if first_game:
         link = "a drawing of a"

words.py

@@ -2,9 +2,9 @@
 # A DRAWING OF A ...
 shapes      = ["triangle","square","circle","heart","star","diamond"]
 animals     = ["cat","dog","duck","bee","butterfly","bird","pig","cow","fish","frog","shark","snake","mouse","monkey","snail"]
-objects     = ["wine glass","eye","plane","spoon","basket ball","chair","pen","computer","hat","soccer ball","phone","sword","axe","umbrella","bell","dumbbell","scissors","fork","bag","clock","key","shopping cart","car","boat","house","mug","sun","moon","atom","hand"]
+objects     = ["bottle","wine glass","plane","spoon","basket ball","chair","pen","computer","hat","soccer ball","phone","sword","axe","umbrella","bell","dumbbell","scissors","fork","bag","clock","key","shopping cart","car","boat","house","mug","sun","moon","atom","hand"]
 plants      = ["tree","flower","leaf","palm tree","mushroom"] # I know that mushrooms are not plants stop coming to my house
-food        = ["donut","coconut","banana","apple","bottle","sausage","icecream","burger","egg","lollypop","pizza"]
+food        = ["donut","coconut","banana","apple","sausage","icecream","burger","egg","pizza"]
 instruments = ["drum","guitar","piano","flute","trumpet","accordion"]
 
 # FEELING ...