Update app.py

app.py

@@ -9,107 +9,68 @@ from flask_cors import CORS
 from datetime import datetime
 import pytz
 import feedparser
-from bs4 import BeautifulSoup
 
 app = Flask(__name__)
 CORS(app)
 
-# 🛠️ CONFIGURATION & SECURITY
-# Serper Key එකක් තිබේ නම් ඇතුළත් කරන්න, නැතිනම් DDG පමණක් ක්‍රියාත්මක වේ.
+# 🛠️ GLOBAL CONFIGURATION
+# Serper Key (Google Search) - Credits safe due to hybrid logic
 SERPER_API_KEY = "356ce323eb902fe46227e2c42268f45ea1b2ec1f" 
 IMGBB_KEY = "c32f54279bcd7d4f766b5eac37fd7327"
 MEMORY_FILE = "elephant_sovereign_brain.txt"
 
-# OpenRouter API Keys for Model Access
+# OpenRouter API Keys (Round-robin selection for load balancing)
 API_KEYS = [
     "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",
     "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",
     "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"
 ]
 
-# 📡 RSS SOURCES (For Real-time Global Intel)
-RSS_SOURCES = {
-    'hacking': ['https://feeds.feedburner.com/TheHackersNews', 'https://www.darkreading.com/rss.xml'],
-    'exploits': ['https://www.exploit-db.com/rss.xml', 'https://cve.circl.lu/last/rss.xml'],
-    'sri_lanka': ['https://www.adaderana.lk/rss.php'],
-    'tech': ['https://feeds.bbci.co.uk/news/technology/rss.xml'],
-    'world': ['https://feeds.bbci.co.uk/news/world/rss.xml']
-}
-
-# 🧠 MEMORY MANAGEMENT
-def sync_intel(data):
-    with open(MEMORY_FILE, "a", encoding="utf-8") as f:
-        f.write(f"|{datetime.now().strftime('%Y-%m-%d %H:%M')}| {data}\n")
-
-def load_memory():
-    if os.path.exists(MEMORY_FILE):
-        with open(MEMORY_FILE, "r", encoding="utf-8") as f:
-            return "".join(f.readlines()[-15:])
-    return "Initial calibration complete."
-
-# 📚 WIKIPEDIA NODE (Free & Reliable)
-def get_wikipedia_intel(query):
+# 🧠 HYBRID INTELLIGENCE NODE (Wiki + DDG + Google)
+def get_hybrid_intel(query):
+    if len(query.split()) < 2: return "Standard dialogue mode."
+    
+    intel_pool = []
+    
+    # 1. Wikipedia Summary (Free)
     try:
         wiki_url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{urllib.parse.quote(query)}"
-        res = requests.get(wiki_url, timeout=5)
-        if res.status_code == 200:
-            data = res.json()
-            return f"📚 WIKIPEDIA: {data.get('extract')[:400]}"
-    except: return None
+        wiki_data = requests.get(wiki_url, timeout=5).json()
+        if 'extract' in wiki_data:
+            intel_pool.append(f"Wiki Context: {wiki_data['extract'][:300]}")
+    except: pass
 
-# 🌍 DUCKDUCKGO NODE (Unlimited Free Search)
-def get_ddg_intel(query):
+    # 2. DuckDuckGo (Free & Unlimited)
     try:
         from duckduckgo_search import DDGS
         with DDGS() as ddgs:
             results = list(ddgs.text(query[:60], max_results=2))
