Serper

app.py

@@ -1,38 +1,43 @@
 import gradio as gr
 import requests
+import os
 
-def wikipedia_search(query):
-    url = "https://en.wikipedia.org/w/api.php"
-    params = {
-        "action": "query",
-        "format": "json",
-        "prop": "extracts",
-        "exintro": True,
-        "explaintext": True,
-        "titles": query,
+# You should set your Serper API key as an environment variable
+SERPER_API_KEY = "37228930df7532489158d04f41333979853b1422"
+
+def serper_search(query):
+    url = "https://google.serper.dev/search"
+    payload = {"q": query}
+    headers = {
+        "X-API-KEY": SERPER_API_KEY,
+        "Content-Type": "application/json"
     }
-    response = requests.get(url, params=params)
+    
+    response = requests.post(url, json=payload, headers=headers)
     data = response.json()
-    page = next(iter(data["query"]["pages"].values()))
-    return page.get("extract", "Sorry, I couldn't find any information on that topic.")
+    
+    if 'organic' in data and len(data['organic']) > 0:
+        return data['organic'][0]['snippet']
+    else:
+        return "Sorry, I couldn't find any information on that topic."
 
 def chatbot_response(message, history):
-    response = wikipedia_search(message)
+    response = serper_search(message)
     # Truncate the response if it's too long
     return response[:500] + "..." if len(response) > 500 else response
 
 iface = gr.ChatInterface(
     chatbot_response,
     chatbot=gr.Chatbot(height=400),
-    textbox=gr.Textbox(placeholder="Ask about a topic...", container=False, scale=7),
-    title="Wikipedia Search Chatbot",
-    description="This chatbot searches Wikipedia for information on the topic you ask about.",
+    textbox=gr.Textbox(placeholder="Ask a question...", container=False, scale=7),
+    title="Serper Search Chatbot",
+    description="This chatbot uses Serper API to search for information on the topic you ask about.",
     theme="soft",
     examples=[
-        "Python programming language",
-        "Artificial Intelligence",
-        "Gradio library",
-        "Machine Learning",
+        "What is machine learning?",
+        "Latest news about AI",
+        "How does blockchain work?",
+        "Benefits of exercise",
     ],
     cache_examples=True,
     retry_btn=None,