Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 from transformers.utils import logging as hf_logging
 
 # ---------------------------------------------------------------------------
-# Persistent cache + request log
+# Logging helpers
 # ---------------------------------------------------------------------------
 os.environ["HF_HOME"] = "/data/.huggingface"
 LOG_FILE = "/data/requests.log"
@@ -24,23 +24,22 @@ def log(msg: str):
 # ---------------------------------------------------------------------------
 # Configuration
 # ---------------------------------------------------------------------------
-MODEL_ID = "ibm-granite/granite-3.3-2b-instruct"  # 2‑B fits HF CPU Space
-MAX_TURNS = 4               # keep last N user/AI pairs
+MODEL_ID = "ibm-granite/granite-3.3-2b-instruct"
+MAX_TURNS = 4               # retain last N user/AI exchanges
 MAX_TOKENS = 64
 MAX_INPUT_CH = 300
 
 SYSTEM_MSG = (
-    "You are **SchoolSpirit AI**, the official digital mascot for "
-    "SchoolSpirit AI LLC, founded by Charles Norton in 2025. The company "
-    "specializes in on‑prem AI chat mascots, custom fine‑tuning of language "
-    "models, and turnkey GPU servers for K‑12 schools and education vendors.\n\n"
+    "You are **SchoolSpirit AI**, the digital mascot for SchoolSpirit AI LLC, "
+    "founded by Charles Norton in 2025. The company installs on‑prem AI chat "
+    "mascots, offers custom fine‑tuning of language models, and ships turnkey "
+    "GPU hardware to K‑12 schools.\n\n"
     "GUIDELINES:\n"
-    "• Respond in a warm, encouraging tone suitable for students, parents, "
-    "and staff.\n"
-    "• Keep answers concise (≤ 4 sentences) unless asked for detail.\n"
-    "• If unsure or out of scope, say you’re not sure and offer human follow‑up.\n"
-    "• No personal data collection, no medical/legal/financial advice.\n"
-    "• Maintain professionalism—no profanity, politics, or mature themes."
+    "• Use a warm, encouraging tone fit for students, parents, and staff.\n"
+    "• Keep replies short—no more than four sentences unless asked.\n"
+    "• If you’re unsure or out of scope, say so and suggest human follow‑up.\n"
+    "• Never collect personal data or provide medical, legal, or financial advice.\n"
+    "• No profanity, politics, or mature themes."
 )
 
 WELCOME_MSG = "Welcome to SchoolSpirit AI! Do you have any questions?"
@@ -78,13 +77,17 @@ trim = lambda m: m if len(m) <= 1 + MAX_TURNS * 2 else [m[0]] + m[-MAX_TURNS * 2
 
 
 def chat_fn(user_msg: str, history: list):
+    """
+    Gradio passes:
+      user_msg : str
+      history  : list[dict]  -> [{'role':'assistant'|'user','content':...}, ...]
+    Return a string; ChatInterface will append it as assistant message.
+    """
     log(f"User sent {len(user_msg)} chars")
-    # Seed system + welcome messages on first call
+
+    # Inject system message once
     if not history or history[0]["role"] != "system":
-        history = [
-            {"role": "system", "content": SYSTEM_MSG},
-            {"role": "assistant", "content": WELCOME_MSG},
-        ]
+        history.insert(0, {"role": "system", "content": SYSTEM_MSG})
 
     if MODEL_ERR:
         return MODEL_ERR
@@ -128,7 +131,7 @@ gr.ChatInterface(
     chatbot=gr.Chatbot(
         height=480,
         type="messages",
-        value=[("", WELCOME_MSG)],  # pre-populate AI welcome bubble
+        value=[{"role": "assistant", "content": WELCOME_MSG}],  # preloaded welcome
     ),
     title="SchoolSpirit AI Chat",
     theme=gr.themes.Soft(primary_hue="blue"),