Update src/streamlit_app.py

src/streamlit_app.py

@@ -8,24 +8,31 @@ from translatepy import Translator
 # -------------------- CONFIG --------------------
 REQUEST_TIMEOUT = 20
 HEADERS = {
-    "User-Agent": "WikiHealthNavigator/1.0 (edu-project)"
+    "User-Agent": "WikiHealthNavigator/1.0 (educational-project)"
 }
 translator = Translator()
 
-# -------------------- FETCH SUMMARY --------------------
+# -------------------- FETCH SUMMARY (SAFE + FALLBACK) --------------------
 @st.cache_data(ttl=3600)
 def fetch_summary(query, lang):
     try:
         title = urllib.parse.quote(query.strip())
         url = f"https://{lang}.wikipedia.org/api/rest_v1/page/summary/{title}"
         r = requests.get(url, headers=HEADERS, timeout=REQUEST_TIMEOUT)
-        r.raise_for_status()
+
+        # Handle missing pages (404 etc.)
+        if r.status_code != 200:
+            return None, None
+
         data = r.json()
-        return data.get("extract"), data.get("content_urls", {}).get("desktop", {}).get("page")
+        return (
+            data.get("extract"),
+            data.get("content_urls", {}).get("desktop", {}).get("page")
+        )
     except Exception:
         return None, None
 
-# -------------------- FETCH IMAGES (SAFE VERSION) --------------------
+# -------------------- FETCH IMAGES (WIKIMEDIA COMMONS – SAFE) --------------------
 @st.cache_data(ttl=3600)
 def fetch_images(query, max_images=4):
     api = "https://commons.wikimedia.org/w/api.php"
@@ -43,18 +50,13 @@ def fetch_images(query, max_images=4):
     try:
         r = requests.get(api, params=params, headers=HEADERS, timeout=REQUEST_TIMEOUT)
 
-        # ✅ Ensure response is JSON
         if r.status_code != 200:
             return []
 
         if "application/json" not in r.headers.get("Content-Type", ""):
             return []
 
-        try:
-            data = r.json()
-        except ValueError:
-            return []
-
+        data = r.json()
         pages = data.get("query", {}).get("pages", {})
         images = []
 
@@ -63,7 +65,6 @@ def fetch_images(query, max_images=4):
                 images.append(page["imageinfo"][0]["url"])
 
         return images
-
     except Exception:
         return []
 
@@ -77,16 +78,16 @@ def narrate(text, lang):
     except Exception as e:
         st.error(f"Audio error: {e}")
 
-# -------------------- SESSION STATE --------------------
+# -------------------- SESSION STATE INIT --------------------
 for key in ["summary", "images", "url", "lang"]:
     if key not in st.session_state:
         st.session_state[key] = None
 
 # -------------------- UI --------------------
-st.set_page_config("WikiHealth Navigator", layout="centered")
+st.set_page_config(page_title="WikiHealth Navigator", layout="centered")
 st.title("🩺 WikiHealth Navigator")
 st.markdown(
-    "Health search using Wikipedia text + "
+    "Health topic search using **Wikipedia** text and "
     "**** images"
 )
 
@@ -104,13 +105,16 @@ query = st.text_input("Enter health topic")
 
 # -------------------- SEARCH --------------------
 if st.button("🔍 Search") and query:
+    # 1️⃣ Try selected language
     summary, url = fetch_summary(query, lang_code)
 
-    if not summary and lang_code != "en":
+    # 2️⃣ Fallback to English if page not found
+    if not summary:
         summary, url = fetch_summary(query, "en")
 
-    if summary and lang_code != "en":
-        summary = translator.translate(summary, lang_code).result
+        # 3️⃣ Translate English → selected language
+        if summary and lang_code != "en":
+            summary = translator.translate(summary, lang_code).result
 
     st.session_state.summary = summary
     st.session_state.images = fetch_images(query)
@@ -125,8 +129,8 @@ if st.session_state.summary:
     st.subheader("🖼️ Related Images")
     if st.session_state.images:
         cols = st.columns(len(st.session_state.images))
-        for c, img in zip(cols, st.session_state.images):
-            c.image(img, use_container_width=True)
+        for col, img in zip(cols, st.session_state.images):
+            col.image(img, use_container_width=True)
     else:
         st.info("Images temporarily unavailable. Please try again.")