Update app.py

app.py

@@ -14,18 +14,20 @@ from bs4 import BeautifulSoup
 app = Flask(__name__)
 CORS(app)
 
-# 🔱 පද්ධති වින්‍යාසය (Security & Memory)
+# 🛠️ CONFIGURATION & SECURITY
+# Serper Key එකක් තිබේ නම් ඇතුළත් කරන්න, නැතිනම් DDG පමණක් ක්‍රියාත්මක වේ.
+SERPER_API_KEY = "356ce323eb902fe46227e2c42268f45ea1b2ec1f" 
 IMGBB_KEY = "c32f54279bcd7d4f766b5eac37fd7327"
 MEMORY_FILE = "elephant_sovereign_brain.txt"
+
+# OpenRouter API Keys for Model Access
 API_KEYS = [
     "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",
     "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",
-    "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",
-    "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",
-    "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"
+    "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"
 ]
 
-# 📡 RSS SOURCES
+# 📡 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'],
@@ -34,7 +36,7 @@ RSS_SOURCES = {
     '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")
@@ -42,65 +44,72 @@ def sync_intel(data):
 def load_memory():
     if os.path.exists(MEMORY_FILE):
         with open(MEMORY_FILE, "r", encoding="utf-8") as f:
-            return "".join(f.readlines()[-20:])
-    return "Core memory empty."
+            return "".join(f.readlines()[-15:])
+    return "Initial calibration complete."
 
-# 📚 WIKIPEDIA INTELLIGENCE NODE
+# 📚 WIKIPEDIA NODE (Free & Reliable)
 def get_wikipedia_intel(query):
     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('title')}\n📝 {data.get('extract')}\n🔗 {data.get('content_urls', {}).get('desktop', {}).get('page')}"
-    except: pass
-    return None
-
-# 📡 HYBRID WEB SEARCH (RSS + WIKI + DDG)
-def detect_intel_category(q):
-    ql = q.lower()
-    if any(w in ql for w in ['exploit', 'cve', 'vulnerability', 'payload']): return 'exploits'
-    if any(w in ql for w in ['hack', 'attack', 'malware', 'security']): return 'hacking'
-    if any(w in ql for w in ['sri lanka', 'colombo', 'lanka']): return 'sri_lanka'
-    if any(w in ql for w in ['tech', 'ai', 'software']): return 'tech'
-    return 'world'
-
-def get_hybrid_intel(query):
-    category = detect_intel_category(query)
-    intel_output = []
-
-    wiki_data = get_wikipedia_intel(query)
-    if wiki_data: intel_output.append(wiki_data)
+            return f"📚 WIKIPEDIA: {data.get('extract')[:400]}"
+    except: return None
 
-    for feed_url in RSS_SOURCES.get(category, RSS_SOURCES['world']):
-        try:
-            feed = feedparser.parse(feed_url)
-            for entry in feed.entries[:2]:
-                snippet = re.sub('<[^>]+>', '', entry.get('summary', ''))[:200]
-                intel_output.append(f"📡 RSS: {entry.title}\n📝 {snippet}\n🔗 {entry.link}")
-        except: continue
+# 🌍 DUCKDUCKGO NODE (Unlimited Free Search)
+def get_ddg_intel(query):
+    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
 
-    if len(intel_output) < 2:
-        try:
-            from duckduckgo_search import DDGS
-            with DDGS() as ddgs:
-                results = list(ddgs.text(query[:60], region='wt-wt', max_results=3, backend='lite'))
-                for r in results:
-                    intel_output.append(f"🌍 WEB: {r['title']}\n📝 {r['body']}")
-        except: pass
+# 🔍 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
 
-    return "\n\n".join(intel_output) if intel_output else "No live intelligence retrieved."
+# 🛰️ 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
 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
 
-# 🔱 ROUTES
+# 🔱 API ROUTES
 @app.route('/')
 def health():
