Update app.py

app.py

@@ -5,8 +5,9 @@ import base64
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 from datetime import datetime
+import pytz # Timezone සඳහා
 
-# 🔱 Stable DDGS Integration
+# 🔱 Stable Web Search Integration
 try:
     from duckduckgo_search import DDGS
 except ImportError:
@@ -27,21 +28,24 @@ API_KEYS = [
 ]
 
 def search_intel(query):
-    """RAG-style web retrieval"""
+    """වැඩිදියුණු කළ සෙවුම් පද්ධතිය"""
     intel_data = ""
     try:
         with DDGS() as ddgs:
-            results = list(ddgs.text(query, region='wt-wt', max_results=4))
+            # 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']}"
-    except: intel_data = "Web access limited."
+            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_to_imgbb(base64_image):
-    """Backend-side Image Upload to ImgBB"""
+def upload_to_imgbb(b64_image):
+    """Internal Image Hosting Logic"""
     try:
-        payload = {"key": IMGBB_KEY, "image": base64_image}
+        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
@@ -51,20 +55,23 @@ def chat():
     try:
         data = request.json
         user_msg = data.get("message", "")
-        # Frontend එකෙන් 'image_base64' නමින් data එක එවිය යුතුයි
         image_base_64 = data.get("image_base64") 
         tier = data.get("tier", "prime")
 
-        final_image_url = None
-        if image_base_64:
-            # පින්තූරයක් තිබේ නම් එය Backend එකෙන් Upload කරයි
-            final_image_url = upload_to_imgbb(image_base_64)
+        # 🔱 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 Selection Logic
+        # Model Logic
         if final_image_url:
             selected_model = "nvidia/llama-nemotron-embed-vl-1b-v2:free"
             active_tier = "VISION-CORE"
-            web_context = "Image analysis mode active."
+            web_context = "Visual mode active. Bypassing search for visual precision."
         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")
@@ -74,10 +81,22 @@ def chat():
         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"
         }
 
-        # Payload Assembly
+        # 🔱 ශක්තිමත් කළ 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}})
@@ -85,20 +104,27 @@ def chat():
         payload = {
             "model": selected_model,
             "messages": [
-                {"role": "system", "content": f"You are {active_tier} mode of Elephant AI by MINZO-PRIME. Web Data: {web_context}"},
+                {"role": "system", "content": system_content},
                 {"role": "user", "content": content}
-            ]
+            ],
+            "temperature": 0.5
         }
 
         resp = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload, timeout=60)
-        return jsonify({"reply": resp.json()['choices'][0]['message']['content']})
+        resp_json = resp.json()
+
+        if resp.status_code == 200 and 'choices' in resp_json:
+            return jsonify({"reply": resp_json['choices'][0]['message']['content']})
+        else:
+            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"
+    return "🐘 Elephant AI Core: ONLINE | Operator: MINZO-PRIME | Node: Stable"
 
 if __name__ == "__main__":
     app.run(host='0.0.0.0', port=7860)