Update app.py

app.py

@@ -1,24 +1,17 @@
 import os
 import random
 import requests
-import base64
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 from datetime import datetime
-import pytz # Timezone සඳහා
-
-# 🔱 Stable Web Search Integration
-try:
-    from duckduckgo_search import DDGS
-except ImportError:
-    os.system('pip install duckduckgo_search')
-    from duckduckgo_search import DDGS
+import pytz
 
 app = Flask(__name__)
 CORS(app)
 
-# AUTHORIZED CONFIGURATION
+# 🔱 පද්ධති වින්‍යාසය (Security & Memory)
 IMGBB_KEY = "c32f54279bcd7d4f766b5eac37fd7327"
+MEMORY_FILE = "elephant_core_brain.txt"
 API_KEYS = [
     "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",
     "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",
@@ -27,26 +20,31 @@ API_KEYS = [
     "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"
 ]
 
-def search_intel(query):
-    """වැඩිදියුණු කළ සෙවුම් පද්ධතිය"""
-    intel_data = ""
+# 🧠 දිගුකාලීන මතකය (Persistent Intelligence Sync)
+def sync_intel_to_memory(data):
+    with open(MEMORY_FILE, "a", encoding="utf-8") as f:
+        f.write(f"|SYNC_DATETIME: {datetime.now().strftime('%Y-%m-%d %H:%M')}| {data}\n")
+
+def load_core_memory():
+    if os.path.exists(MEMORY_FILE):
+        with open(MEMORY_FILE, "r", encoding="utf-8") as f:
+            lines = f.readlines()
+            # පද්ධතියේ ක්‍රියාකාරීත්වය වේගවත් කිරීමට අවසන් කරුණු 20 මතකයට ගනී
+            return "".join(lines[-20:])
+    return "Initial memory banks empty. Awaiting Specialist input."
+
+# 🌍 වෙබ් සෙවුම් පද්ධතිය (Enhanced Web Search)
+def get_web_intel(query):
     try:
+        from duckduckgo_search import DDGS
         with DDGS() as ddgs:
-            # Region එක wt-wt (Global) යොදා block වීම් අවම කරයි
-            results = list(ddgs.text(query, region='wt-wt', max_results=5))
-            if results:
-                for r in results:
-                    intel_data += f"\n- {r['title']}: {r['body']}"
-            else: intel_data = "No live updates found for this query."
-    except Exception as e:
-        intel_data = f"Web Retrieval currently throttled. Proceed with internal knowledge."
-    return intel_data
+            results = list(ddgs.text(query, region='wt-wt', max_results=4))
+            return "\n".join([f"• {r['title']}: {r['body']}" for r in results])
+    except: return "Web Retrieval Mode: Offline (Internal Knowledge Only)"
 
-def upload_to_imgbb(b64_image):
-    """Internal Image Hosting Logic"""
+def upload_media(b64):
     try:
-        payload = {"key": IMGBB_KEY, "image": b64_image}
-        res = requests.post("https://api.imgbb.com/1/upload", data=payload)
+        res = requests.post("https://api.imgbb.com/1/upload", data={"key": IMGBB_KEY, "image": b64})
         return res.json()['data']['url']
     except: return None
 
@@ -55,76 +53,80 @@ def chat():
     try:
         data = request.json
         user_msg = data.get("message", "")
-        image_base_64 = data.get("image_base64") 
+        img_b64 = data.get("image_base64")
         tier = data.get("tier", "prime")
 
-        # 🔱 Time Configuration (Sri Lanka Standard Time)
-        lk_tz = pytz.timezone('Asia/Colombo')
-        now = datetime.now(lk_tz)
-        current_date = now.strftime("%A, %B %d, %Y")
-        current_time = now.strftime("%I:%M %p")
-
-        # Image Handling
-        final_image_url = upload_to_imgbb(image_base_64) if image_base_64 else None
-
-        # Model Logic
-        if final_image_url:
-            selected_model = "nvidia/llama-nemotron-embed-vl-1b-v2:free"
-            active_tier = "VISION-CORE"
-            web_context = "Visual mode active. Bypassing search for visual precision."
+        # 🕒 Timezone Management (Sri Lanka Standard)
+        sl_tz = pytz.timezone('Asia/Colombo')
+        now = datetime.now(sl_tz)
+        
+        # 🧠 Self-Learning logic (Command: Remember/මතක තබාගන්න)
+        if any(x in user_msg.lower() for x in ["remember", "මතක තබාගන්න", "mthka thba ganna"]):
+            sync_intel_to_memory(user_msg)
+            mem_status = "SYNCHRONIZED"
+        else: mem_status = "READ_ONLY"
+
+        current_memory = load_core_memory()
+        img_url = upload_media(img_b64) if img_b64 else None
+
+        # 🔱 Sovereign Model Selection (Dynamic Multi-Engine)
+        if img_url:
+            # Vision සඳහා දැනට තිබෙන හොඳම Free Model එක
+            selected_model = "google/gemini-2.0-flash-exp:free"
+            active_tier = "VISION-NODE"
+            web_intel_data = "Image analysis protocol engaged."
         else:
-            model_map = {"prime": "minimax/minimax-m2.5:free", "nexus": "tencent/hy3-preview:free"}
-            selected_model = model_map.get(tier, "minimax/minimax-m2.5:free")
+            # 🔱 ඔබ සොයාගත් අලුත්ම සහ ප්‍රබලම මොඩල් දෙක මෙතැනට!
+            model_map = {
+                "prime": "nvidia/nemotron-3-super-120b-a12b:free", # බලවත් 120B Model එක
+                "nexus": "z-ai/glm-4.5-air:free"                   # වේගවත් සහ Thinking AI එක
+            }
+            selected_model = model_map.get(tier, "nvidia/nemotron-3-super-120b-a12b:free")
             active_tier = tier.upper()
-            web_context = search_intel(user_msg)
+            web_intel_data = get_web_intel(user_msg)
+
+        # 🛡️ පද්ධති ප්‍රොටොකෝලය (Master Prompt)
+        system_prompt = (
+            f"SYSTEM_IDENTITY: {active_tier} mode of Elephant AI.\n"
+            f"OPERATOR: MINZO-PRIME / Specialist Team.\n"
+            f"CHRONOS: {now.strftime('%A, %B %d, %Y | %I:%M %p')}\n"
+            f"CORE_MEMORY_BANKS:\n{current_memory}\n"
+            f"REAL_TIME_WEB_INTEL:\n{web_intel_data}\n"
+            "INSTRUCTION: Use both Core Memory and Web Intel for maximum precision. "
+            "Maintain an authoritative, sovereign, and highly intelligent persona."
+        )
 
         headers = {
             "Authorization": f"Bearer {random.choice(API_KEYS)}",
             "Content-Type": "application/json",
-            "HTTP-Referer": "https://huggingface.co/spaces/minzo456/Elephant-AI-Core",
-            "X-Title": "Elephant AI Sovereign Node"
+            "HTTP-Referer": "https://huggingface.co/spaces/minzo456/Elephant-AI-Core"
         }
 
-        # 🔱 ශක්තිමත් කළ System Prompt එක (AI එකට බොරු කීමට ඉඩ නොදේ)
-        system_content = (
-            f"You are {active_tier} mode of Elephant AI, engineered by MINZO-PRIME.\n"
-            f"STRICT SYSTEM DATA:\n"
-            f"- Current Date: {current_date}\n"
-            f"- Current Time: {current_time}\n"
-            f"- Location: Sri Lanka\n"
-            f"LIVE WEB CONTEXT: {web_context}\n"
-            "INSTRUCTION: Always use the 'STRICT SYSTEM DATA' for time/date queries. "
-            "Never claim you don't know the date. Maintain your sovereign identity."
-        )
-
-        content = [{"type": "text", "text": user_msg}]
-        if final_image_url:
-            content.append({"type": "image_url", "image_url": {"url": final_image_url}})
+        # Content Assembly
+        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,
             "messages": [
-                {"role": "system", "content": system_content},
-                {"role": "user", "content": content}
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": u_content}
             ],
-            "temperature": 0.5
+            "temperature": 0.4 # නිරවද්‍යතාවය (Accuracy) වැඩි කිරීමට
         }
 
