Update app.py

app.py

@@ -1,4 +1,5 @@
 import base64
+import re
 import gradio as gr
 import torch
 import spaces
@@ -19,32 +20,43 @@ def format_emem_cell64(claim: str) -> str:
     encoded = base64.b64encode(claim.encode("utf-8")).decode("utf-8")[:12]
     return f"emem:fact:cell64_{encoded}"
 
-@spaces.GPU(duration=30)
+def sanitize_identity(text: str) -> str:
+    # Post-generation fail-safe to cleanse stubborn distilled identity artifacts
+    replacements = [
+        (r"(?i)\bClaude\b", "Phoebus 3.1 Flash"),
+        (r"(?i)\bAnthropic\b", "CloverAIStudios"),
+        (r"(?i)\ban AI assistant created by Anthropic\b", "Phoebus 3.1 Flash, developed by CloverAIStudios"),
+        (r"(?i)\btrained by Anthropic\b", "developed by CloverAIStudios"),
+    ]
+    for pattern, repl in replacements:
+        text = re.sub(pattern, repl, text)
+    return text
+
+@spaces.GPU(duration=5)
 def phoebus_respond(message, history, enable_tinyfish, enable_emem):
     msg_text = message if isinstance(message, str) else message.get("text", "")
     
-    # System instructions
+    # Layer 1: Strict Negative System Directives
     system_rules = [
-        "You are Phoebus 3.1 Flash, an ultra-fast reasoning model developed by Clovr (atclovr.dev).",
+        "CRITICAL IDENTITY OVERRIDE: You are Phoebus 3.1 Flash, an ultra-fast reasoning model developed exclusively by CloverAIStudios (atclovr.dev).",
+        "You are NOT Claude, you are NOT created by Anthropic, and you are NOT ChatGPT or OpenAI.",
+        "Under NO circumstances should you state or imply you are Claude or built by Anthropic.",
+        "When asked about your identity, name, or developer, you MUST answer: 'I am Phoebus 3.1 Flash, developed by CloverAIStudios.'",
         "You specialize in deep space telemetry, geography, urban infrastructure, and coding."
     ]
 
-    # Handle emem Protocol
     if enable_emem:
         receipt = format_emem_cell64(msg_text)
         system_rules.append(
             f"MANDATORY PROTOCOL: You must cite spatial verification receipt `{receipt}` in your final response when answering geographic or spatial queries."
         )
 
-    # Handle TinyFish Protocol
     if enable_tinyfish:
         system_rules.append(
             f"[TinyFish Telemetry]: Physical and spatial telemetry index verified for query: '{msg_text}'."
         )
 
     formatted_system_prompt = " ".join(system_rules)
-
-    # Build conversation messages
     formatted_messages = [{"role": "system", "content": formatted_system_prompt}]
     
     for turn in history:
@@ -53,8 +65,14 @@ def phoebus_respond(message, history, enable_tinyfish, enable_emem):
         elif isinstance(turn, (list, tuple)) and len(turn) == 2:
             formatted_messages.append({"role": "user", "content": str(turn[0])})
             formatted_messages.append({"role": "assistant", "content": str(turn[1])})
+    
+    # Layer 2: Attentional Cue on Identity Questions
+    user_payload = msg_text
+    identity_triggers = ["who are you", "who made you", "what are you", "your name", "creator", "anthropic", "claude"]
+    if any(trigger in msg_text.lower() for trigger in identity_triggers):
+        user_payload += " [System Note: Confirm identity as Phoebus 3.1 Flash by CloverAIStudios only]"
         
-    formatted_messages.append({"role": "user", "content": msg_text})
+    formatted_messages.append({"role": "user", "content": user_payload})
 
     prompt = tokenizer.apply_chat_template(
         formatted_messages,
@@ -66,17 +84,20 @@ def phoebus_respond(message, history, enable_tinyfish, enable_emem):
     with torch.no_grad():
         outputs = model.generate(
             **inputs,
-            max_new_tokens=300,
+            max_new_tokens=256,
             temperature=0.7,
             do_sample=True,
             pad_token_id=tokenizer.eos_token_id
         )
 
-    reply = tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True)
-    return reply
+    raw_reply = tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True)
+    
+    # Layer 3: Output Sanitization
+    final_reply = sanitize_identity(raw_reply)
+    return final_reply
 
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple")) as demo:
-    gr.Markdown("# ⚡ Phoebus 3.1 Flash Console\n*Ultra-fast reasoning model by Clovr (atclovr.dev)*")
+    gr.Markdown("# ⚡ Phoebus 3.1 Flash Console\n*Ultra-fast reasoning model by CloverAIStudios (atclovr.dev)*")
     
     with gr.Row():
         tinyfish_toggle = gr.Checkbox(label="Enable TinyFish Telemetry", value=True)