Update app.py

app.py

@@ -1,41 +1,107 @@
-from flask import Flask, render_template, request
-import requests
-import os
+from flask import Flask, render_template, request, jsonify
+from time import sleep
+from bs4 import BeautifulSoup
+from requests import get
+import urllib
 
 app = Flask(__name__)
 
-BASE_URL = "https://oevortex-webscout-api.hf.space"
+def _req(term, results, lang, start, proxies, timeout, safe, ssl_verify):
+    resp = get(
+        url="https://www.google.com/search",
+        headers={
+            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36 Edg/111.0.1661.62"
+        },
+        params={
+            "q": term,
+            "num": results + 2,  # Prevents multiple requests
+            "hl": lang,
+            "start": start,
+            "safe": safe,
+        },
+        proxies=proxies,
+        timeout=timeout,
+        verify=ssl_verify,
+    )
+    resp.raise_for_status()
+    return resp
+
+
+class SearchResult:
+    def __init__(self, url, title, description):
+        self.url = url
+        self.title = title
+        self.description = description
+
+    def __repr__(self):
+        return f"SearchResult(url={self.url}, title={self.title}, description={self.description})"
+
+    def to_dict(self):  # Add a method to convert to a dictionary
+        return {
+            "url": self.url,
+            "title": self.title,
+            "description": self.description,
+        }
+
+
+def search(term, num_results=10, lang="en", proxy=None, advanced=False, sleep_interval=0, timeout=5, safe="active", ssl_verify=None):
+    """Search the Google search engine"""
+
+    escaped_term = urllib.parse.quote_plus(term)  # make 'site:xxx.xxx.xxx ' works.
+
+    # Proxy
+    proxies = None
+    if proxy:
+        if proxy.startswith("https"):
+            proxies = {"https": proxy}
+        else:
+            proxies = {"http": proxy}
+
+    # Fetch
+    start = 0
+    results = []
+    while start < num_results:
+        # Send request
+        resp = _req(escaped_term, num_results - start,
+                    lang, start, proxies, timeout, safe, ssl_verify)
+
+        # Parse
+        soup = BeautifulSoup(resp.text, "html.parser")
+        result_block = soup.find_all("div", attrs={"class": "g"})
+        if len(result_block) == 0:
+            start += 1
+        for result in result_block:
+            # Find link, title, description
+            link = result.find("a", href=True)
+            title = result.find("h3")
+            description_box = result.find(
+                "div", {"style": "-webkit-line-clamp:2"})
+            if description_box:
+                description = description_box.text
+                if link and title and description:
+                    start += 1
+                    if advanced:
+                        results.append(SearchResult(link["href"], title.text, description))
+                    else:
+                        results.append(link["href"])
+        sleep(sleep_interval)
+
+    return results
 
 @app.route("/", methods=["GET", "POST"])
 def index():
     return render_template("index.html")
 
-@app.route("/api/suggestions", methods=["GET"])
-def get_suggestions():
-    search_query = request.args.get("q")
-    api_endpoint = f"{BASE_URL}/api/suggestions?q={search_query}"
-    response = requests.get(api_endpoint)
-    return response.json()
-
-
 @app.route("/api/search", methods=["GET"])
 def perform_search():
     search_query = request.args.get("q")
-    results = request.args.get("max_results", "10")
-    time_limit = request.args.get("timelimit", "none")
-    safe_search = request.args.get("safesearch", "off")
-    api_endpoint = f"{BASE_URL}/api/search?q={search_query}&max_results={results}&timelimit={time_limit}&safesearch={safe_search}"
-    response = requests.get(api_endpoint)
-    return response.json()
-
-@app.route("/api/answers", methods=["GET"])
-def get_people_also_search():
-    search_query = request.args.get("q")
-    api_endpoint = f"{BASE_URL}/api/answers?q={search_query}"
-    response = requests.get(api_endpoint)
-    return response.json()
+    max_results = int(request.args.get("max_results", 10))  # Default to 10 results
+
+    search_results = search(term=search_query, num_results=max_results, advanced=True)
 
+    # Convert search results to a list of dictionaries
+    results_json = [result.to_dict() for result in search_results]
+    return jsonify(results_json) 
 
 if __name__ == "__main__":
-    port = int(os.environ.get('PORT', 7860)) 
-    app.run(host='0.0.0.0', port=port)
+    app.run(debug=True)