Update app.py

app.py

@@ -153,18 +153,53 @@ def ask_ollama(user_message, model='llama-3.3-70b-versatile', system_prompt=sear
             "1": "Error parsing response. Please try again.",
             "2": "Error parsing response. Please try again."
         }
-
+# Using DuckDuckGo API
+# def search_web(topic, max_references=5, data_type="pdf"):
+#     """Search the web using DuckDuckGo and return results."""
+#     doc_list = []
+#     with DDGS(verify=False) as ddgs:
+#         i = 0
+#         for r in ddgs.text(topic, region='wt-wt', safesearch='On', timelimit='n'):
+#             if i >= max_references:
+#                 break
+#             doc_list.append({"type": data_type, "title": r['title'], "body": r['body'], "url": r['href']})
+#             i += 1
+#     return doc_list
+# Using Brave Search API
 def search_web(topic, max_references=5, data_type="pdf"):
-    """Search the web using DuckDuckGo and return results."""
-    doc_list = []
-    with DDGS(verify=False) as ddgs:
-        i = 0
-        for r in ddgs.text(topic, region='wt-wt', safesearch='On', timelimit='n'):
-            if i >= max_references:
-                break
-            doc_list.append({"type": data_type, "title": r['title'], "body": r['body'], "url": r['href']})
-            i += 1
-    return doc_list
+    """Search the web using the Brave API and return results."""
+    url = "https://game4all-serpent.hf.space/search_brave"
+    payload = {
+        "queries": [topic],
+        "n_results": max_references
+    }
+
+    try:
+        response = requests.post(url, json=payload)
+        response.raise_for_status()
+        data = response.json()
+        
+        if data.get("error"):
+            raise ValueError(f"API Error: {data['error']}")
+
+        results = data.get("results", [])
+        doc_list = [
+            {
+                "type": data_type,
+                "title": r.get("title", ""),
+                "body": r.get("body", ""),
+                "url": r.get("href", "")
+            }
+            for r in results[:max_references]
+        ]
+        return doc_list
+
+    except requests.RequestException as e:
+        print(f"HTTP Request failed: {e}")
+        return []
+    except ValueError as ve:
+        print(ve)
+        return []
 
 def analyze_pdf_novelty(patent_background, url, data_type="pdf"):
     """Extract full document text from PDF or background from patent and evaluate novelty"""