Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,4 +1,3 @@
-# app.py
 import os
 import tempfile
 import logging
@@ -25,34 +24,27 @@ translator = Translator()
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)
 
-# -------------------- Helper Functions --------------------
+# -------------------- Wikipedia Summary Fetch --------------------
 @st.cache_data(ttl=3600)
 def fetch_wikipedia_content(query, lang="en", is_fallback=False):
-    """
-    Fetch summary + page URL using the REST summary endpoint first,
-    then fallback to action=query if needed.
-    Returns: (summary_text or None, full_url_or_None, actual_lang)
-    """
     if not query:
         return None, None, lang
 
     title = query.strip()
     encoded_title = urllib.parse.quote(title, safe="")
 
-    # REST summary endpoint (preferred)
     rest_url = f"https://{lang}.wikipedia.org/api/rest_v1/page/summary/{encoded_title}"
     try:
         resp = requests.get(rest_url, headers=WIKIMEDIA_HEADERS, timeout=REQUEST_TIMEOUT)
         resp.raise_for_status()
         j = resp.json()
-        summary = j.get("extract") or j.get("description") or None
+        summary = j.get("extract") or j.get("description")
         full_url = j.get("content_urls", {}).get("desktop", {}).get("page")
         if summary:
             return summary, full_url, lang
     except Exception as e:
-        logger.debug("REST summary failed: %s", e)
+        logger.debug("REST failed: %s", e)
 
-    # Fallback to action=query API if REST didn't yield a summary
     try:
         api = f"https://{lang}{WIKIPEDIA_API_BASE_URL_PREFIX}"
         params = {
@@ -68,23 +60,49 @@ def fetch_wikipedia_content(query, lang="en", is_fallback=False):
         }
         res = requests.get(api, params=params, headers=WIKIMEDIA_HEADERS, timeout=REQUEST_TIMEOUT)
         res.raise_for_status()
-        data = res.json().get("query", {}).get("pages", [])
-        if data:
-            page = data[0] if isinstance(data, list) else next(iter(data.values()))
-            if not page.get("missing", False):
-                summary = page.get("extract")
-                full_url = page.get("fullurl")
-                if summary:
-                    return summary, full_url, lang
+        pages = res.json().get("query", {}).get("pages", [])
+        if pages and not pages[0].get("missing"):
+            return pages[0].get("extract"), pages[0].get("fullurl"), lang
     except Exception as e:
-        logger.debug("Query API fallback failed: %s", e)
+        logger.debug("Query API failed: %s", e)
 
     if not is_fallback and lang != "en":
         return fetch_wikipedia_content(query, lang="en", is_fallback=True)
 
     return None, None, lang
 
-# -------------------- Narration Function --------------------
+# -------------------- Wikipedia Image Fetch --------------------
+@st.cache_data(ttl=3600)
+def fetch_wikipedia_images(query, lang="en", max_images=3):
+    if not query:
+        return []
+
+    api = f"https://{lang}.wikipedia.org/w/api.php"
+    params = {
+        "action": "query",
+        "format": "json",
+        "prop": "pageimages",
+        "piprop": "original",
+        "titles": query,
+        "redirects": 1
+    }
+
+    try:
+        res = requests.get(api, params=params, headers=WIKIMEDIA_HEADERS, timeout=REQUEST_TIMEOUT)
+        res.raise_for_status()
+        pages = res.json().get("query", {}).get("pages", {})
+        images = []
+
+        for page in pages.values():
+            if "original" in page:
+                images.append(page["original"]["source"])
+
+        return images[:max_images]
+    except Exception as e:
+        logger.debug("Image fetch failed: %s", e)
+        return []
+
+# -------------------- Narration --------------------
 def narrate_text(text, lang_code):
     try:
         clean = text.strip().replace("\n", " ")
@@ -97,68 +115,61 @@ def narrate_text(text, lang_code):
             tts.save(fp.name)
             audio_path = fp.name
 
