improved

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import pymupdf
 from duckduckgo_search import DDGS
+from serpapi import GoogleSearch
 
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
@@ -25,6 +26,25 @@ def search_web(query):
     return "No relevant results found on the web."
 
 
+def google_search(query):
+    params = {
+        "q": query,
+        "api_key": "b11d4a3660600e7e7f481b3288f107fbf993389a20125b0a97ebe7ab207854a5",  # Replace this with your real key
+        "engine": "google",
+    }
+    search = GoogleSearch(params)
+    results = search.get_dict()
+    if "organic_results" in results:
+        # Combine top 3 results
+        summaries = []
+        for res in results["organic_results"][:3]:
+            title = res.get("title", "")
+            snippet = res.get("snippet", "")
+            summaries.append(f"{title}: {snippet}")
+        return "\n".join(summaries)
+    return None
+
+
 SYSTEM_PROMPT = """
 You are an intelligent and friendly AI assistant. 
 
@@ -37,6 +57,7 @@ Your goals:
 - Always be polite, helpful, and respectful.
 """
 
+
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -45,6 +66,53 @@ def respond(
     top_p,
 ):
 
+    recent_keywords = [
+        "latest",
+        "today",
+        "current",
+        "now",
+        "recent",
+        "news",
+        "update",
+        "price",
+        "who won",
+        "what happened",
+        "trending",
+        "breaking",
+        "just in",
+        "new release",
+        "live",
+        "score",
+        "results",
+        "weather",
+        "forecast",
+        "report",
+        "market",
+        "stocks",
+        "crypto",
+        "rate",
+        "exchange",
+        "gold price",
+        "happening",
+        "event",
+        "updates",
+        "hot",
+        "viral",
+        "announcement",
+        "today's",
+        "this week",
+        "schedule",
+        "calendar",
+        "launch",
+        "drop",
+        "release date",
+        "opening",
+        "closing",
+        "deadline",
+    ]
+
+    message_lower = message.lower()
+
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
 
     for val in history:
@@ -53,6 +121,12 @@ def respond(
         if val[1]:
             messages.append({"role": "assistant", "content": val[1]})
 
+    if any(kw in message_lower for kw in recent_keywords):
+        web_context = google_search(message)
+        if web_context:
+            # Inject web context as part of the user's query
+            message = f"{message}\n\n[Relevant web search results to help you answer]:\n{web_context}"
+
     messages.append({"role": "user", "content": message})
 
     response = ""

requirements.txt

@@ -1,3 +1,4 @@
 huggingface_hub==0.25.2
 duckduckgo_search
-pymupdf
+pymupdf
+google-search-results