Update app.py

app.py

@@ -1,16 +1,36 @@
 import os
 import random
 import requests
+import json
+import feedparser
 from flask import Flask, request, Response, stream_with_context
 from flask_cors import CORS
 
 app = Flask(__name__)
-# 🔱 ලෝකයේ ඕනෑම තැනක සිට එන HTML එකකට සම්බන්ධ වීමට ඉඩ දීම
 CORS(app)
 
-# Hugging Face Secrets වලින් Keys ටික ලබා ගැනීම
+# 🔱 SECURE KEY LOADING & CLEANING
 RAW_KEYS = os.getenv("OR_KEYS", "")
-API_KEYS = RAW_KEYS.split(",") if RAW_KEYS else []
+API_KEYS = [k.strip() for k in RAW_KEYS.replace("\n", ",").split(",") if k.strip()]
+
+MODEL_NAME = "minimax/minimax-m2.5:free"
+
+# 🔱 WEB SEARCH & INTEL SYSTEM (RSS & Google News Engine)
+def fetch_web_intel(query):
+    intel = ""
+    # RSS News Feeds (Security, Tech & Global News)
+    RSS_FEEDS = [
+        f'https://news.google.com/rss/search?q={query}&hl=en-US&gl=US&ceid=US:en',
+        'https://feeds.feedburner.com/TheHackersNews'
+    ]
+    
+    for url in RSS_FEEDS:
+        try:
+            feed = feedparser.parse(url)
+            for entry in feed.entries[:3]: # පළමු ප්‍රතිඵල 3 බැගින්
+                intel += f"\n- Title: {entry.title}\n  Source: {entry.link}\n"
+        except: pass
+    return intel
 
 @app.route('/api/chat', methods=['POST'])
 def chat():
@@ -20,7 +40,9 @@ def chat():
     user_data = request.json
     user_msg = user_data.get("message", "")
     
-    # Random ලෙස Key එකක් තෝරා ගැනීම (Key Rotation)
+    # පද්ධතිය සජීවීව අන්තර්ජාලයේ තොරතුරු සෙවීම
+    web_data = fetch_web_intel(user_msg)
+    
     current_key = random.choice(API_KEYS)
     
     headers = {
@@ -28,32 +50,51 @@ def chat():
         "Content-Type": "application/json"
     }
     
+    # System Prompt එක Elephant AI ලෙස සකස් කිරීම
+    system_instruction = (
+        "You are Elephant AI, a powerful real-time technical assistant created by Commander MINZO-PRIME. "
+        "Use the provided web research data to give accurate, up-to-date answers. "
+        f"Live Web Data: {web_data if web_data else 'No live data found, use internal knowledge.'}"
+    )
+    
     payload = {
-        "model": "minimax/minimax-m2.5:free",
-        "messages": [{"role": "user", "content": user_msg}],
+        "model": MODEL_NAME,
+        "messages": [
+            {"role": "system", "content": system_instruction},
+            {"role": "user", "content": user_msg}
+        ],
         "stream": True
     }
 
     def generate():
-        resp = requests.post(
-            "https://openrouter.ai/api/v1/chat/completions",
-            headers=headers,
-            json=payload,
-            stream=True
-        )
-        for line in resp.iter_lines():
-            if line:
-                decoded = line.decode('utf-8')
-                if "data: " in decoded and "[DONE]" not in decoded:
-                    try:
-                        import json
-                        chunk = json.loads(decoded.replace("data: ", ""))
-                        content = chunk['choices'][0]['delta'].get('content', '')
-                        yield content
-                    except: pass
+        try:
+            resp = requests.post(
+                "https://openrouter.ai/api/v1/chat/completions",
+                headers=headers,
+                json=payload,
+                stream=True,
+                timeout=40
+            )
+            
+            for line in resp.iter_lines():
+                if line:
+                    decoded = line.decode('utf-8')
+                    if decoded.startswith("data: "):
+                        data_str = decoded[6:]
+                        if data_str == "[DONE]": break
+                        try:
+                            chunk = json.loads(data_str)
+                            content = chunk['choices'][0]['delta'].get('content', '')
+                            yield content
+                        except: pass
+        except Exception as e:
+            yield f"\n[Elephant Core Error: {str(e)}]"
 
     return Response(stream_with_context(generate()), mimetype='text/plain')
 
+@app.route('/')
+def health():
+    return f"🐘 Elephant AI (Web-Search Enabled) is Online. Keys Active: {len(API_KEYS)}"
+
 if __name__ == "__main__":
-    # HF Spaces සඳහා අනිවාර්යයෙන්ම port 7860 විය යුතුයි
     app.run(host='0.0.0.0', port=7860)