Update app.py

app.py

@@ -1,5 +1,7 @@
 import os
 import requests
+import urllib.parse
+from bs4 import BeautifulSoup
 from flask import Flask, request, Response, stream_with_context, render_template_string
 
 app = Flask(__name__)
@@ -17,6 +19,53 @@ if BASE_URL:
     else:
         INVOKE_URL = BASE_URL
 
+
+# 🌐 --- GOOGLE SEARCH SCRAPER ENGINE --- 🌐
+def google_search_scraper(query, num_results=4):
+    """
+    यह फंक्शन Google सर्च से डेटा निकालता है और उसे AI के समझने लायक बनाता है।
+    """
+    # Google को धोखा देने के लिए असली ब्राउज़र का User-Agent
+    headers = {
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
+    }
+    url = f"https://www.google.com/search?q={urllib.parse.quote(query)}&hl=en"
+    
+    try:
+        response = requests.get(url, headers=headers, timeout=10)
+        soup = BeautifulSoup(response.text, 'html.parser')
+        results = []
+        
+        # Google के सर्च रिजल्ट्स 'div.g' के अंदर होते हैं
+        for g in soup.find_all('div', class_='g'):
+            title_elem = g.find('h3')
+            link_elem = g.find('a')
+            
+            if title_elem and link_elem:
+                title = title_elem.text
+                link = link_elem.get('href', '')
+                
+                # Snippet (डिस्क्रिप्शन) निकालना
+                snippet = ""
+                snippet_box = g.find('div', class_='VwiC3b') 
+                if snippet_box:
+                    snippet = snippet_box.text
+                
+                results.append({
+                    "title": title,
+                    "link": link,
+                    "snippet": snippet
+                })
+                
+                if len(results) >= num_results:
+                    break
+                    
+        return results
+    except Exception as e:
+        return [{"error": str(e)}]
+# ---------------------------------------------
+
+
 @app.route('/')
 def home():
     try:
@@ -36,20 +85,36 @@ def chat():
     user_message = data.get("message", "")
     attachments = data.get("attachments", []) 
     system_prompt = data.get("system_prompt", "")
-    
-    # 🛠️ FIX 1: यहाँ फ्रंटएंड से आ रही हिस्ट्री को रिसीव किया
     history = data.get("history", []) 
-    
     max_tokens = data.get("max_tokens", 4096)
     temperature = data.get("temperature", 0.6)
 
+    # 🚀 --- INTERCEPT & SEARCH LOGIC --- 🚀
+    # अगर यूजर के मैसेज में /search या /google है, तो स्क्रैपर चालू करें
+    if user_message.strip().lower().startswith("/search "):
+        search_query = user_message[8:].strip()
+        scraped_data = google_search_scraper(search_query)
+        
+        search_context = f"Real-time Google Search Results for '{search_query}':\n\n"
+        for idx, res in enumerate(scraped_data):
+            if "error" in res:
+                search_context += f"Search Error: {res['error']}\n"
+            else:
+                search_context += f"{idx+1}. Title: {res['title']}\nSnippet: {res['snippet']}\nLink: {res['link']}\n\n"
+        
+        search_context += "\n[INSTRUCTION FOR AI: Formulate a comprehensive, professional response to the user's query using ONLY the real-time search data provided above. Cite sources if applicable.]"
+        
+        # एआई के लिए प्रॉम्प्ट को मॉडिफाई करना (यूजर को यह बैकग्राउंड डेटा नहीं दिखेगा)
+        user_message = f"User Query: {search_query}\n\n[SYSTEM BACKGROUND CONTEXT]\n{search_context}"
+    # ---------------------------------------
+
     messages = []
 
     # 1. सिस्टम प्रॉम्प्ट
     if system_prompt.strip():
         messages.append({"role": "system", "content": system_prompt})
 
-    # 🛠️ FIX 2: पुरानी हिस्ट्री को मॉडल के पेलोड में जोड़ना (इससे यह पुरानी बातें याद रखेगा)
+    # 2. पुरानी हिस्ट्री को मॉडल के पेलोड में जोड़ना
     for msg in history:
         role = msg.get("role", "user")
         content = msg.get("content", "")
@@ -106,4 +171,4 @@ def chat():
         return Response("Internal Error: Unable to process request securely.", status=500)
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+    app.run(host='0.0.0.0', port=7860)