Updated wiki function

app.py

@@ -45,13 +45,33 @@ def save_chat_history(messages):
     with shelve.open("chat_history") as db:
         db["messages"] = messages
 
-def fetch_wikipedia_link(topic):
+'''def fetch_wikipedia_link(topic):
     api_url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{topic}"
     response = requests.get(api_url)
     if response.status_code == 200:
         return response.json()['content_urls']['desktop']['page']
     st.error("Failed to fetch Wikipedia content.")
-    return None
+    return None'''
+
+from urllib.parse import quote
+
+def fetch_wikipedia_link(topic: str) -> str | None:
+    try:
+        encoded_topic = quote(topic)
+        api_url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{encoded_topic}"
+        response = requests.get(api_url, timeout=5)
+
+        if response.status_code != 200:
+            return None
+
+        data = response.json()
+        return data.get("content_urls", {}) \
+                   .get("desktop", {}) \
+                   .get("page")
+
+    except requests.RequestException:
+        return None
+
 
 load_css()
 initialize_session_state()