Update app.py

app.py

@@ -33,29 +33,33 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-import wikipedia
-
+import wikipediaapi
+        
 @tool
 def get_wikipedia_summary(query: str) -> str:
-    """
-    A tool that fetches a Wikipedia summary for a given topic.
-
+    """ A tool that fetches a Wikipedia summary for a given topic.
     Args:
-        query: The topic to search for in Wikipedia.
-
+        query (str): The topic to search for in Wikipedia.
     Returns:
-        A short summary of the topic from Wikipedia.
+        str: A short summary of the topic from Wikipedia or an error message if the page does not exist.
     """
-
     try:
-        summary = wikipedia.summary(query, sentences=2)
-        return f"Here is a short summary of {query} from Wikipedia: {summary}"
-    except wikipedia.exceptions.DisambiguationError as e:
-        return f"The topic '{query}' is ambiguous. Try a more specific term. Suggestions: {e.options[:5]}"
-    except wikipedia.exceptions.PageError:
-        return f"Sorry, no page was found for '{query}' on Wikipedia."
+        # Initialize the Wikipedia API object
+        wiki_wiki = wikipediaapi.Wikipedia('en')
+
+        # Fetch the page for the given query
+        page = wiki_wiki.page(query)
+
+        # Check if the page exists
+        if not page.exists():
+            return f"Sorry, no page was found for '{query}' on Wikipedia."
+
+        # Return a truncated summary (up to 500 characters)
+        return f"Here is a short summary of '{query}': {page.summary[:500]}..."
+    
     except Exception as e:
-        return f"An error occurred: {str(e)}"
+        # Catch any unexpected errors and return a user-friendly message
+        return f"An error occurred while fetching the Wikipedia summary: {str(e)}"
 
 
 final_answer = FinalAnswerTool()