Update app.py

shifted to qwant API search

app.py

@@ -7,54 +7,60 @@ client = InferenceClient("mistralai/Mistral-Nemo-Instruct-2407")
 
 def translate_to_english(text: str) -> str:
     try:
-        # Translate the text to English.
         return GoogleTranslator(source='auto', target='en').translate(text)
     except Exception:
         return text
 
 def translate_to_bisaya(text: str) -> str:
     try:
-        # 'ceb' is the ISO code for Cebuano/Bisaya.
         return GoogleTranslator(source='auto', target='ceb').translate(text)
     except Exception:
         return text
 
 def get_internet_data(query: str) -> str:
-    url = "https://api.duckduckgo.com"
+    """
+    Uses Qwant's free search API to fetch a snippet based on the query.
+    """
+    url = "https://api.qwant.com/v3/search/web"
     params = {
         "q": query,
-        "format": "json",
-        "no_redirect": 1,
-        "skip_disambig": 1,
+        "count": 10,
+        "offset": 0,
+        "t": "web",
+        "safesearch": 1,
+        "locale": "en_US",
+        "uiv": 4,
     }
     try:
         response = requests.get(url, params=params, timeout=5)
         response.raise_for_status()
         data = response.json()
-        result = data.get("AbstractText", "")
-        if not result:
-            result = "Wala koy nakuha nga impormasyon gikan sa internet."
+        items = data.get("data", {}).get("result", {}).get("items", [])
+        if items:
+            snippet = items[0].get("desc", "")
+            if not snippet:
+                snippet = items[0].get("title", "")
+        else:
+            snippet = "Wala koy nakuha nga impormasyon gikan sa Qwant search."
     except Exception:
-        result = "Naay problema sa pagkuha sa impormasyon gikan sa internet."
-    return result
+        snippet = "Naay problema sa pagkuha sa impormasyon gikan sa Qwant search."
+    return snippet
 
 def respond(message, history: list[tuple[str, str]]):
     # Step 1: Translate the query from Bisaya to English.
     english_query = translate_to_english(message)
     
-    # Step 2: Search the web using the translated query.
+    # Step 2: Search the web using Qwant's API with the translated query.
     search_result = get_internet_data(english_query)
     
-    # Step 3: Translate the search result to Bisaya.
+    # Step 3: Translate the search result back to Bisaya.
     bisaya_search_result = translate_to_bisaya(search_result)
     
-    # Enrich the original query with the Bisaya search result.
+    # Enrich the original query with the translated search result.
     enriched_message = (
-        f"{message}\n\nMga resulta gikan sa internet (isinalin sa bisaya): "
-        f"{bisaya_search_result}"
+        f"{message}\n\nMga resulta gikan sa internet (isinalin sa bisaya): {bisaya_search_result}"
     )
     
-    # Build the conversation messages.
     system_message = (
         "Ikaw usa ka buotan nga Chatbot. Tubaga lang sa binisaya. "
         "Gamiton ang bag-ong kasayuran nga nakuha gikan sa internet. "
@@ -63,7 +69,7 @@ def respond(message, history: list[tuple[str, str]]):
     max_tokens = 4096
     temperature = 0.6
     top_p = 0.95
-    
+
     messages = [{"role": "system", "content": system_message}]
     for user_text, assistant_text in history:
         if user_text:
@@ -72,7 +78,7 @@ def respond(message, history: list[tuple[str, str]]):
             messages.append({"role": "assistant", "content": assistant_text})
     messages.append({"role": "user", "content": enriched_message})
     
-    # Collect the full response from the model.
+    # Get the complete response from the model.
     full_response = ""
     for token_message in client.chat_completion(
         messages,
@@ -87,8 +93,8 @@ def respond(message, history: list[tuple[str, str]]):
         full_response += token
         if len(full_response) > 3000:
             break
-    
-    # Post-translate the final answer to Bisaya.
+
+    # Translate the final response to Bisaya.
     final_response = translate_to_bisaya(full_response)
     yield final_response