Update search.py

search.py

@@ -1,6 +1,13 @@
-# search.py
 from googlesearch import search
 
+PREFERRED_DOMAINS = [
+    'amazon.in', 'flipkart.com', 'wikipedia.org', 
+    'makemytrip.com', 'yatra.com', 'skyscanner.co.in'
+]
+
 def search_google(query, num_results=5):
-    """Search Google and return list of URLs."""
-    return list(search(query, num_results=num_results))
+    raw_urls = list(search(query, num_results=num_results))
+    # Prioritize known trusted domains
+    preferred = [u for u in raw_urls if any(dom in u for dom in PREFERRED_DOMAINS)]
+    others = [u for u in raw_urls if u not in preferred]
+    return preferred + others