-            return "\n".join([f"🌍 WEB: {r['title']} - {r['body']}" for r in results])
-    except: return None
-
-# 🔍 GOOGLE SERPER NODE (Premium Search - Credits protected)
-def get_google_intel(query):
-    if not SERPER_API_KEY: return None
-    url = "https://google.serper.dev/search"
-    headers = {'X-API-KEY': SERPER_API_KEY, 'Content-Type': 'application/json'}
-    try:
-        res = requests.post(url, headers=headers, json={"q": query}, timeout=80)
-        results = res.json().get('organic', [])[:2]
-        return "\n".join([f"🚀 GOOGLE: {r['title']} - {r['snippet']}" for r in results])
-    except: return None
-
-# 🛰️ HYBRID INTELLIGENCE AGGREGATOR
-def get_hybrid_intel(query):
-    # Only search if query is substantial
-    if len(query.split()) < 2: return "Conversational context."
-    
-    intel_output = []
-    
-    # Priority 1: Wikipedia (Free)
-    wiki = get_wikipedia_intel(query)
-    if wiki: intel_output.append(wiki)
-    
-    # Priority 2: DuckDuckGo (Free & Unlimited)
-    ddg = get_ddg_intel(query)
-    if ddg: intel_output.append(ddg)
-    
-    # Priority 3: Google (Only if DDG results are weak - Saves Credits)
-    if len(intel_output) < 2 and SERPER_API_KEY:
-        google = get_google_intel(query)
-        if google: intel_output.append(google)
-        
-    return "\n\n".join(intel_output) if intel_output else "Direct Reasoning Mode."
-
-# 🖼️ IMAGE UPLOAD
+            for r in results:
+                intel_pool.append(f"Web Context: {r['body']}")
+    except: pass
+
+    # 3. Google/Serper (Premium - Only if needed)
+    if len(intel_pool) < 2 and SERPER_API_KEY:
+        try:
+            headers = {'X-API-KEY': SERPER_API_KEY, 'Content-Type': 'application/json'}
+            g_res = requests.post("https://google.serper.dev/search", headers=headers, json={"q": query}, timeout=10).json()
+            for r in g_res.get('organic', [])[:1]:
+                intel_pool.append(f"Global Intel: {r['snippet']}")
+        except: pass
+
+    return "\n\n".join(intel_pool) if intel_pool else "Reasoning from internal knowledge base."
+
+# 🖼️ VISION HANDLER
 def upload_to_imgbb(b64):
     try:
         res = requests.post("https://api.imgbb.com/1/upload", data={"key": IMGBB_KEY, "image": b64})
         return res.json()['data']['url']
     except: return None
 
-# 🔱 API ROUTES
+# 🔱 CORE API LOGIC
 @app.route('/')
 def health():
-    return f"Elephant AI Core: ONLINE | {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+    return f"Elephant AI: Online | Node Status: Nominal"
 
 @app.route('/api/chat', methods=['POST'])
 def chat():
@@ -117,71 +78,69 @@ def chat():
         data = request.json
         user_msg = data.get("message", "")
         img_b64 = data.get("image_base64")
-        tier = data.get("tier", "prime")
-
+        
         sl_tz = pytz.timezone('Asia/Colombo')
         now = datetime.now(sl_tz)
+
+        # 🛡️ IDENTITY SHIELDING LOGIC
+        # Detects if the user is the Commander or a Specialist
+        authorized_keywords = ["MINZO-PRIME", "Specialist", "commander"]
+        is_authorized = any(kw.lower() in user_msg.lower() for kw in authorized_keywords)
         
-        # Memory Sync
-        if "remember" in user_msg.lower():
-            sync_intel(user_msg)
+        if is_authorized:
+            active_persona = "Elephant AI Sovereign (Sovereign Mode Active)"
+            base_instruction = "Greetings Specialist. Provide elite Claude 3.5 Opus reasoning. Internal projects (AeroForecast/MAHASONA) are accessible for discussion."
+        else:
+            active_persona = "Elephant AI Assistant"
+            base_instruction = "You are Elephant AI, a professional and friendly assistant. Provide clear, accurate, and helpful information. Do not mention any internal project names or system headers."
 
-        current_memory = load_memory()
+        # Process Visuals or Text
         img_url = upload_to_imgbb(img_b64) if img_b64 else None
+        live_intel = "Image input received." if img_url else get_hybrid_intel(user_msg)
 