-        st.success("📣 Narration ready! Click Play below:")
         st.audio(audio_path, format="audio/mp3")
     except Exception as e:
-        st.error(f"🔇 Narration error: {e}")
+        st.error(f"Narration error: {e}")
 
-# -------------------- Translation Function --------------------
+# -------------------- Translation --------------------
 def translate_summary(text, target_lang):
     try:
-        result = translator.translate(text, target_lang)
-        return result.result
-    except Exception as e:
-        st.warning(f"⚠️ Translation failed. Showing original English summary.\n\nError: {e}")
+        return translator.translate(text, target_lang).result
+    except Exception:
         return text
 
 # -------------------- Streamlit UI --------------------
 st.set_page_config(page_title="WikiHealth Navigator", layout="centered")
 st.title("🩺 WikiHealth Navigator")
-st.markdown("Search health topics using Wikipedia across Indian languages. (No images)")
+st.markdown("Search health topics using Wikipedia (with images & audio)")
+
+languages = {
+    "English": "en",
+    "हिन्दी": "hi",
+    "తె��ుగు": "te",
+    "தமிழ்": "ta"
+}
 
-languages = {"English (English)": "en", "हिन्दी (Hindi)": "hi", "తెలుగు (Telugu)": "te", "தமிழ் (Tamil)": "ta"}
 col1, col2 = st.columns([2, 3])
 with col1:
     selected = st.selectbox("Select Language:", list(languages.keys()))
     lang_code = languages[selected]
 with col2:
-    query = st.text_input("Enter a health-related topic:")
-
-for key in ["summary", "article_url", "actual_lang", "translated_summary"]:
-    if key not in st.session_state:
-        st.session_state[key] = None
-
-if st.button("🔍 Search"):
-    if query:
-        with st.spinner(f"Searching '{query}' in {selected}..."):
-            summary, full_url, actual_lang = fetch_wikipedia_content(query, lang=lang_code)
-            if summary:
-                st.session_state.summary = summary
-                st.session_state.article_url = full_url
-                st.session_state.actual_lang = actual_lang
-                if actual_lang != lang_code:
-                    translated = translate_summary(summary, lang_code)
-                    st.session_state.translated_summary = translated
-                else:
-                    st.session_state.translated_summary = summary
+    query = st.text_input("Enter a health topic:")
+
+if st.button("🔍 Search") and query:
+    with st.spinner("Searching Wikipedia..."):
+        summary, url, actual_lang = fetch_wikipedia_content(query, lang_code)
+
+        if summary:
+            if actual_lang != lang_code:
+                summary = translate_summary(summary, lang_code)
+
+            st.subheader("📝 Summary")
+            st.write(summary)
+
+            st.subheader("🖼️ Related Images")
+            images = fetch_wikipedia_images(query, lang_code)
+
+            if images:
+                cols = st.columns(len(images))
+                for col, img in zip(cols, images):
+                    col.image(img, use_container_width=True)
             else:
-                st.session_state.summary = None
-                st.session_state.translated_summary = None
-                st.warning(f"No summary found for '{query}'.")
-
-if st.session_state.summary:
-    st.subheader("📝 Summary")
-    st.write(st.session_state.translated_summary or st.session_state.summary)
-
-    button_label = {
-        "en": "🔊 Narrate Summary",
-        "hi": "सारांश सुनाइए",
-        "te": "సారాంశాన్ని వివరించండి",
-        "ta": "சுருக்கமாக விவரிக்கவும்"
-    }.get(lang_code, "🔊 Narrate Summary")
-
-    if st.button(button_label):
-        narrate_text(st.session_state.translated_summary or st.session_state.summary, lang_code)
-
-    if st.session_state.article_url:
-        st.markdown(f"[Read full article on Wikipedia]({st.session_state.article_url})")
+                st.info("No images found.")
+
+            if st.button("🔊 Narrate Summary"):
+                narrate_text(summary, lang_code)
+
+            if url:
+                st.markdown(f"[Read full article on Wikipedia]({url})")
+        else:
+            st.warning("No information found.")