Update app.py

app.py

@@ -14,56 +14,44 @@ app = Flask(__name__)
 CORS(app)
 
 # 🛠️ 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 (Round-robin selection for load balancing)
 API_KEYS = [
     "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",
     "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",
     "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"
 ]
 
-# 🧠 HYBRID INTELLIGENCE NODE (Wiki + DDG + Google)
+# 🧠 HYBRID INTELLIGENCE NODE
 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)}"
         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
-
-    # 2. DuckDuckGo (Free & Unlimited)
     try:
         from duckduckgo_search import DDGS
         with DDGS() as ddgs:
             results = list(ddgs.text(query[:60], max_results=2))
-            for r in results:
-                intel_pool.append(f"Web Context: {r['body']}")
+            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']}")
+            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})
+        res = requests.post("https://api.imgbb.com/1/upload", data={"key": IMGBB_KEY, "image": b64}, timeout=15)
         return res.json()['data']['url']
     except: return None
 
@@ -82,23 +70,30 @@ def chat():
         sl_tz = pytz.timezone('Asia/Colombo')
         now = datetime.now(sl_tz)
 
-        # 🛡️ IDENTITY SHIELDING LOGIC
-        # Detects if the user is the Commander or a Specialist
+        # 🛡️ IDENTITY & MODEL ROUTING
         authorized_keywords = ["MINZO-PRIME", "Specialist", "commander"]
         is_authorized = any(kw.lower() in user_msg.lower() for kw in authorized_keywords)
+        is_nexus = "nexus" in user_msg.lower() # "Nexus" කියන වචනය තිබේ නම් Nexus model එකට යයි
         
         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."
+            base_instruction = "Greetings Specialist. Provide elite reasoning. Internal projects (AeroForecast/MAHASONA) are accessible."
+            # Uncensored Power + Thinking
+            selected_model = "cognitivecomputations/dolphin-mistral-24b-venice-edition:free"
+        elif is_nexus:
+            active_persona = "Elephant AI Nexus"
+            base_instruction = "You are in Nexus Thinking Mode. Use advanced logic to break down complex queries."
+            selected_model = "liquid/lfm-2.5-1.2b-thinking:free" # Nexus Model
         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."
+            base_instruction = "You are Elephant AI Prime. Provide clear, accurate, and professional information."
+            selected_model = "minimax/minimax-m2.5:free" # Prime Model
 
         # 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)
 
-        # 🛡️ SYSTEM PROMPT (Final Public Version)
+        # 🛡️ SYSTEM PROMPT
         system_prompt = (
             f"You are {active_persona}.\n"
             f"Role: {base_instruction}\n"
@@ -106,9 +101,9 @@ def chat():
             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."
+            "2. CLEAN UI: No system metadata or internal headers in output.\n"
+            "3. NO LEAKS: If not in Sovereign Mode, never mention internal projects.\n"
+            "4. PRECISION: Maintain professional tone."
         )
 
         headers = {
@@ -120,7 +115,7 @@ def chat():
         if img_url: u_content.append({"type": "image_url", "image_url": {"url": img_url}})
 
         payload = {
-            "model": "nvidia/nemotron-3-super-120b-a12b:free",
+            "model": selected_model,
             "messages": [
                 {"role": "system", "content": system_prompt},
                 {"role": "user", "content": u_content}
@@ -131,16 +126,14 @@ def chat():
         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 FILTER (Anti-Leak & Formatting)
-        # Removes common AI 'Identity' prefixes if they leak into the response
+        # 🛡️ FINAL FILTER
         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
+        ai_reply = re.sub(r'[^\x00-\x7F]+', '', ai_reply) # Removes non-ascii
 
         return jsonify({"reply": ai_reply})
 
     except Exception as e:
-        # Professional fallback for public users
-        return jsonify({"reply": "The system is processing high traffic volumes. Please re-submit your query in a moment."}), 500
+        return jsonify({"reply": "The system is processing high traffic volumes. Please re-submit in a moment."}), 500
 
 if __name__ == "__main__":
     app.run(host='0.0.0.0', port=7860)