Update app.py

app.py

@@ -1,31 +1,172 @@
-import requests, base64
-
-invoke_url = "https://integrate.api.nvidia.com/v1/chat/completions"
-stream = False
-
-def read_b64(path):
-  with open(path, "rb") as f:
-    return base64.b64encode(f.read()).decode()
-
-headers = {
-  "Authorization": "Bearer $NVIDIA_API_KEY",
-  "Accept": "text/event-stream" if stream else "application/json"
-}
-
-payload = {
-  "model": "google/diffusiongemma-26b-a4b-it",
-  "messages": [{"role":"user","content":""}],
-  "max_tokens": 4096,
-  "temperature": 1.00,
-  "top_p": 0.95,
-  "stream": stream,
-  "chat_template_kwargs": {"enable_thinking":True},
-}
-
-response = requests.post(invoke_url, headers=headers, json=payload, stream=stream)
-if stream:
-    for line in response.iter_lines():
-        if line:
-            print(line.decode("utf-8"))
-else:
-    print(response.json())
+import os
+import requests
+import urllib.parse
+from bs4 import BeautifulSoup
+from datetime import datetime, timedelta, timezone
+from flask import Flask, request, Response, stream_with_context, render_template_string
+
+app = Flask(__name__)
+
+# 🔐 --- SECURE ENVIRONMENT VARIABLES ---
+# यह आपके Hugging Face Secrets से अपने आप लोड हो जाएगा
+API_KEY = os.environ.get("NVIDIA_API_KEY") or os.environ.get("YOUR_VEDIKA_API_KEY") 
+MODEL_ID = os.environ.get("MODEL_ID")
+
+# NVIDIA का डायरेक्ट URL (BASE_URL का लॉजिक पूरी तरह हटा दिया गया है)
+INVOKE_URL = "https://integrate.api.nvidia.com/v1/chat/completions"
+
+# 🌐 --- BULLETPROOF SCRAPER (News + Wiki) ---
+def web_search_scraper(query, num_results=4):
+    results = []
+    headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64)"}
+
+    # 1. Google News RSS (Never blocks, gives real latest news)
+    try:
+        rss_url = f"https://news.google.com/rss/search?q={urllib.parse.quote(query)}&hl=en-IN&gl=IN&ceid=IN:en"
+        res = requests.get(rss_url, headers=headers, timeout=5)
+        soup = BeautifulSoup(res.content, 'xml') # Use XML parser
+        items = soup.find_all('item')
+        
+        for item in items[:num_results]:
+            title = item.title.text if item.title else ""
+            link = item.link.text if item.link else ""
+            pub_date = item.pubDate.text if item.pubDate else ""
+            if title:
+                results.append({
+                    "title": title,
+                    "link": link,
+                    "snippet": f"Published: {pub_date}"
+                })
+    except Exception as e:
+        pass
+
+    # 2. Wikipedia API Fallback (Never blocks)
+    if len(results) < 2:
+        try:
+            wiki_url = f"https://en.wikipedia.org/w/api.php?action=query&list=search&srsearch={urllib.parse.quote(query)}&utf8=&format=json"
+            res = requests.get(wiki_url, headers=headers, timeout=5).json()
+            for item in res.get('query', {}).get('search', [])[:num_results]:
+                clean_snippet = BeautifulSoup(item['snippet'], "html.parser").text
+                results.append({
+                    "title": item['title'], 
+                    "link": f"https://en.wikipedia.org/wiki/{urllib.parse.quote(item['title'])}", 
+                    "snippet": clean_snippet
+                })
+        except Exception as e:
+            pass
+
+    return results
+
+# ----------------------------------------------------
+
+@app.route('/')
+def home():
+    try:
+        with open('index.html', 'r', encoding='utf-8') as f:
+            return render_template_string(f.read())
+    except Exception as e:
+        return f"<h1>System Error</h1><p>index.html missing: {str(e)}</p>"
+
+@app.route('/api/chat', methods=['POST'])
+def chat():
+    if not API_KEY or not INVOKE_URL or not MODEL_ID:
+        return Response("Server Error: Secrets missing.", status=500)
+
+    data = request.get_json() or {}
+    user_message = data.get("message", "")
+    attachments = data.get("attachments", []) 
+    is_search = data.get("is_search", False)
+    history = data.get("history", []) 
+    max_tokens = data.get("max_tokens", 4096)
+    temperature = data.get("temperature", 1.00) # Updated to NVIDIA default if not passed
+
+    # 🕒 --- REAL-TIME IST INJECTION ---
+    ist_time = datetime.now(timezone.utc) + timedelta(hours=5, minutes=30)
+    current_date = ist_time.strftime("%A, %d %B %Y, %I:%M %p IST")
+
+    # 🧠 --- STRICT SYSTEM PROMPT (ANTI-HALLUCINATION) ---
+    system_prompt = f"""
+    You are CODE VED, an advanced AI System engineered EXCLUSIVELY by DIVY PATEL.
+    Current Date: {current_date}.
+    
+    STRICT DIRECTIVES:
+    1. NEVER invent, guess, or hallucinate product launches, dates, news, or facts.
+    2. If you are provided with live search data, base your answer PURELY on that data.
+    3. If you lack data to answer a question about a recent event, you MUST say: "I do not have the verified live data for this right now."
+    """
+
+    # 🚀 --- AUTO-SEARCH LOGIC WITH FAILSAFE ---
+    if is_search:
+        scraped_data = web_search_scraper(user_message)
+        
+        if scraped_data:
+            search_context = "\n\n--- [LIVE VERIFIED WEB DATA] ---\n"
+            for idx, res in enumerate(scraped_data):
+                search_context += f"{idx+1}. TITLE: {res['title']}\nSNIPPET: {res['snippet']}\nURL: {res['link']}\n\n"
+            
+            search_context += "[SYSTEM: Use the above verified data to answer. Cite sources.]"
+        else:
+            search_context = """
+            \n\n[CRITICAL SYSTEM ALERT: LIVE SEARCH FAILED OR BLOCKED. 
+            YOU HAVE NO NEW DATA FOR THIS QUERY. 
+            MANDATORY RULE: DO NOT INVENT ANY NEWS OR DATES. TELL THE USER HONESTLY THAT YOU COULD NOT FETCH THE LATEST LIVE DATA.]
+            """
+            
+        user_message = f"{search_context}\n\nUSER QUERY: {user_message}"
+
+    messages = [{"role": "system", "content": system_prompt}]
+
+    for msg in history:
+        role = msg.get("role", "user")
+        content = msg.get("content", "")
+        if content:
+            messages.append({"role": role, "content": content})
+
+    content_payload = []
+    if user_message.strip():
+        content_payload.append({"type": "text", "text": user_message})
+        
+    for att in attachments:
+        att_type = att.get("type")
+        b64_data = att.get("data")
+        if att_type == "image":
+            # Image formatting compatible with standard APIs
+            content_payload.append({"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{b64_data}"}})
+        elif att_type in ["audio", "file"]:
+            content_payload.append({"type": "input_audio", "input_audio": {"data": b64_data, "format": "wav"}})
+
+    if not content_payload:
+        content_payload.append({"type": "text", "text": "Hello"})
+
+    messages.append({"role": "user", "content": content_payload})
+
+    headers = {
+        "Authorization": f"Bearer {API_KEY}", 
+        "Accept": "text/event-stream"
+    }
+    
+    # 🌟 --- NEW NVIDIA FORMAT PAYLOAD INTEGRATED HERE --- 🌟
+    payload = {
+        "model": MODEL_ID,  # Dynamically fetching from your secrets
+        "messages": messages,
+        "max_tokens": int(max_tokens),
+        "temperature": float(temperature),
+        "top_p": 0.95, # As per your format
+        "stream": True, # Stream is kept True for fast UI response
+        "chat_template_kwargs": {"enable_thinking": True} # 👈 New NVIDIA feature added
+    }
+
+    try:
+        response = requests.post(INVOKE_URL, headers=headers, json=payload, stream=True)
+        def generate():
+            for line in response.iter_lines():
+                if line:
+                    decoded_line = line.decode("utf-8")
+                    if decoded_line.startswith("data: "):
+                        yield decoded_line + "\n\n"
+        return Response(stream_with_context(generate()), mimetype='text/event-stream')
+    except Exception as e:
+        return Response(f"Internal Error: {str(e)}", status=500)
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)