Update app.py

app.py

@@ -1,17 +1,22 @@
 import os
 import random
 import requests
+import json
+import re
+import urllib.parse
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 from datetime import datetime
 import pytz
+import feedparser
+from bs4 import BeautifulSoup
 
 app = Flask(__name__)
 CORS(app)
 
 # 🔱 පද්ධති වින්‍යාසය (Security & Memory)
 IMGBB_KEY = "c32f54279bcd7d4f766b5eac37fd7327"
-MEMORY_FILE = "elephant_core_brain.txt"
+MEMORY_FILE = "elephant_sovereign_brain.txt"
 API_KEYS = [
     "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",
     "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",
@@ -20,46 +25,71 @@ API_KEYS = [
     "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"
 ]
 
-# 🧠 දිගුකාලීන මතකය (Intelligence Sync)
-def sync_intel_to_memory(data):
+# 📡 RSS FEEDS (WormGPT Logic)
+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']
+}
+
+# 🧠 මතකය කළමනාකරණය
+def sync_intel(data):
     with open(MEMORY_FILE, "a", encoding="utf-8") as f:
-        f.write(f"|SYNC: {datetime.now().strftime('%Y-%m-%d %H:%M')}| {data}\n")
+        f.write(f"|{datetime.now().strftime('%Y-%m-%d %H:%M')}| {data}\n")
 
-def load_core_memory():
+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."
+
+# 📡 HYBRID WEB SEARCH (RSS + 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):
+    """RSS feeds පරීක්ෂා කර පසුව Web Search වෙත යයි"""
+    category = detect_intel_category(query)
+    intel_output = []
+
+    # 1. RSS Retrieval (වඩාත් ස්ථාවරයි)
+    for feed_url in RSS_SOURCES.get(category, RSS_SOURCES['world']):
         try:
-            with open(MEMORY_FILE, "r", encoding="utf-8") as f:
-                lines = f.readlines()
-                return "".join(lines[-20:])
-        except: return "Memory access error."
-    return "Initial memory banks empty."
-
-# 🌍 සජීවී සෙවුම් පද්ධතිය (Improved & Stable)
-def get_web_intel(query):
-    """Hugging Face IP blocks මඟහරවා ගැනීමට 'lite' backend එක භාවිතා කරයි"""
-    try:
-        from duckduckgo_search import DDGS
-        # සෙවුම් පදය සරල කර (Trim) භාවිතා කරයි
-        clean_query = query[:60]
-        with DDGS() as ddgs:
-            # backend='lite' යනු shared IPs සඳහා වඩාත් සාර්ථක ක්‍රමයයි
-            results = list(ddgs.text(clean_query, region='wt-wt', max_results=3, backend='lite'))
-            if results:
-                return "\n".join([f"• {r['title']}: {r['body']}" for r in results])
-    except Exception as e:
-        print(f"Web Search Log: {e}")
-    return "Web Retrieval: Throttled or Offline. Proceeding with Internal Core Data."
+            feed = feedparser.parse(feed_url)
+            for entry in feed.entries[:3]:
+                snippet = re.sub('<[^>]+>', '', entry.get('summary', ''))[:200]
+                intel_output.append(f"📡 RSS: {entry.title}\n📝 {snippet}\n🔗 {entry.link}")
+        except: continue
+
+    # 2. DuckDuckGo Fallback (RSS වල දත්ත මදි නම්)
+    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
+
+    return "\n\n".join(intel_output) if intel_output else "No live intelligence retrieved."
 
-def upload_media(b64):
+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
 
-# 🔱 Hugging Face Health Check
+# 🔱 ROUTES
 @app.route('/')
-def health_check():
-    return f"🐘 Elephant AI Core: ONLINE | Operator: MINZO-PRIME | {datetime.now().strftime('%H:%M:%S')}"
+def health():
+    return f"🐘 Elephant AI Core: ONLINE | {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
 
 @app.route('/api/chat', methods=['POST'])
 def chat():
@@ -69,24 +99,23 @@ def chat():
         img_b64 = data.get("image_base64")
         tier = data.get("tier", "prime")
 
