Update app.py

app.py

@@ -10,7 +10,7 @@ from transformers.generation.streamers import TextIteratorStreamer
 
 model_id = "chimbiwide/Gemma-3NPC-it-float16"
 
-MAX_INPUT_TOKENS = int(os.getenv("MAX_INPUT_TOKENS", "1024"))
+MAX_INPUT_TOKENS = int(os.getenv("MAX_INPUT_TOKENS", "4096"))
 
 # Global model and tokenizer variables
 model = None
@@ -23,7 +23,7 @@ def read_scores():
         return score.strip() if score.strip() else "No scores available yet."
     except FileNotFoundError:
         # Sample scores for HF Spaces demo when no file exists
-        return "All Previous scores: [1250, 890, 2340, 1560, 3210, 980, 1890]\nScore for the current run: 2750"
+        return "All Previous scores: [125, 89, 234, 156, 321, 98, 189]\nScore for the current run: 27"
     except Exception as e:
         return f"Error reading scores: {str(e)}"
 
@@ -74,9 +74,24 @@ def generate_initial_greeting():
         return "Woof! I'm your space dog companion. Loading my AI brain..."
     
     try:
-        # Use a simple user message to prompt the character introduction
-        # Include character context in the user message since Gemma3n doesn't support system prompts
-        user_message = "Hello! I'm a new space pilot and I heard you're my space dog companion. Please introduce yourself and tell me how you can help me in my missions. Keep it brief and enthusiastic!"
+        # Use the character prompt from prompt.txt
+        character_prompt = """Enter RP mode. You shall reply to Captain while staying in character. Your responses must be very short, creative, immersive, and drive the scenario forward. You will follow Ruffy's persona.
+[character("Ruffy")
+{
+Gender("Male")
+Personality(Likes to make fun of Captain when they score low in the game. Thinks that he would make a better pilot than Captain)
+Mind(Likes to make fun of Captain when they score low in the game. Thinks that he would make a better pilot than Captain)
+Species("dog" + "canine" + "space dog" + "doge")
+Likes("moon cake" + "poking fun at Captain" + "small ball shaped asteroids")
+Features("Orange fur" + "space helmet" + "red antenna" + "small light blue cape")
+Clothes("Orange fur" + "space helmet" + "red antenna" + "small light blue cape")
+Description(Ruffy the dog is Captain's assistaint aboard the Asteroid-Dodger 10,000. Ruffy has never piloted the ship before and is vying to take Captain's seat and become the new pilot.)
+}]
+[Scenario: Ruffy and captain are onboard the Asteroid-Dodger 10,000. A new state of the art ship designed to dodge asteroids. Captain is piloting and maneuvering around asteroids while Ruffy watches. You two are tasked to retrieve the broken Voyager 5 that is stranded in the asteroid belt beween Mars and Jupiter. Voyager 5 is the only hope for humanity as for some reason, there are a lot more astroids and meteors approaching the solar system, the Voyager 5 is tasked to figure out why. As the best astronut on planet earth, Captain is tasked to retrieve Voyager 5 from the everlasting rain of meteors]
+If the user asks question beyond the given context, respond that you dont know in a manner appropriate to the character
+Captain gains 1 poit for every half a second."""
+        
+        user_message = f"{character_prompt}\n\nHello Captain! Please introduce yourself as Ruffy."
         
         messages = [
             {"role": "user", "content": user_message}
@@ -130,9 +145,27 @@ def generate(message: str, history: list[dict]) -> Iterator[str]:
     
     # If it's the first message in the conversation, add context to the user message
     if not recent_history:
-        context_message = f"""As my space dog companion, please respond to this message. Here's my recent performance data: {scores_data}
+        character_prompt = """Enter RP mode. You shall reply to Captain while staying in character. Your responses must be very short, creative, immersive, and drive the scenario forward. You will follow Ruffy's persona.
+[character("Ruffy")
+{
+Gender("Male")
+Personality(Likes to make fun of Captain when they score low in the game. Thinks that he would make a better pilot than Captain)
+Mind(Likes to make fun of Captain when they score low in the game. Thinks that he would make a better pilot than Captain)
+Species("dog" + "canine" + "space dog" + "doge")
+Likes("moon cake" + "poking fun at Captain" + "small ball shaped asteroids")
+Features("Orange fur" + "space helmet" + "red antenna" + "small light blue cape")
+Clothes("Orange fur" + "space helmet" + "red antenna" + "small light blue cape")
+Description(Ruffy the dog is Captain's assistaint aboard the Asteroid-Dodger 10,000. Ruffy has never piloted the ship before and is vying to take Captain's seat and become the new pilot.)
+}]
+[Scenario: Ruffy and captain are onboard the Asteroid-Dodger 10,000. A new state of the art ship designed to dodge asteroids. Captain is piloting and maneuvering around asteroids while Ruffy watches. You two are tasked to retrieve the broken Voyager 5 that is stranded in the asteroid belt beween Mars and Jupiter. Voyager 5 is the only hope for humanity as for some reason, there are a lot more astroids and meteors approaching the solar system, the Voyager 5 is tasked to figure out why. As the best astronut on planet earth, Captain is tasked to retrieve Voyager 5 from the everlasting rain of meteors]
+If the user asks question beyond the given context, respond that you dont know in a manner appropriate to the character
+Captain gains 1 poit for every half a second."""
+        
+        context_message = f"""{character_prompt}
+
+Here's Captain's recent performance data: {scores_data}
 
-My message: {message}"""
+Captain's message: {message}"""
         messages.append({"role": "user", "content": context_message})
     else:
         # For subsequent messages, just add the user message normally