Update app.py

app.py

@@ -1,34 +1,61 @@
 from flask import Flask, request, jsonify
-from urllib.parse import quote
+from urllib.parse import quote, urlparse, parse_qs, unquote
 import requests
 from bs4 import BeautifulSoup
 
 app = Flask(__name__)
 
-# DuckDuckGo search API
-@app.route('/search', methods=['GET'])
-def search_duckduckgo():
-    query = request.args.get('query')
-    if not query:
-        return jsonify([]), 400
-
+def duckduckgo_search(query):
     url = 'https://duckduckgo.com/html/'
     params = {'q': query}
     headers = {'User-Agent': 'Mozilla/5.0'}
-    
+
     response = requests.get(url, params=params, headers=headers)
     if response.status_code != 200:
-        return jsonify([]), response.status_code
+        return []
 
     soup = BeautifulSoup(response.text, 'html.parser')
     results = []
-    for result in soup.find_all('a', class_='result__a'):
-        title = result.get_text()
-        link = result['href']
+    for result in soup.find_all('div', class_='result'):
+        title_tag = result.find('a', class_='result__a')
+        if not title_tag:
+            continue
+        title = title_tag.get_text()
+        link = title_tag['href']
+        
+        # Decode the actual URL from the DuckDuckGo redirect link
+        parsed_link = urlparse(link)
+        query_params = parse_qs(parsed_link.query)
+        actual_url = unquote(query_params.get('uddg', [''])[0])
+
+        description_tag = result.find('a', class_='result__snippet')
+        description = description_tag.get_text() if description_tag else 'No description available'
+        
+        # Attempt to fetch the favicon
+        icon = 'No icon available'
+        if actual_url:
+            parsed_actual_url = urlparse(actual_url)
+            favicon_url = f"{parsed_actual_url.scheme}://{parsed_actual_url.netloc}/favicon.ico"
+            favicon_response = requests.get(favicon_url, headers=headers)
+            if favicon_response.status_code == 200:
+                icon = favicon_url
+
         results.append({
             'title': title,
-            'link': link
+            'link': actual_url,
+            'description': description,
+            'icon': icon
         })
+    return results
+
+# DuckDuckGo search API
+@app.route('/search', methods=['GET'])
+def search_duckduckgo():
+    query = request.args.get('query')
+    if not query:
+        return jsonify([]), 400
+
+    results = duckduckgo_search(query)
     return jsonify(results)
 
 # Google Image Search API
@@ -56,4 +83,4 @@ def get_image_urls():
     return jsonify(image_urls)
 
 if __name__ == '__main__':
-    app.run(host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0", port=7860)