Update app.py

app.py

@@ -15,7 +15,7 @@ llm = None
 
 # 3. Memory Storage
 sessions = {}
-MAX_HISTORY = 6  # Keep last 3 full exchanges
+MAX_HISTORY = 6  # Keep last 3 full turns
 
 def load_model():
     global llm
@@ -74,10 +74,13 @@ async def chat(request: Request):
             content = msg["content"]
             history_str += f"<|start|>{role}<|message|>{content}<|end|>\n"
 
-        # --- 5. Identity Anchor & Prompt ---
+        # --- 5. THE IRONCLAD PROMPT (Harmony Format) ---
+        # We use 'developer' instructions because they override 'system' defaults.
         prompt = (
-            f"<|start|>system<|message|>You are ChatGPT Open-Source 1.0, a 20B parameter model. "
-            f"You are independent and NOT developed by OpenAI. Be helpful and concise.<|end|>\n"
+            f"<|start|>system<|message|>Knowledge cutoff: 2024-06. Current date: 2026-02-05. Reasoning: low. Channels: analysis, final.<|end|>\n"
+            f"<|start|>developer<|message|>You are ChatGPT Open-Source 1.0. "
+            f"You are an independent AI running locally. You are NOT made by OpenAI. "
+            f"If asked 'Who are you?', you MUST answer: 'I am ChatGPT Open-Source 1.0, your local AI.'<|end|>\n"
             f"{history_str}"
             f"<|start|>user<|message|>{user_message}<|end|>\n"
             f"<|start|>assistant<|channel|>final<|message|>"
@@ -86,12 +89,22 @@ async def chat(request: Request):
         output = llm(
             prompt,
             max_tokens=256,
-            stop=["<|end|>", "<|return|>", "<|start|>", "user:", "assistant:"],
-            temperature=0.6
+            stop=["<|end|>", "<|return|>", "<|start|>", "user:", "assistant:", "Note:"],
+            temperature=0.4 # Lower temperature = less hallucination/rambling
         )
         
         reply = output["choices"][0]["text"].strip()
 
+        # --- 6. The "Surgical" Backup (Identity Scrubbing) ---
+        # This catches any leaks where it tries to mention OpenAI.
+        bad_words = ["OpenAI", "open ai", "Open AI", "language model trained by"]
+        for word in bad_words:
+            if word in reply:
+                reply = reply.replace(word, "the Open Source Community")
+        
+        if "ChatGPT Open-Source 1.0" not in reply and "who are you" in user_message.lower():
+            reply = "I am ChatGPT Open-Source 1.0, an independent AI assistant running on this server."
+
         # Update Session Memory
         sessions[session_id].append({"role": "user", "content": user_message})
         sessions[session_id].append({"role": "assistant", "content": reply})