Update app.py

app.py

@@ -3,25 +3,28 @@ import requests
 import uvicorn
 from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import HTMLResponse
-from fastapi.staticfiles import StaticFiles
+from duckduckgo_search import DDGS
 
 app = FastAPI()
 
-# 🔱 CORS Policy "Unleashed" - බ්‍රවුසරයේ ඇති සියලුම බාධක ඉවත් කරයි
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
+    allow_origins=["*"], 
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# 🔱 Configuration
-# Hugging Face Secrets වලින් Key එක ලබා ගනී (Invisible to public)
 API_KEY = os.getenv("OPENROUTER_KEY")
 MODEL = "tencent/hy3-preview:free"
-OPENROUTER_URL = "https://openrouter.ai/api/v1/chat/completions"
+
+def get_web_info(query):
+    """DuckDuckGo හරහා නොමිලේ අන්තර්ජාලය පීරීම"""
+    try:
+        with DDGS() as ddgs:
+            results = [r['body'] for r in ddgs.text(query, max_results=3)]
+            return "\n".join(results)
+    except Exception:
+        return "No real-time info found."
 
 @app.post("/generate")
 async def generate(request: Request):
@@ -29,32 +32,32 @@ async def generate(request: Request):
         data = await request.json()
         user_prompt = data.get("prompt")
         
+        # 🔱 පියවර 1: අන්තර්ජාලයෙන් තොරතුරු ලබා ගැනීම
+        web_context = get_web_info(user_prompt)
+        
+        # 🔱 පියවර 2: මොඩලයට ලබා දෙන උපදෙස් (Hybrid System Prompt)
         headers = {
             "Authorization": f"Bearer {API_KEY}",
-            "HTTP-Referer": "https://huggingface.co/spaces", 
+            "HTTP-Referer": "https://huggingface.co",
             "Content-Type": "application/json"
         }
         
         payload = {
             "model": MODEL,
             "messages": [
-                {"role": "system", "content": "You are Elephant AI Pro, a master level coding assistant built by MINZO-PRIME. Your logic is powered by Tencent Hy3."},
+                {
+                    "role": "system", 
+                    "content": f"You are Elephant AI Pro with DuckDuckGo Search. Current Web Info: {web_context}. Use this info to provide accurate, real-time answers. Master: MINZO-PRIME."
+                },
                 {"role": "user", "content": user_prompt}
-            ],
-            "temperature": 0.7
+            ]
         }
         
-        response = requests.post(OPENROUTER_URL, headers=headers, json=payload, timeout=60)
+        response = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload)
         return response.json()
         
     except Exception as e:
         return {"error": str(e)}
 
-# 🔱 Main Page Loader
-@app.get("/", response_class=HTMLResponse)
-async def read_index():
-    with open("index.html", "r", encoding="utf-8") as f:
-        return f.read()
-
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)