Update app.py

app.py

@@ -1,39 +1,41 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-from serpapi import GoogleSearch  # Import SerpAPI
-
-SERPAPI_KEY = "your_serpapi_key_here"  # Replace with your API key
+from selenium import webdriver
+from selenium.webdriver.common.by import By
+from selenium.webdriver.chrome.service import Service
+from webdriver_manager.chrome import ChromeDriverManager
+import time
 
 client = InferenceClient("Futuresony/future_ai_12_10_2024.gguf")
 
 def is_uncertain(question, response):
     """Check if the model's response is unreliable."""
-    if len(response.split()) < 4:  # Too short = likely incorrect
-        return True
-    if response.lower() in question.lower():  # Repeats question = unsure
-        return True
-    uncertain_phrases = [
-        "Kulingana na utafiti", "Inaaminika kuwa", "Ninadhani", 
-        "It is believed that", "Some people say", "Inasemekana kuwa"
-    ]
-    if any(phrase.lower() in response.lower() for phrase in uncertain_phrases):
+    if len(response.split()) < 4 or response.lower() in question.lower():
         return True
-    return False
+    uncertain_phrases = ["Kulingana na utafiti", "Inaaminika kuwa", "Ninadhani", "It is believed that", "Some people say"]
+    return any(phrase.lower() in response.lower() for phrase in uncertain_phrases)
 
 def google_search(query):
-    """Fetch search results using SerpAPI."""
-    params = {
-        "q": query,
-        "hl": "en",
-        "gl": "us",
-        "api_key": SERPAPI_KEY
-    }
-    search = GoogleSearch(params)
-    results = search.get_dict()
-    
-    if "organic_results" in results:
-        return results["organic_results"][0]["snippet"]  # First search result
-    return "Sorry, I couldn't find an answer on Google."
+    """Fetch search results using Selenium."""
+    options = webdriver.ChromeOptions()
+    options.add_argument("--headless")  # Run in background
+    driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options)
+
+    driver.get(f"https://www.google.com/search?q={query}")
+    time.sleep(2)  # Wait for page to load
+
+    try:
+        # Extract answer from featured snippet if available
+        snippet = driver.find_element(By.CLASS_NAME, "hgKElc").text
+    except:
+        # Extract first search result
+        try:
+            snippet = driver.find_element(By.CSS_SELECTOR, "div.BNeawe.s3v9rd.AP7Wnd").text
+        except:
+            snippet = "Sorry, I couldn't find an answer on Google."
+
+    driver.quit()
+    return snippet
 
 def respond(message, history, system_message, max_tokens, temperature, top_p):
     messages = [{"role": "system", "content": system_message}]
@@ -48,7 +50,6 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
         response += token
         yield response  # Stream the response
 
-    # If the model's response is unreliable, fetch from Google
     if is_uncertain(message, response):
         google_response = google_search(message)
         yield f"🤖 AI: {response}\n\n🌍 Google: {google_response}"