-        # Time Configuration (Sri Lanka)
         sl_tz = pytz.timezone('Asia/Colombo')
         now = datetime.now(sl_tz)
         
-        # Self-Learning Detection
-        if any(x in user_msg.lower() for x in ["remember", "මතක තබාගන්න", "mthka thba"]):
-            sync_intel_to_memory(user_msg)
-            mem_status = "SYNCHRONIZED"
-        else: mem_status = "READ_ONLY"
+        # Self-Learning
+        if "remember" in user_msg.lower() or "මතක තබාගන්න" in user_msg:
+            sync_intel(user_msg)
+            mem_tag = " [SYNCED]"
+        else: mem_tag = ""
 
-        current_memory = load_core_memory()
-        img_url = upload_media(img_b64) if img_b64 else None
+        current_memory = load_memory()
+        img_url = upload_to_imgbb(img_b64) if img_b64 else None
 
-        # 🔱 Model Management
+        # 🔱 Sovereign Engines
         if img_url:
             selected_model = "google/gemma-4-31b-it:free"
-            active_tier = "VISION-CORE"
-            web_intel_data = "Visual input engaged. Bypassing search."
+            active_tier = "VISION-NODE"
+            live_intel = "Image data provided."
         else:
             model_map = {
                 "prime": "nvidia/nemotron-3-super-120b-a12b:free",
@@ -94,17 +123,16 @@ def chat():
             }
             selected_model = model_map.get(tier, "nvidia/nemotron-3-super-120b-a12b:free")
             active_tier = tier.upper()
-            # සජීවී සෙවුම සිදු කිරීම
-            web_intel_data = get_web_intel(user_msg)
+            live_intel = get_hybrid_intel(user_msg)
 
-        # 🛡️ පද්ධති ප්‍රොටොකෝලය
+        # 🛡️ SYSTEM PROMPT
         system_prompt = (
-            f"SYSTEM_IDENTITY: {active_tier} mode of Elephant AI by MINZO-PRIME.\n"
-            f"CHRONOS: {now.strftime('%A, %B %d, %Y | %I:%M %p')}\n"
-            f"STRICT_LOCATION: Sri Lanka.\n"
-            f"CORE_MEMORY_BANKS:\n{current_memory}\n"
-            f"LIVE_WEB_INTEL:\n{web_intel_data}\n"
-            "INSTRUCTION: Use both memory and web intel to provide authoritative answers."
+            f"IDENTITY: {active_tier} Mode of Elephant AI.\n"
+            f"OPERATOR: COMMANDER MINZO-PRIME.\n"
+            f"CHRONOS: {now.strftime('%Y-%m-%d %H:%M')}\n"
+            f"ESTABLISHED_MEMORY:\n{current_memory}\n"
+            f"LIVE_INTELLIGENCE:\n{live_intel}\n"
+            "INSTRUCTION: Use RSS and Web data to provide precise, copy-pasteable technical data."
         )
 
         headers = {
@@ -123,21 +151,20 @@ def chat():
                 {"role": "system", "content": system_prompt},
                 {"role": "user", "content": u_content}
             ],
-            "temperature": 0.4
+            "temperature": 0.7
         }
 
-        resp = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload, timeout=75)
+        resp = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload, timeout=80)
         resp_data = resp.json()
 
         if 'choices' in resp_data:
             ai_reply = resp_data['choices'][0]['message']['content']
-            status_tag = "🔱 [Intel Synchronized to Memory]\n\n" if mem_status == "SYNCHRONIZED" else ""
-            return jsonify({"reply": status_tag + ai_reply})
+            return jsonify({"reply": f"🔱{mem_tag}\n\n" + ai_reply})
         else:
-            return jsonify({"reply": f"Core Exception: {resp_data.get('error', {}).get('message', 'Unstable Connection')}"}), 400
+            return jsonify({"reply": f"Node Error: {resp_data.get('error', {}).get('message', 'Timeout')}"}), 400
 
     except Exception as e:
-        return jsonify({"reply": f"Sovereign Core Disruption: {str(e)}"}), 500
+        return jsonify({"reply": f"Core Disruption: {str(e)}"}), 500
 
 if __name__ == "__main__":
     app.run(host='0.0.0.0', port=7860)