Updated Wiki function

app.py

@@ -45,13 +45,34 @@ 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:
+        # URL encode the topic
+        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:
+            st.error(f"Failed to fetch Wikipedia content for '{topic}' (status {response.status_code})")
+            return None
+
+        data = response.json()
+        return data.get("content_urls", {}).get("desktop", {}).get("page")
+
+    except requests.RequestException as e:
+        st.error(f"Network error: {e}")
+        return None
 
 
 load_css()