Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-import wikipediaapi
+import wikipedia
 from PIL import Image
 import requests
 import torch
@@ -9,8 +9,6 @@ from torchvision.models import resnet50
 
 # Initialize inference client for chat
 chat_client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
-# Initialize Wikipedia API
-wiki_wiki = wikipediaapi.Wikipedia('en')
 
 # Load pre-trained image classification model
 model = resnet50(pretrained=True)
@@ -23,14 +21,15 @@ transform = transforms.Compose([
 ])
 
 def search_wikipedia(query):
-    page = wiki_wiki.page(query)
-    if page.exists():
-        return page.summary
-    else:
+    try:
+        summary = wikipedia.summary(query)
+        return summary
+    except wikipedia.exceptions.DisambiguationError as e:
+        return f"Disambiguation error: {e}"
+    except wikipedia.exceptions.PageError:
         return "No information found on that topic."
 
 def respond(message, history, system_message, max_tokens, temperature, top_p):
-    # Search Wikipedia for information
     search_response = search_wikipedia(message)
     
     # Prepare the chat messages