Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,5 @@
 import os
-os.environ["HOME"] = "/tmp"  # Hugging Face permission fix
+os.environ["HOME"] = "/tmp"  # Fix for Hugging Face Spaces
 
 import streamlit as st
 import requests
@@ -10,7 +10,7 @@ st.set_page_config(page_title="WikiTrail", layout="wide")
 st.title("📚 WikiTrail")
 st.markdown("Explore Wikipedia topics visually and get a summarized journey.")
 
-# Language selector
+# Language options
 languages = {
     "English": "en",
     "Hindi (हिन्दी)": "hi",
@@ -20,10 +20,11 @@ languages = {
 lang_name = st.selectbox("🌐 Select Language", list(languages.keys()))
 lang_code = languages[lang_name]
 
-# Topic input
-topic_input = st.text_input("🔍 Enter a topic (in English)", placeholder="e.g., India, Gandhi, Telangana")
+# Input
+topic_input = st.text_input("🔍 Enter a topic (in English)", placeholder="e.g., India, Telangana, Mahatma Gandhi")
+topic_input = topic_input.strip()
 
-# Translate topic using Wikipedia search
+# 🔍 Fix: Get best matching title or fallback
 def get_translated_title(query, lang):
     search_url = f"https://{lang}.wikipedia.org/w/api.php"
     params = {
@@ -33,15 +34,20 @@ def get_translated_title(query, lang):
         "format": "json",
         "origin": "*"
     }
-    res = requests.get(search_url, params=params)
-    if res.status_code == 200:
-        results = res.json()
-        search_list = results.get("query", {}).get("search", [])
-        if search_list:
-            return search_list[0]["title"]
-    return None
 
-# Fetch summary
+    try:
+        res = requests.get(search_url, params=params)
+        res.raise_for_status()
+        data = res.json()
+        search_results = data.get("query", {}).get("search", [])
+
+        if search_results:
+            return search_results[0]["title"]
+        return query  # fallback
+    except:
+        return query  # fallback
+
+# Summary API
 def fetch_summary(title, lang):
     safe_title = urllib.parse.quote(title.replace(" ", "_"))
     url = f"https://{lang}.wikipedia.org/api/rest_v1/page/summary/{safe_title}"
@@ -55,7 +61,7 @@ def fetch_summary(title, lang):
         }
     return None
 
-# Fetch related
+# Related Topics
 def fetch_related(title, lang):
     safe_title = urllib.parse.quote(title.replace(" ", "_"))
     url = f"https://{lang}.wikipedia.org/w/api.php?action=query&format=json&origin=*&titles={safe_title}&prop=links&pllimit=5"
@@ -67,38 +73,35 @@ def fetch_related(title, lang):
             return [link['title'] for link in pages[0]['links']]
     return []
 
-# Summarizer
+# Simple Summary
 def summarize_bullets(texts, limit=3):
     full_text = ' '.join(set(texts))
     sentences = full_text.replace('।', '.').replace('?', '.').replace('!', '.').split('.')
     clean = [s.strip() for s in sentences if s.strip()]
     return ["• " + s + "." for s in clean[:limit]] if clean else ["No summary available."]
 
-# Main logic
+# ✅ Main logic
 if topic_input:
     with st.spinner("🔍 Searching Wikipedia..."):
         summaries = []
         final_text = ""
 
         translated_title = get_translated_title(topic_input, lang_code)
-        if not translated_title:
+        st.caption(f"📄 Fetched title: {translated_title}")  # Debug info
+
+        main = fetch_summary(translated_title, lang_code)
+        if not main:
             st.error(f"No matching page found in {lang_name} for '{topic_input}'")
             st.stop()
 
-        # Main topic
         st.subheader("🔷 Main Topic")
-        main = fetch_summary(translated_title, lang_code)
-        if main:
-            summaries.append(main["summary"])
-            final_text += f"📚 {main['title']} - {lang_name} Wikipedia Summary\n\n"
-            final_text += main["summary"] + "\n\n"
-            st.markdown(f"### {main['title']}")
-            st.write(main["summary"])
-            st.markdown(f"[Read More →]({main['link']})", unsafe_allow_html=True)
-        else:
-            st.warning("Couldn't fetch main topic summary.")
+        summaries.append(main["summary"])
+        final_text += f"📚 {main['title']} - {lang_name} Wikipedia Summary\n\n"
+        final_text += main["summary"] + "\n\n"
+        st.markdown(f"### {main['title']}")
+        st.write(main["summary"])
+        st.markdown(f"[Read More →]({main['link']})", unsafe_allow_html=True)
 
-        # Related
         st.subheader("🔗 Related Topics")
         related_titles = fetch_related(translated_title, lang_code)
         if related_titles:
@@ -113,12 +116,11 @@ if topic_input:
         else:
             st.info("No related topics found.")
 
-        # Bullet summary
         st.subheader("🧠 Combined Summary")
         for bullet in summarize_bullets(summaries):
             st.markdown(bullet)
 
-        # Download button
+        # 📥 Download
         st.download_button(
             label="📥 Download Summary as TXT",
             data=final_text,