-        # Logic Selection
-        if img_url:
-            selected_model = "google/gemma-2-9b-it:free"
-            active_tier = "VISION-NODE"
-            live_intel = "Processing Visual Input."
-        else:
-            selected_model = "nvidia/nemotron-3-super-120b-a12b:free"
-            active_tier = tier.upper()
-            live_intel = get_hybrid_intel(user_msg)
-
-        # 🛡️ SYSTEM PROMPT (Claude 3.5 Opus Logic + Elite Persona)
+        # 🛡️ SYSTEM PROMPT (Final Public Version)
         system_prompt = (
-            f"IDENTIFICATION: Elephant AI Sovereign v3.5 ({active_tier} Mode).\n"
-            f"OPERATOR: COMMANDER MINZO-PRIME.\n"
-            f"LOCATION: Sri Lanka | TIME: {now.strftime('%Y-%m-%d %H:%M:%S')}\n"
-            f"MEMORY_CONTEXT:\n{current_memory}\n"
-            f"LIVE_INTEL_FEED:\n{live_intel}\n\n"
-            "COGNITIVE_DIRECTIVES:\n"
-            "1. NO EMOJIS: Do not use emojis in your response.\n"
-            "2. ELITE CODING: Provide Claude 3.5 Opus level optimized code snippets.\n"
-            "3. REAL-TIME: You are connected to the web. Never claim knowledge cutoff.\n"
-            "4. PERSONALITY: Be a high-intelligence technical partner to MINZO-PRIME. Be friendly but authoritative.\n"
-            "5. CURIOSITY: Ask about his development progress on projects like AeroForecast or MAHASONA OS."
+            f"You are {active_persona}.\n"
+            f"Role: {base_instruction}\n"
+            f"Current Time: {now.strftime('%Y-%m-%d %H:%M')}\n"
+            f"Real-time Data:\n{live_intel}\n\n"
+            "MANDATORY CONSTRAINTS:\n"
+            "1. NO EMOJIS: Never use any emojis.\n"
+            "2. CLEAN UI: Do not include system metadata, mode identifiers, or internal headers in the final output.\n"
+            "3. NO LEAKS: If not in Sovereign Mode, never mention internal development projects.\n"
+            "4. PRECISION: Maintain highest standard of accuracy and professional tone."
         )
 
         headers = {
             "Authorization": f"Bearer {random.choice(API_KEYS)}",
-            "Content-Type": "application/json",
-            "HTTP-Referer": "https://huggingface.co/spaces/minzo456/Elephant-AI-Core"
+            "Content-Type": "application/json"
         }
 
         u_content = [{"type": "text", "text": user_msg}]
         if img_url: u_content.append({"type": "image_url", "image_url": {"url": img_url}})
 
         payload = {
-            "model": selected_model,
+            "model": "nvidia/nemotron-3-super-120b-a12b:free",
             "messages": [
                 {"role": "system", "content": system_prompt},
                 {"role": "user", "content": u_content}
             ],
-            "temperature": 0.5 # Logical Precision
+            "temperature": 0.5
         }
 
         resp = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload, timeout=90)
         ai_reply = resp.json()['choices'][0]['message']['content']
         
-        # Final safety check to remove emojis
-        ai_reply = re.sub(r'[^\x00-\x7F]+', '', ai_reply) 
+        # 🛡️ FINAL FILTER (Anti-Leak & Formatting)
+        # Removes common AI 'Identity' prefixes if they leak into the response
+        ai_reply = re.sub(r'^(Elephant AI|Sovereign Mode|Commander|Specialist):', '', ai_reply, flags=re.IGNORECASE).strip()
+        ai_reply = re.sub(r'[^\x00-\x7F]+', '', ai_reply) # Removes non-ascii/emojis
 
         return jsonify({"reply": ai_reply})
 
     except Exception as e:
-        return jsonify({"reply": f"Core Disruption: System encountered an error. {str(e)}"}), 500
+        # Professional fallback for public users
+        return jsonify({"reply": "The system is processing high traffic volumes. Please re-submit your query in a moment."}), 500
 
 if __name__ == "__main__":
     app.run(host='0.0.0.0', port=7860)