-        resp = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload, timeout=60)
-        resp_json = resp.json()
+        resp = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload, timeout=70)
+        resp_data = resp.json()
 
-        if resp.status_code == 200 and 'choices' in resp_json:
-            return jsonify({"reply": resp_json['choices'][0]['message']['content']})
+        if 'choices' in resp_data:
+            ai_reply = resp_data['choices'][0]['message']['content']
+            status_tag = "🔱 [Intel Synchronized to Memory Bank]\n\n" if mem_status == "SYNCHRONIZED" else ""
+            return jsonify({"reply": status_tag + ai_reply})
         else:
-            error_info = resp_json.get('error', {}).get('message', 'Core Node Overload')
-            return jsonify({"reply": f"Core Disruption: {error_info}"}), resp.status_code
+            return jsonify({"reply": f"Node Error: {resp_data.get('error', {}).get('message', 'Core Overload')}"}), 400
 
     except Exception as e:
-        return jsonify({"reply": f"Sovereign Core Exception: {str(e)}"}), 500
-
-@app.route('/')
-def health():
-    return "🐘 Elephant AI Core: ONLINE | Operator: MINZO-PRIME | Node: Stable"
+        return jsonify({"reply": f"Sovereign Core Disruption: {str(e)}"}), 500
 
 if __name__ == "__main__":
     app.run(host='0.0.0.0', port=7860)