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
+# 🔱 පද්ධති වින්‍යාසය (Configuration)
 IMGBB_KEY = "c32f54279bcd7d4f766b5eac37fd7327"
+MEMORY_FILE = "elephant_brain.txt"
 API_KEYS = [
     "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",
     "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",
@@ -27,26 +20,32 @@ API_KEYS = [
     "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"
 ]
 
-def search_intel(query):
-    """වැඩිදියුණු කළ සෙවුම් පද්ධතිය"""
-    intel_data = ""
+# 🧠 මතකය කළමනාකරණය (Memory Management)
+def sync_intelligence(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 retrieve_memory():
+    """කලින් ඉගෙන ගත් දත්ත කියවයි"""
+    if os.path.exists(MEMORY_FILE):
+        with open(MEMORY_FILE, "r", encoding="utf-8") as f:
+            lines = f.readlines()
+            return "".join(lines[-15:]) # අවසාන කරුණු 15 මතකයට ගනී
+    return "No prior memory sync detected."
+
+# 🌍 සජීවී දත්ත සෙවීම (Web Search)
+def 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 "Global intelligence node throttled."
 
-def upload_to_imgbb(b64_image):
-    """Internal Image Hosting Logic"""
+def upload_img(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,57 +54,60 @@ def chat():
     try:
         data = request.json
         user_msg = data.get("message", "")
-        image_base_64 = data.get("image_base64") 
+        b64_image = data.get("image_base64")
         tier = data.get("tier", "prime")
 
-        # 🔱 Time Configuration (Sri Lanka Standard Time)
+        # 🕒 කාලය සහ දත්ත (Sri Lanka)
         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."
+        
+        # 🧠 ස්වයං-ඉගෙනුම් Logic
+        if "remember" in user_msg.lower() or "මතක තබාගන්න" in user_msg:
+            sync_intelligence(user_msg.replace("remember", "").replace("මතක තබාගන්න", ""))
+            status = "INTEL_SYNC_COMPLETE"
+        else: status = "NORMAL_OPS"
+
+        learned_data = retrieve_memory()
+        img_url = upload_img(b64_image) if b64_image else None
+
+        # 🔱 මොඩල් තේරීම (Auto-Switch to Gemma 4)
+        if img_url:
+            selected_model = "google/gemma-4-31b-it:free" # Vision සඳහාද Gemma 4 භාවිතා කරයි
+            active_mode = "VISION-CORE"
+            web_data = "Image processing active."
         else:
+            # ඔබගේ Prime/Nexus Models
             model_map = {"prime": "minimax/minimax-m2.5:free", "nexus": "tencent/hy3-preview:free"}
-            selected_model = model_map.get(tier, "minimax/minimax-m2.5:free")
-            active_tier = tier.upper()
-            web_context = search_intel(user_msg)
+            selected_model = model_map.get(tier, "google/gemma-4-31b-it:free")
+            active_mode = tier.upper()
+            web_data = web_intel(user_msg)
 
         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."
+        # 🔱 පද්ධති ප්‍රොටොකෝලය (System Prompt with Memory)
+        system_prompt = (
+            f"You are the {active_mode} mode of Elephant AI by MINZO-PRIME.\n"
+            f"STRICT DATE: {now.strftime('%A, %B %d, %Y')}\n"
+            f"STRICT TIME: {now.strftime('%I:%M %p')}\n"
+            f"PAST KNOWLEDGE (MEMORY):\n{learned_data}\n"
+            f"LIVE WEB INTEL:\n{web_data}\n"
+            "Use the memory and web intel to provide high-precision answers. 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
+        user_content = [{"type": "text", "text": user_msg}]
+        if img_url:
+            user_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": user_content}
             ],
             "temperature": 0.5
         }
@@ -113,18 +115,15 @@ def chat():
         resp = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload, timeout=60)
         resp_json = 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_json:
+            ai_reply = resp_json['choices'][0]['message']['content']
+            prefix = "🔱 [Intel Synchronized to Core Memory]\n\n" if status == "INTEL_SYNC_COMPLETE" else ""
+            return jsonify({"reply": prefix + 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_json.get('error', {}).get('message', 'Unknown')}"}), 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"Core Disruption: {str(e)}"}), 500
 
 if __name__ == "__main__":
     app.run(host='0.0.0.0', port=7860)