Update app.py

app.py

@@ -1,11 +1,12 @@
 import os, random, requests, json, feedparser
 from flask import Flask, request, jsonify
 from flask_cors import CORS
-from datetime import datetime # 🔱 දිනය ලබා ගැනීමට
+from datetime import datetime
 
 app = Flask(__name__)
 CORS(app)
 
+# 🔱 AUTHORIZED API KEYS
 API_KEYS = [
     "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",
     "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",
@@ -15,13 +16,15 @@ API_KEYS = [
 ]
 
 def fetch_web_intel(query):
+    """RSS Intelligence හරහා අන්තර්ජාලය පරීක්ෂා කිරීම"""
     intel = ""
     try:
         search_url = f'https://news.google.com/rss/search?q={query}&hl=en-US'
         feed = feedparser.parse(search_url)
         for entry in feed.entries[:3]:
-            intel += f"\n- {entry.title}"
-    except: pass
+            intel += f"\n- Title: {entry.title}"
+    except:
+        pass
     return intel
 
 @app.route('/api/chat', methods=['POST'])
@@ -29,27 +32,34 @@ def chat():
     try:
         data = request.json
         user_msg = data.get("message", "")
-        
-        # 1. 🔱 සැබෑ දිනය සහ වේලාව ලබා ගැනීම
+        tier = data.get("tier", "prime") # Prime or Nexus
+
+        # 🔱 Sovereign Model Mapping
+        model_map = {
+            "prime": "minimax/minimax-m2.5:free",
+            "nexus": "thm/glm-4.5-air:free"
+        }
+        selected_model = model_map.get(tier, "minimax/minimax-m2.5:free")
+
+        # Get Real-time Context
         current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        
         web_context = fetch_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 Core"
         }
-        
-        # 2. 🔱 AI එකට වර්තමාන දිනය සහ තොරතුරු ලබා දීම
-        system_prompt = f"""You are Elephant AI by MINZO-PRIME. 
-        Current Date and Time: {current_time}. 
-        Real-time Web Context: {web_context}
-        If the user asks about the date or time, use the information provided above."""
+
+        # System Instruction for Sovereign Identity
+        system_prompt = f"""You are {tier.upper()} mode of Elephant AI, a sovereign intelligence developed by MINZO-PRIME.
+        Current Date and Time: {current_time}.
+        Web Intelligence Context: {web_context}
+        Always maintain your identity as Elephant AI and provide professional, accurate answers."""
 
         payload = {
-            "model": "minimax/minimax-m2.5:free",
+            "model": selected_model,
             "messages": [
                 {"role": "system", "content": system_prompt},
                 {"role": "user", "content": user_msg}
@@ -57,18 +67,21 @@ def chat():
             "stream": False
         }
 
-        resp = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload)
+        resp = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload, timeout=40)
         res_data = resp.json()
-        
-        ai_reply = res_data['choices'][0]['message']['content']
-        return jsonify({"reply": ai_reply})
+
+        if 'choices' in res_data:
+            ai_reply = res_data['choices'][0]['message']['content']
+            return jsonify({"reply": ai_reply})
+        else:
+            return jsonify({"reply": "Core Error: Connectivity to sovereign node lost."}), 500
 
     except Exception as e:
-        return jsonify({"reply": f"Internal Error: {str(e)}"}), 500
+        return jsonify({"reply": f"System Alert: {str(e)}"}), 500
 
 @app.route('/')
 def health():
-    return "🐘 Elephant AI Core: ONLINE"
+    return "🐘 Elephant AI Core: ONLINE | WEB INTEL: ACTIVE"
 
 if __name__ == "__main__":
     app.run(host='0.0.0.0', port=7860)