Update app.py

app.py

@@ -9,9 +9,9 @@ For more information on `huggingface_hub` Inference API support, please check th
 #client = InferenceClient("DrDomedag/LocoLama")
 #client = InferenceClient("unsloth/llama-3.2-1b-instruct-bnb-4bit")
 #client = InferenceClient("unsloth/llama-3.2-1b-bnb-4bit")
-#client = InferenceClient("unsloth/Llama-3.2-1B-Instruct") # Works!
+client = InferenceClient("unsloth/Llama-3.2-1B-Instruct") # Works!
 #client = InferenceClient("unsloth/Llama-3.2-3B-Instruct") # Works!
-client = InferenceClient("DrDomedag/LocoLamav2")
+#client = InferenceClient("DrDomedag/LocoLamav2")
 #client = InferenceClient("T3lli/lora_model")
 
 
@@ -54,7 +54,7 @@ words = [
 
 class WordGame:
 
-    def __init__(self):
+    def __init__(self, ui):
         self.points = 0
         self.target_word = ""
         self.attempts = 3
@@ -69,6 +69,8 @@ class WordGame:
         if self.target_word in string:
             print(f"The player input the target word and the task was reset.")
             self.generate_task()
+            self.points -= 1
+            self.update_status(message="You input the target word yourself, so you lost one point and the game reset.")
         else:
             print(f"The player did not input the target word, so that's good.")
             
@@ -77,7 +79,9 @@ class WordGame:
         if self.target_word in string:
             self.points += self.attempts
             print(f"The player scored {self.attempts} points and has a total of {self.points}.")
+            score_gained = self.attempts
             self.generate_task()
+            self.update_status(message=f"Success! You earned {score_gained} points!")
         else:
             print("The response did not contain the word.")
             self.attempts -= 1
@@ -85,11 +89,15 @@ class WordGame:
             if self.attempts <= 0:
                 self.generate_task()
                 print(f"The player ran out of attempts.")
+                self.update_status(message=f"You did not win in three attempts. Generating new target word.")
             else:
                 print(f"The player has attempts left.")
+
+    def update_status(self, message=""):
+        demo.title = f'Current score: {self.points}, remaining attempts: {self.attempts}, target word: "{self.target_word}" {message}'
     
 
-game = WordGame()
+
 
 
 
@@ -138,6 +146,7 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 """
 demo = gr.ChatInterface(
     respond,
+    title="Hi Elli",
     additional_inputs=[
         gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
@@ -152,6 +161,8 @@ demo = gr.ChatInterface(
     ],
 )
 
+game = WordGame(demo)
+
 
 if __name__ == "__main__":
     demo.launch()