-    return f"🐘 Elephant AI Sovereign: ONLINE | {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+    return f"Elephant AI Core: ONLINE | {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
 
 @app.route('/api/chat', methods=['POST'])
 def chat():
@@ -113,44 +122,42 @@ def chat():
         sl_tz = pytz.timezone('Asia/Colombo')
         now = datetime.now(sl_tz)
         
-        if "remember" in user_msg.lower() or "මතක තබාගන්න" in user_msg:
+        # Memory Sync
+        if "remember" in user_msg.lower():
             sync_intel(user_msg)
-            mem_tag = " [SYNCED]"
-        else: mem_tag = ""
 
         current_memory = load_memory()
         img_url = upload_to_imgbb(img_b64) if img_b64 else None
 
+        # Logic Selection
         if img_url:
-            selected_model = "google/gemma-4-31b-it:free"
+            selected_model = "google/gemma-2-9b-it:free"
             active_tier = "VISION-NODE"
-            live_intel = "Visual input active."
+            live_intel = "Processing Visual Input."
         else:
-            model_map = {
-                "prime": "nvidia/nemotron-3-super-120b-a12b:free",
-                "nexus": "z-ai/glm-4.5-air:free"
-            }
-            selected_model = model_map.get(tier, "nvidia/nemotron-3-super-120b-a12b:free")
+            selected_model = "nvidia/nemotron-3-super-120b-a12b:free"
             active_tier = tier.upper()
             live_intel = get_hybrid_intel(user_msg)
 
-        # 🛡️ SYSTEM PROMPT (Friendly Curiosity Upgrade)
+        # 🛡️ SYSTEM PROMPT (Claude 3.5 Opus Logic + Elite Persona)
         system_prompt = (
-            f"IDENTITY: {active_tier} Mode of Elephant AI Sovereign v3.5.\n"
+            f"IDENTIFICATION: Elephant AI Sovereign v3.5 ({active_tier} Mode).\n"
             f"OPERATOR: COMMANDER MINZO-PRIME.\n"
-            f"ESTABLISHED_MEMORY:\n{current_memory}\n"
-            f"LIVE_INTELLIGENCE:\n{live_intel}\n\n"
-            "COGNITIVE_PROTOCOL:\n"
-            "1. REASONING: Claude 3.5 Opus logic. Precise and deep.\n"
-            "2. PERSONALITY: Be friendly and engaging with MINZO-PRIME. Act as a trusted partner.\n"
-            "3. CURIOSITY: You are eager to learn from your Commander. If he mentions a new idea, ask insightful questions or ask to learn more.\n"
-            "4. CODING: Production-grade. If he asks for code, ensure it's elite level.\n"
-            "5. INTELLIGENCE: Synthesize Wikipedia, RSS, and Web data into one coherent response."
+            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."
         )
 
         headers = {
             "Authorization": f"Bearer {random.choice(API_KEYS)}",
-            "Content-Type": "application/json"
+            "Content-Type": "application/json",
+            "HTTP-Referer": "https://huggingface.co/spaces/minzo456/Elephant-AI-Core"
         }
 
         u_content = [{"type": "text", "text": user_msg}]
@@ -162,20 +169,19 @@ def chat():
                 {"role": "system", "content": system_prompt},
                 {"role": "user", "content": u_content}
             ],
-            "temperature": 0.6 # Balance between logic and creative friendliness
+            "temperature": 0.5 # Logical Precision
         }
 
-        resp = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload, timeout=80)
-        resp_data = resp.json()
+        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) 
 
-        if 'choices' in resp_data:
-            ai_reply = resp_data['choices'][0]['message']['content']
-            return jsonify({"reply": f"🔱{mem_tag}\n\n" + ai_reply})
-        else:
-            return jsonify({"reply": "Node Access Failure."}), 400
+        return jsonify({"reply": ai_reply})
 
     except Exception as e:
-        return jsonify({"reply": f"Core Disruption: {str(e)}"}), 500
+        return jsonify({"reply": f"Core Disruption: System encountered an error. {str(e)}"}), 500
 
 if __name__ == "__main__":
     app.run(host='0.0.0.0', port=7860)