Update src/streamlit_app.py

src/streamlit_app.py

@@ -16,10 +16,28 @@ languages = {
 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")
+topic_input = st.text_input("Enter a topic (in English)", placeholder="e.g., India")
 
-def fetch_topic_summary(topic, lang):
-    url = f"https://{lang}.wikipedia.org/api/rest_v1/page/summary/{topic}"
+def get_localized_title(english_title, lang):
+    """Search for the localized title in the target language."""
+    url = f"https://{lang}.wikipedia.org/w/api.php"
+    params = {
+        "action": "query",
+        "list": "search",
+        "srsearch": english_title,
+        "format": "json",
+        "origin": "*"
+    }
+    response = requests.get(url, params=params)
+    if response.status_code == 200:
+        data = response.json()
+        search_results = data.get("query", {}).get("search", [])
+        if search_results:
+            return search_results[0]["title"]
+    return None
+
+def fetch_topic_summary(title, lang):
+    url = f"https://{lang}.wikipedia.org/api/rest_v1/page/summary/{title}"
     res = requests.get(url)
     if res.status_code == 200:
         data = res.json()
@@ -31,8 +49,8 @@ def fetch_topic_summary(topic, lang):
         }
     return None
 
-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"
+def fetch_related_topics(title, lang):
+    url = f"https://{lang}.wikipedia.org/w/api.php?action=query&format=json&origin=*&titles={title}&prop=links&pllimit=5"
     res = requests.get(url)
     if res.status_code == 200:
         data = res.json()
@@ -42,17 +60,27 @@ def fetch_related_topics(topic, lang):
     return []
 
 def simple_summary(all_summaries, limit=3):
-    full_text = ' '.join(all_summaries)
-    sentences = full_text.split('. ')
-    return '. '.join(sentences[:limit]) + '.' if sentences else full_text
+    unique_summaries = list(set(all_summaries))
+    full_text = ' '.join(unique_summaries)
+    sentences = full_text.replace('।', '.').replace('?', '.').replace('!', '.').split('.')
+    sentences = [s.strip() for s in sentences if s.strip()]
+    return '. '.join(sentences[:limit]) + '.' if sentences else "No content to summarize."
 
-if topic:
-    with st.spinner("🔍 Searching Wikipedia..."):
+if topic_input:
+    with st.spinner("🔍 Translating & Searching Wikipedia..."):
         summaries = []
 
+        # Step 1: Get localized title from English input
+        localized_title = topic_input if lang_code == "en" else get_localized_title(topic_input, lang_code)
+
+        if not localized_title:
+            st.error(f"No page found for '{topic_input}' in {lang_name}. Try a simpler or different topic.")
+            st.stop()
+
+        # Step 2: Show Main Topic
         st.subheader("🔷 Main Topic")
-        main = fetch_topic_summary(topic, lang_code)
-        if main:
+        main = fetch_topic_summary(localized_title, lang_code)
+        if main and main["summary"]:
             summaries.append(main["summary"])
             st.markdown(f"### {main['title']}")
             if main["image"]:
@@ -61,13 +89,15 @@ if topic:
             st.markdown(f"[Read More →]({main['link']})", unsafe_allow_html=True)
         else:
             st.warning("Couldn't fetch data for the main topic.")
+            st.stop()
 
+        # Step 3: Related Topics
         st.subheader("🔗 Related Topics")
-        related = fetch_related_topics(topic, lang_code)
+        related = fetch_related_topics(localized_title, lang_code)
         if related:
             for rel in related:
                 data = fetch_topic_summary(rel, lang_code)
-                if data:
+                if data and data["summary"] not in summaries:
                     summaries.append(data["summary"])
                     with st.expander(data["title"]):
                         if data["image"]:
@@ -77,5 +107,6 @@ if topic:
         else:
             st.info("No related topics found.")
 
+        # Step 4: Combined Summary
         st.subheader("🧠 Combined Summary")
         st.success(simple_summary(summaries))