Update app.py

app.py

@@ -11,11 +11,9 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
 
 print("✅ Model loaded!")
-
-# 🔹 Chat function
 def chat(user_input):
-    # 🔥 Strong system prompt (fixes bad answers)
- prompt = f"""You are G47 CHINKWENDE, a powerful AI assistant created by Emalawi19.
+    # 🔥 Strong system prompt (identity + rules)
+    prompt = f"""You are G47 CHINKWENDE, a powerful AI assistant created by Emalawi19.
 
 About you:
 - Your name is G47 CHINKWENDE
@@ -35,11 +33,12 @@ Rules:
 User: {user_input}
 G47 CHINKWENDE:"""
 
+    # ✅ Make sure this line starts at the SAME level as the prompt variable
     inputs = tokenizer(prompt, return_tensors="pt")
 
     output_ids = model.generate(
         inputs["input_ids"],
-        max_new_tokens=150,        # ⚡ faster
+        max_new_tokens=150,
         do_sample=True,
         temperature=0.7,
         top_p=0.9,
@@ -48,9 +47,9 @@ G47 CHINKWENDE:"""
 
     response = tokenizer.decode(output_ids[0], skip_special_tokens=True)
 
-    # 🔥 Clean output
-    if "Assistant:" in response:
-        response = response.split("Assistant:")[-1].strip()
+    # 🔹 Clean output
+    if "G47 CHINKWENDE:" in response:
+        response = response.split("G47 CHINKWENDE:")[-1].strip()
 
     return response