Update app.py

app.py

@@ -1,17 +1,24 @@
 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
+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
 
 app = Flask(__name__)
 CORS(app)
 
-# 🔱 පද්ධති වින්‍යාසය (Configuration)
+# AUTHORIZED CONFIGURATION
 IMGBB_KEY = "c32f54279bcd7d4f766b5eac37fd7327"
-MEMORY_FILE = "elephant_brain.txt"
 API_KEYS = [
     "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",
     "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",
@@ -20,32 +27,26 @@ API_KEYS = [
     "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"
 ]
 
-# 🧠 මතකය කළමනාකරණය (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):
+def search_intel(query):
+    """වැඩිදියුණු කළ සෙවුම් පද්ධතිය"""
+    intel_data = ""
     try:
-        from duckduckgo_search import DDGS
         with DDGS() as ddgs:
-            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."
+            # 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
 
-def upload_img(b64):
+def upload_to_imgbb(b64_image):
+    """Internal Image Hosting Logic"""
     try:
-        res = requests.post("https://api.imgbb.com/1/upload", data={"key": IMGBB_KEY, "image": b64})
+        payload = {"key": IMGBB_KEY, "image": b64_image}
+        res = requests.post("https://api.imgbb.com/1/upload", data=payload)
         return res.json()['data']['url']
     except: return None
 
@@ -54,60 +55,57 @@ def chat():
     try:
         data = request.json
         user_msg = data.get("message", "")
-        b64_image = data.get("image_base64")
+        image_base_64 = data.get("image_base64") 
         tier = data.get("tier", "prime")
 
-        # 🕒 කාලය සහ දත්ත (Sri Lanka)
+        # 🔱 Time Configuration (Sri Lanka Standard Time)
         lk_tz = pytz.timezone('Asia/Colombo')
         now = datetime.now(lk_tz)
-        
-        # 🧠 ස්වයං-ඉගෙනුම් 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."
+        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."
         else:
-            # ඔබගේ Prime/Nexus Models
             model_map = {"prime": "minimax/minimax-m2.5:free", "nexus": "tencent/hy3-preview:free"}
-            selected_model = model_map.get(tier, "google/gemma-4-31b-it:free")
-            active_mode = tier.upper()
-            web_data = web_intel(user_msg)
+            selected_model = model_map.get(tier, "minimax/minimax-m2.5:free")
+            active_tier = tier.upper()
+            web_context = search_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"
+            "HTTP-Referer": "https://huggingface.co/spaces/minzo456/Elephant-AI-Core",
+            "X-Title": "Elephant AI Sovereign Node"
         }
 
-        # 🔱 පද්ධති ප්‍රොටොකෝලය (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."
+        # 🔱 ශක්තිමත් කළ 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 Assembly
-        user_content = [{"type": "text", "text": user_msg}]
-        if img_url:
-            user_content.append({"type": "image_url", "image_url": {"url": img_url}})
+        content = [{"type": "text", "text": user_msg}]
+        if final_image_url:
+            content.append({"type": "image_url", "image_url": {"url": final_image_url}})
 
         payload = {
             "model": selected_model,
             "messages": [
-                {"role": "system", "content": system_prompt},
-                {"role": "user", "content": user_content}
+                {"role": "system", "content": system_content},
+                {"role": "user", "content": content}
             ],
             "temperature": 0.5
         }
@@ -115,15 +113,18 @@ def chat():
         resp = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload, timeout=60)
         resp_json = resp.json()
 
-        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})
+        if resp.status_code == 200 and 'choices' in resp_json:
+            return jsonify({"reply": resp_json['choices'][0]['message']['content']})
         else:
-            return jsonify({"reply": f"Node Error: {resp_json.get('error', {}).get('message', 'Unknown')}"}), 400
+            error_info = resp_json.get('error', {}).get('message', 'Core Node Overload')
+            return jsonify({"reply": f"Core Disruption: {error_info}"}), resp.status_code
 
     except Exception as e:
-        return jsonify({"reply": f"Core Disruption: {str(e)}"}), 500
+        return jsonify({"reply": f"Sovereign Core Exception: {str(e)}"}), 500
+
+@app.route('/')
+def health():
+    return "🐘 Elephant AI Core: ONLINE | Operator: MINZO-PRIME | Node: Stable"
 
 if __name__ == "__main__":
     app.run(host='0.0.0.0', port=7860)