Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,10 +6,20 @@ st.set_page_config(page_title="WikiTrail", layout="wide")
 st.markdown("<h1 style='text-align: center;'>📚 WikiTrail</h1>", unsafe_allow_html=True)
 st.markdown("<p style='text-align: center;'>Explore Wikipedia topics visually and get a summarized journey.</p>", unsafe_allow_html=True)
 
+# Language selector
+languages = {
+    "English": "en",
+    "Hindi (हिन्दी)": "hi",
+    "Telugu (తెలుగు)": "te",
+    "Tamil (தமிழ்)": "ta"
+}
+lang_name = st.selectbox("🌐 Select Language", list(languages.keys()), index=0)
+lang_code = languages[lang_name]
+
 topic = st.text_input("Enter a topic", placeholder="e.g., India")
 
-def fetch_topic_summary(topic):
-    url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{topic}"
+def fetch_topic_summary(topic, lang):
+    url = f"https://{lang}.wikipedia.org/api/rest_v1/page/summary/{topic}"
     res = requests.get(url)
     if res.status_code == 200:
         data = res.json()
@@ -21,8 +31,8 @@ def fetch_topic_summary(topic):
         }
     return None
 
-def fetch_related_topics(topic):
-    url = f"https://en.wikipedia.org/w/api.php?action=query&format=json&origin=*&titles={topic}&prop=links&pllimit=5"
+def fetch_related_topics(topic, lang):
+    url = f"https://{lang}.wikipedia.org/w/api.php?action=query&format=json&origin=*&titles={topic}&prop=links&pllimit=5"
     res = requests.get(url)
     if res.status_code == 200:
         data = res.json()
@@ -34,14 +44,14 @@ def fetch_related_topics(topic):
 def simple_summary(all_summaries, limit=3):
     full_text = ' '.join(all_summaries)
     sentences = full_text.split('. ')
-    return '. '.join(sentences[:limit]) + '.'
+    return '. '.join(sentences[:limit]) + '.' if sentences else full_text
 
 if topic:
     with st.spinner("🔍 Searching Wikipedia..."):
         summaries = []
 
         st.subheader("🔷 Main Topic")
-        main = fetch_topic_summary(topic)
+        main = fetch_topic_summary(topic, lang_code)
         if main:
             summaries.append(main["summary"])
             st.markdown(f"### {main['title']}")
@@ -53,10 +63,10 @@ if topic:
             st.warning("Couldn't fetch data for the main topic.")
 
         st.subheader("🔗 Related Topics")
-        related = fetch_related_topics(topic)
+        related = fetch_related_topics(topic, lang_code)
         if related:
             for rel in related:
-                data = fetch_topic_summary(rel)
+                data = fetch_topic_summary(rel, lang_code)
                 if data:
                     summaries.append(data["summary"])
                     with st.expander(data["title"]):